Update app.py

app.py

@@ -1,239 +1,132 @@
-import streamlit as st
 import requests
 import json
 import os
-import io
-import base64
 
-# --- Configuration ---
-TALKBOT_TTS_URL = "https://talkbot.ir/TTS-tkun"
-TALKBOT_CHAT_API_URL = "https://talkbot.ir/api/v1/chat/completions"
-DEEPSEEK_API_KEY = "sk-4fb61f56acfccf731e801b904cd89f5" # This key is publicly visible, consider environment variables for production!
+# اطلاعات API TTS
+TTS_API_BASE_URL = "https://talkbot.ir/TTS-tkun"
+TTS_API_FORMAT = "?text="
 
-# --- Helper Functions ---
-
-@st.cache_data
-def get_tts_audio_url(text: str) -> str:
-    """Fetches a WAV audio URL from Talkbot TTS."""
-    try:
-        params = {"text": text}
-        response = requests.get(TALKBOT_TTS_URL, params=params)
-        response.raise_for_status()  # Raise an exception for bad status codes
-        return response.text.strip()
-    except requests.exceptions.RequestException as e:
-        st.error(f"Error fetching TTS audio URL for '{text}': {e}")
-        return ""
-
-@st.cache_data
-def generate_podcast_script_ai(topic: str, num_sections: int = 3) -> str:
-    """Generates a podcast script using the Talkbot Deepseek API."""
-    prompt = (
-        f"Generate a concise podcast script about '{topic}'. "
-        f"The script should have {num_sections} main sections, each starting with a clear heading. "
-        "Each section should contain interesting facts or engaging discussions. "
-        "Conclude with a short outro. The script should be conversational and engaging. "
-        "Keep the total length reasonable for a short podcast (e.g., 300-500 words)."
-        "\n\nFormat your response as follows:\n\n"
-        "Section 1: [Title of Section 1]\n[Content of Section 1]\n\n"
-        "Section 2: [Title of Section 2]\n[Content of Section 2]\n\n"
-        # Add more sections as needed based on num_sections if hardcoding
-        # For simplicity, we'll let the AI decide the exact number based on prompt
-        "..."
-        "Outro:\n[Outro content]"
-    )
+# اطلاعات API OpenAI (برای Talkbot.ir)
+OPENAI_API_BASE_URL = "https://talkbot.ir/api/v1/chat/completions"
+OPENAI_API_KEY = "sk-4fb61f56acfcc731e801b904cd89f5" # دقت کنید که این کلید API از Talkbot.ir است و ممکن است با OpenAI اصلی متفاوت باشد.
+OPENAI_MODEL = "deepseek-v3-0324"
 
+def get_text_from_ai(prompt_text):
+    """
+    تولید متن پادکست با استفاده از API هوش مصنوعی.
+    """
     headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {DEEPSEEK_API_KEY}"
+        "Authorization": f"Bearer {OPENAI_API_KEY}",
+        "Content-Type": "application/json"
     }
 
     payload = {
-        "model": "deepseek-v3-0324",
+        "model": OPENAI_MODEL,
         "messages": [
-            {"role": "system", "content": "You are a helpful podcast script writer."},
-            {"role": "user", "content": prompt}
+            {"role": "system", "content": "You are a helpful assistant that generates podcast scripts."},
+            {"role": "user", "content": f"Generate a short podcast script about: {prompt_text}"}
         ],
-        "max_tokens": 700,
-        "temperature": 0.7,
-        "top_p": 1,
-        "frequency_penalty": 0,
-        "presence_penalty": 0
+        "max_tokens": 500,
+        "temperature": 0.7
     }
 
     try:
-        response = requests.post(TALKBOT_CHAT_API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        result = response.json()
-        if result and 'choices' in result and result['choices']:
-            return result['choices'][0]['message']['content'].strip()
+        response = requests.post(OPENAI_API_BASE_URL, headers=headers, json=payload)
+        response.raise_for_status()  # Raises an HTTPError for bad responses (4xx or 5xx)
+        data = response.json()
+        if data and data.get("choices"):
+            return data["choices"][0]["message"]["content"].strip()
         else:
-            st.error("AI response did not contain content.")
-            return "Could not generate podcast script. Please try again."
+            print("AI response did not contain expected data.")
+            return None
     except requests.exceptions.RequestException as e:
-        st.error(f"Error generating podcast script with AI: {e}")
-        return "Could not generate podcast script. API error."
+        print(f"Error connecting to AI API: {e}")
+        return None
     except json.JSONDecodeError:
-        st.error("Could not parse AI response JSON.")
-        return "Could not generate podcast script. Invalid AI response."
-
-def create_audio_player_from_url(audio_url: str, caption: str = ""):
-    """Creates an HTML audio player for a given URL."""
-    if audio_url:
-        st.audio(audio_url, format="audio/wav", caption=caption)
-    else:
-        st.write(f"No audio available for: {caption}")
-
-def get_audio_from_url_as_download(audio_url: str, filename: str):
-    """Fetches audio from a URL and returns it as a bytes object for download."""
+        print("Error decoding JSON response from AI API.")
+        return None
+
+def get_audio_wav_link(text):
+    """
+    دریافت لینک فایل WAV از API تبدیل متن به گفتار.
+    """
+    encoded_text = requests.utils.quote(text) # Encode URL-unsafe characters
+    tts_url = f"{TTS_API_BASE_URL}{TTS_API_FORMAT}{encoded_text}"
+    
     try:
-        response = requests.get(audio_url, stream=True)
-        response.raise_for_status()
-        audio_bytes = io.BytesIO(response.content).read()
-        return audio_bytes
+        # برای این API، فقط ساختار URL مهم است، نیازی به درخواست مستقیم برای دریافت لینک نیست.
+        # فرض می‌شود که API بلافاصله لینک wav را به عنوان پاسخ GET برمی‌گرداند.
+        # اما طبق توضیحات شما، خروجی به صورت لینک wav است.
+        # این یعنی درخواست GET به آن URL، مستقیماً لینک را برمی‌گرداند.
+        # اگر API مستقیماً لینک WAV را در بدنه پاسخ برمی‌گرداند، می‌توان از request.text استفاده کرد.
+        # در غیر این صورت، شاید باید یک درخواست واقعی انجام داد.
+        # با توجه به "خروجی صوت به صورت لینک wav است"، فرض می‌شود که خود URL ورودی، لینک WAV خروجی است.
+        # اگر API یک JSON با لینک wav برمی‌گرداند، باید کد را تغییر داد.
+        return tts_url
+        
     except requests.exceptions.RequestException as e:
-        st.error(f"Error downloading audio from {audio_url}: {e}")
+        print(f"Error generating TTS link: {e}")
         return None
 
-# --- Streamlit UI ---
-
-st.set_page_config(
-    page_title="پادکست ساز هوشمند",
-    page_icon="🎙️",
-    layout="centered"
-)
-
-st.title("🎙️ پادکست ساز هوشمند")
-st.markdown("پادکست خود را با هوش مصنوعی و صدای طبیعی بسازید!")
-
-with st.expander("درباره این ابزار"):
-    st.write(
-        """
-        این ابزار به شما کمک می‌کند تا به راحتی پادکست تولید کنید:
-        1. **ایده پادکست**: موضوع پادکست خود را وارد کنید.
-        2. **تولید متن**: هوش مصنوعی (Deepseek) متنی جذاب برای پادکست شما تولید می‌کند.
-        3. **تولید صدا**: متن تولید شده به دو صدای مجزا (صدای اول و صدای دوم) تبدیل می‌شود.
-        4. **پخش و دانلود**: می‌توانید پادکست را گوش دهید یا فایل‌های صوتی آن را دانلود کنید.
-
-        **توجه**: کلید API شما برای Talkbot Deepseek باید معتبر باشد.
-        """
-    )
-    st.markdown("[سورس کد در گیتهاب](لینک-گیتهاب-خود-را-اینجا-قرار-دهید)") # Add your GitHub link here
-
-# --- Input Section ---
-st.header("۱. موضوع پادکست را وارد کنید")
-podcast_topic = st.text_input(
-    "موضوع پادکست چیست؟",
-    placeholder="مثال: تاریخچه قهوه، فیزیک کوانتوم به زبان ساده، مزایای مدیتیشن"
-)
-
-if st.button("تولید پادکست", type="primary") and podcast_topic:
-    st.session_state['podcast_generated'] = False
-    st.session_state['script'] = None
-    st.session_state['audio1_url'] = None
-    st.session_state['audio2_url'] = None
-
-    with st.spinner(" در حال تولید متن پادکست با هوش مصنوعی..."):
-        script_content = generate_podcast_script_ai(podcast_topic)
-        st.session_state['script'] = script_content
-
-    if script_content and script_content != "Could not generate podcast script. Please try again.":
-        st.success("متن پادکست با موفقیت تولید شد!")
-        st.subheader("۲. متن پادکست تولید شده:")
-        st.code(script_content, language="text")
-
-        st.info("💡 می‌توانید متن را ویرایش کنید تا برای تولید صدا بهتر باشد.")
-        # Allow editing the script
-        edited_script = st.text_area("ویرایش متن پادکست (اختیاری):", value=script_content, height=300)
-        st.session_state['edited_script'] = edited_script
-
-        # Split script for two voices
-        # A simple split strategy:
-        # Voice 1 takes the first half, Voice 2 takes the second half.
-        # This can be made more sophisticated (e.g., based on paragraphs, specific markers).
-        lines = edited_script.split('\n')
-        mid_point = len(lines) // 2
-        voice1_text = "\n".join(lines[:mid_point])
-        voice2_text = "\n".join(lines[mid_point:])
-
-        st.subheader("۳. تولید صدا")
-
-        col1, col2 = st.columns(2)
-
-        with col1:
-            st.write("**صدای اول:**")
-            st.write(voice1_text)
-            if st.button("تولید صدای اول"):
-                with st.spinner("در حال تولید صدای اول..."):
-                    audio1_url = get_tts_audio_url(voice1_text)
-                    st.session_state['audio1_url'] = audio1_url
-                    if audio1_url:
-                        st.session_state['podcast_generated'] = True
-                        st.success("صدای اول با موفقیت تولید شد!")
-                    else:
-                        st.error("خطا در تولید صدای اول. لطفاً دوباره تلاش کنید.")
-
-        with col2:
-            st.write("**صدای دوم:**")
-            st.write(voice2_text)
-            if st.button("تولید صدای دوم"):
-                with st.spinner("در حال تولید صدای دوم..."):
-                    audio2_url = get_tts_audio_url(voice2_text)
-                    st.session_state['audio2_url'] = audio2_url
-                    if audio2_url:
-                        st.session_state['podcast_generated'] = True
-                        st.success("صدای دوم با موفقیت تولید شد!")
-                    else:
-                        st.error("خطا در تولید صدای دوم. لطفاً دوباره تلاش کنید.")
-
-    elif script_content: # This means an error message was returned
-        st.error("خطا در تولید متن پادکست.")
-
-# --- Display Section (rendered after the script is generated) ---
-if 'podcast_generated' in st.session_state and st.session_state['podcast_generated']:
-    st.subheader("۴. پادکست شما آماده است!")
-
-    col1_player, col2_player = st.columns(2)
-
-    with col1_player:
-        if st.session_state['audio1_url']:
-            st.markdown("### صدای اول")
-            create_audio_player_from_url(st.session_state['audio1_url'], "صدای اول")
-            audio1_bytes = get_audio_from_url_as_download(st.session_state['audio1_url'], "podcast_voice1.wav")
-            if audio1_bytes:
-                st.download_button(
-                    label="دانلود صدای اول",
-                    data=audio1_bytes,
-                    file_name="podcast_voice1.wav",
-                    mime="audio/wav"
-                )
+def main():
+    print("--------------------------------------------------")
+    print("             Talkbot Podcast Generator            ")
+    print("--------------------------------------------------")
+    print("این اسکریپت یک متن پادکست تولید و لینک‌های صوتی را فراهم می‌کند.")
+    print(" برای خروج، 'exit' را تایپ کنید.")
+    print("--------------------------------------------------")
+
+    while True:
+        topic = input("\nموضوع پادکست را وارد کنید (یا 'exit' برای خروج): ")
+        if topic.lower() == 'exit':
+            break
+
+        print(f"\nدر حال تولید اسکریپت پادکست برای '{topic}' با هوش مصنوعی...")
+        podcast_script = get_text_from_ai(topic)
+
+        if podcast_script:
+            print("\n------------------- اسکریپت پادکست -------------------")
+            print(podcast_script)
+            print("------------------------------------------------------")
+
+            # تقسیم متن به دو بخش برای صدای اول و صدای دوم
+            # این یک تقسیم‌بندی ساده است. ممکن است نیاز به منطق دقیق‌تری داشته باشید.
+            sentences = [s.strip() for s in podcast_script.split('.') if s.strip()]
+            
+            if len(sentences) >= 2:
+                speaker1_text = sentences[0]
+                speaker2_text = sentences[1]
+            elif len(sentences) == 1:
+                speaker1_text = sentences[0]
+                speaker2_text = "" # یا بخشی از جمله اول
+            else:
+                print("متن کافی برای تقسیم بین دو گوینده وجود ندارد.")
+                speaker1_text = podcast_script
+                speaker2_text = ""
+
+            print("\nدر حال ساخت لینک‌های صوتی...")
+            
+            audio_link_speaker1 = get_audio_wav_link(speaker1_text)
+            audio_link_speaker2 = get_audio_wav_link(speaker2_text) # اگر speaker2_text خالی باشد، لینک معناداری تولید نمی‌شود.
+
+            if audio_link_speaker1:
+                print(f"لینک صدای اول (Speaker 1): {audio_link_speaker1}")
+            else:
+                print("خطا در تولید لینک صدای اول.")
+            
+            if audio_link_speaker2 and speaker2_text: # فقط اگر متن برای گوینده دوم وجود داشته باشد
+                print(f"لینک صدای دوم (Speaker 2): {audio_link_speaker2}")
+            elif speaker2_text: # اگر speaker2_text داشتیم و لینکی تولید نشد
+                 print("خطا در تولید لینک صدای دوم.")
+            else:
+                print("متن کافی برای تولید صدای دوم وجود نداشت.")
+
+            print("\n------------------------------------------------------")
         else:
-            st.warning("صدای اول هنوز تولید نشده یا با خطا مواجه شده است.")
-
-    with col2_player:
-        if st.session_state['audio2_url']:
-            st.markdown("### صدای دوم")
-            create_audio_player_from_url(st.session_state['audio2_url'], "صدای دوم")
-            audio2_bytes = get_audio_from_url_as_download(st.session_state['audio2_url'], "podcast_voice2.wav")
-            if audio2_bytes:
-                st.download_button(
-                    label="دانلود صدای دوم",
-                    data=audio2_bytes,
-                    file_name="podcast_voice2.wav",
-                    mime="audio/wav"
-                )
-        else:
-            st.warning("صدای دوم هنوز تولید نشده یا با خطا مواجه شده است.")
-
-    st.markdown("---")
-    st.info("برای تولید پادکست جدید، موضوع جدیدی وارد کنید و دکمه 'تولید پادکست' را بزنید.")
+            print("متن پادکست از هوش مصنوعی دریافت نشد.")
+            print("لطفاً اتصال به اینترنت و کلید API را بررسی کنید.")
 
-# --- How to run this app ---
-# 1. Save the code as `app.py`.
-# 2. Make sure you have Streamlit installed: `pip install streamlit requests`
-# 3. Run from your terminal: `streamlit run app.py`
+    print("\nخروج از برنامه. موفق باشید!")
 
-# IMPORTANT SECURITY NOTE:
-# The Deepseek API key is hardcoded directly in the Python script.
-# For production applications, it's highly recommended to use environment variables
-# (e.g., `os.getenv("DEEPSEEK_API_KEY")`) and set them outside of your code
-# for security and flexibility. For this simple example as requested, it's left as is.
+if __name__ == "__main__":
+    main()