Update app.py

app.py

@@ -1,132 +1,109 @@
 import requests
-import json
-import os
 
-# اطلاعات API TTS
-TTS_API_BASE_URL = "https://talkbot.ir/TTS-tkun"
-TTS_API_FORMAT = "?text="
-
-# اطلاعات 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):
+def get_tts_link(text, voice_id):
     """
-    تولید متن پادکست با استفاده از API هوش مصنوعی.
-    """
-    headers = {
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
-        "Content-Type": "application/json"
-    }
-
-    payload = {
-        "model": OPENAI_MODEL,
-        "messages": [
-            {"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": 500,
-        "temperature": 0.7
-    }
+    از API Talkbot.ir برای تبدیل متن به صوت با صدای مشخص استفاده می کند.
+    Args:
+        text: متن ورودی برای تبدیل به صوت.
+        voice_id: شناسه صدا (مثلا 1 یا 2).
 
+    Returns:
+        لینک فایل صوتی WAV در صورت موفقیت، در غیر این صورت None.
+    """
+    url = f"https://talkbot.ir/TTS-tkun?text={requests.utils.quote(text)}&voice={voice_id}"
     try:
-        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()
+        response = requests.get(url)
+        # فرض می کنیم API در صورت موفقیت لینک مستقیم wav برمی گرداند
+        # نیاز به بررسی دقیق پاسخ واقعی API Talkbot.ir است.
+        # اگر پاسخ JSON است، نیاز به پارس کردن آن است.
+        # اگر پاسخ تنها لینک است، این کد کار می کند.
+        if response.status_code == 200:
+            return response.text.strip()  # فرض می کنیم پاسخ مستقیم لینک است
         else:
-            print("AI response did not contain expected data.")
+            print(f"خطا در دریافت صوت: {response.status_code}")
             return None
     except requests.exceptions.RequestException as e:
-        print(f"Error connecting to AI API: {e}")
-        return None
-    except json.JSONDecodeError:
-        print("Error decoding JSON response from AI API.")
+        print(f"خطا در ارسال درخواست: {e}")
         return None
 
-def get_audio_wav_link(text):
-    """
-    دریافت لینک فایل WAV از API تبدیل متن به گفتار.
+def generate_podcast_script(prompt):
     """
-    encoded_text = requests.utils.quote(text) # Encode URL-unsafe characters
-    tts_url = f"{TTS_API_BASE_URL}{TTS_API_FORMAT}{encoded_text}"
-    
-    try:
-        # برای این 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:
-        print(f"Error generating TTS link: {e}")
-        return None
+    با استفاده از API هوش مصنوعی متن پادکست را تولید می کند.
+    Args:
+        prompt: دستور برای تولید متن پادکست.
 
-def main():
-    print("--------------------------------------------------")
-    print("             Talkbot Podcast Generator            ")
-    print("--------------------------------------------------")
-    print("این اسکریپت یک متن پادکست تولید و لینک‌های صوتی را فراهم می‌کند.")
-    print(" برای خروج، 'exit' را تایپ کنید.")
-    print("--------------------------------------------------")
-
-    while True:
-        topic = input("\nموضوع پادکست را وارد کنید (یا 'exit' برای خروج): ")
-        if topic.lower() == 'exit':
-            break
+    Returns:
+        متن تولید شده برای پادکست.
+    """
+    # کلید API هوش مصنوعی شما
+    api_key = "sk-4fb613f56acfccf731e801b904cd89f5"  # لطفاً این کلید را در محیط امن نگهداری کنید و به این شکل در کد قرار ندهید در محیط واقعی
+    api_url = "https://talkbot.ir/api/v1/chat/completions"
 
-        print(f"\nدر حال تولید اسکریپت پادکست برای '{topic}' با هوش مصنوعی...")
-        podcast_script = get_text_from_ai(topic)
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
 
-        if podcast_script:
-            print("\n------------------- اسکریپت پادکست -------------------")
-            print(podcast_script)
-            print("------------------------------------------------------")
+    data = {
+        "model": "deepseek-v3-0324",
+        "messages": [
+            {"role": "system", "content": "شما یک دستیار برای تولید متن پادکست هستید."},
+            {"role": "user", "content": prompt}
+        ]
+    }
 
-            # تقسیم متن به دو بخش برای صدای اول و صدای دوم
-            # این یک تقسیم‌بندی ساده است. ممکن است نیاز به منطق دقیق‌تری داشته باشید.
-            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 = "" # یا بخشی از جمله اول
+    try:
+        response = requests.post(api_url, headers=headers, json=data)
+        if response.status_code == 200:
+            result = response.json()
+            # نیاز به بررسی ساختار پاسخ API هوش مصنوعی شما
+            if result and 'choices' in result and result['choices']:
+                 return result['choices'][0]['message']['content'].strip() if 'message' in result['choices'][0] and 'content' in result['choices'][0]['message'] else "متن پادکست تولید نشد."
             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 خالی باشد، لینک معناداری تولید نمی‌شود.
+                 print(f"پاسخ هوش مصنوعی نامعتبر است: {result}")
+                 return "خطا در تولید متن پادکست: پاسخ نامعتبر هوش مصنوعی"
+        else:
+            print(f"خطا در درخواست به API هوش مصنوعی: {response.status_code}, پاسخ: {response.text}")
+            return f"خطا در تولید متن پادکست: {response.status_code}"
+    except requests.exceptions.RequestException as e:
+        print(f"خطا در ارسال درخواست به API هوش مصنوعی: {e}")
+        return f"خطا در ارسال درخواست به API هوش مصنوعی: {e}"
 
-            if audio_link_speaker1:
-                print(f"لینک صدای اول (Speaker 1): {audio_link_speaker1}")
+# مثال استفاده از توابع (بدون واسط کاربری)
+if __name__ == "__main__":
+    # مرحله 1: تولید متن پادکست
+    podcast_prompt = "متنی کوتاه و جالب در مورد معرفی یک کتاب جدید بنویسید."
+    podcast_text = generate_podcast_script(podcast_prompt)
+    print("متن پادکست تولید شده:")
+    print(podcast_text)
+
+    if podcast_text and podcast_text != "خطا در تولید متن پادکست: پاسخ نامعتبر هوش مصنوعی" and not podcast_text.startswith("خطا"):
+        # برای سادگی، فرض می کنیم متن تولید شده بین دو گوینده تقسیم می شود.
+        # در واقعیت، نیاز به منطق پیچیده تری برای تقسیم متن برای دو گوینده است.
+        parts = podcast_text.split('.') # تقسیم ساده بر اساس نقطه به عنوان مثال
+        half_length = len(parts) // 2
+        text_part1 = '.'.join(parts[:half_length]) + '.' if half_length > 0 else ""
+        text_part2 = '.'.join(parts[half_length:]) if half_length < len(parts) else ""
+
+
+        # مرحله 2: دریافت صوت برای صدای اول
+        if text_part1:
+            print("\nدریافت صوت برای صدای اول...")
+            voice1_link = get_tts_link(text_part1, 1)
+            if voice1_link:
+                print(f"لینک صوت صدای اول: {voice1_link}")
             else:
-                print("خطا در تولید لینک صدای اول.")
-            
-            if audio_link_speaker2 and speaker2_text: # فقط اگر متن برای گوینده دوم وجود داشته باشد
-                print(f"لینک صدای دوم (Speaker 2): {audio_link_speaker2}")
-            elif speaker2_text: # اگر speaker2_text داشتیم و لینکی تولید نشد
-                 print("خطا در تولید لینک صدای دوم.")
+                 print("موفق به دریافت صوت برای صدای اول نشد.")
+
+        # مرحله 3: دریافت صوت برای صدای دوم
+        if text_part2:
+            print("\nدریافت صوت برای صدای دوم...")
+            voice2_link = get_tts_link(text_part2, 2)
+            if voice2_link:
+                print(f"لینک صوت صدای دوم: {voice2_link}")
             else:
-                print("متن کافی برای تولید صدای دوم وجود نداشت.")
+                 print("موفق به دریافت صوت برای صدای دوم نشد.")
 
-            print("\n------------------------------------------------------")
-        else:
-            print("متن پادکست از هوش مصنوعی دریافت نشد.")
-            print("لطفاً اتصال به اینترنت و کلید API را بررسی کنید.")
-
-    print("\nخروج از برنامه. موفق باشید!")
+    else:
+        print("\nبه دلیل خطای تولید متن پادکست، فرآیند تولید صوت انجام نشد.")
 
-if __name__ == "__main__":
-    main()