Update main.py

main.py

@@ -6,19 +6,21 @@ import traceback
 import asyncio
 import base64
 import mimetypes
+import io
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
-# ایمپورت‌های جدید برای هاگینگ فیس و کار با عکس
+# ایمپورت‌های جدید برای هوش مصنوعی و کار با فایل صوتی
 from huggingface_hub import AsyncInferenceClient
 from PIL import Image
+from pydub import AudioSegment  # ایمپورت اضافه شده برای ترکیب پادکست
 
 # --- تنظیمات وب سرور ---
 app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + دانلود فایل ۱۰۰٪ تضمینی + چت چندرسانه‌ای + پرامپت‌نویس هوشمند) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + دانلود فایل ۱۰۰٪ تضمینی + چت چندرسانه‌ای + پرامپت‌نویس هوشمند + ساخت پادکست) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -42,6 +44,11 @@ MAIN_KEYPAD_DICT = {
                 {"id": "tts_btn", "type": "Simple", "button_text": "تبدیل متن به صدا با هوش مصنوعی🎙️"}
             ]
         },
+        {
+            "buttons":[
+                {"id": "podcast_btn", "type": "Simple", "button_text": "ساخت پادکست با هوش مصنوعی 📻"}
+            ]
+        },
         {
             "buttons":[
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل با هوش مصنوعی 📁"}
@@ -149,7 +156,6 @@ GEMINI_KEYS =[k.strip() for k in GEMINI_KEYS_STR.split(",") if k.strip()]
 HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
-# 🔴 تنظیم کردن توکن ربات به صورت سراسری 🔴
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
 
 
@@ -252,15 +258,11 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         return
 
     proc_msg = await send_with_keyboard(client, chat_id, "🧠 در حال پردازش...", False)
-    
     history = user_states[chat_id].get("history",[])
-
     new_parts =[]
     
-    if prompt:
-        new_parts.append({"text": prompt})
-    elif file_bytes:
-        new_parts.append({"text": "لطفاً این فایل را به دقت بررسی کن."})
+    if prompt: new_parts.append({"text": prompt})
+    elif file_bytes: new_parts.append({"text": "لطفاً این فایل را به دقت بررسی کن."})
         
     if file_bytes and file_name:
         base64_data = base64.b64encode(file_bytes).decode('utf-8')
@@ -275,22 +277,14 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
             elif file_name.endswith('.wav'): mime_type = "audio/wav"
             else: mime_type = "image/jpeg"
             
-        new_parts.append({
-            "inlineData": {
-                "mimeType": mime_type,
-                "data": base64_data
-            }
-        })
+        new_parts.append({"inlineData": {"mimeType": mime_type, "data": base64_data}})
 
-    if history and history[-1]["role"] == "user":
-        history[-1]["parts"].extend(new_parts)
-    else:
-        history.append({"role": "user", "parts": new_parts})
+    if history and history[-1]["role"] == "user": history[-1]["parts"].extend(new_parts)
+    else: history.append({"role": "user", "parts": new_parts})
 
     if len(history) > 40:
         history = history[-40:]
-        if history[0]["role"] == "model":
-            history = history[1:]
+        if history[0]["role"] == "model": history = history[1:]
 
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
@@ -307,23 +301,18 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
                         try:
                             final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
                             break
-                        except (KeyError, IndexError):
-                            continue 
-            except Exception:
-                continue
+                        except (KeyError, IndexError): continue 
+            except Exception: continue
     
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
-            if isinstance(proc_msg, dict):
-                msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            if msg_id:
-                await client.delete_messages(chat_id,[msg_id])
+            if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+            if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
     if not final_answer:
-        if history and history[-1]["role"] == "user":
-            history.pop()
+        if history and history[-1]["role"] == "user": history.pop()
         await send_with_keyboard(client, chat_id, "❌ متأسفانه پاسخی دریافت نشد. (شاید سایز فایل بیش از حد مجاز بوده است)", False)
         return
 
@@ -337,77 +326,63 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         
         while len(temp_text) > max_len:
             split_idx = temp_text.rfind('\n', 0, max_len)
-            if split_idx == -1:
-                split_idx = temp_text.rfind(' ', 0, max_len)
-                if split_idx == -1: split_idx = max_len
+            if split_idx == -1: split_idx = temp_text.rfind(' ', 0, max_len)
+            if split_idx == -1: split_idx = max_len
             chunks.append(temp_text[:split_idx])
             temp_text = temp_text[split_idx:].strip()
         if temp_text: chunks.append(temp_text)
         
         for idx, chunk in enumerate(chunks):
-            if idx != len(chunks) - 1:
-                chunk += "\n\n⏳ *(ادامه در پیام بعدی)...* 👇"
+            if idx != len(chunks) - 1: chunk += "\n\n⏳ *(ادامه در پیام بعدی)...* 👇"
             try:
                 await send_with_keyboard(client, chat_id, chunk, False)
                 await asyncio.sleep(2.5) 
-            except Exception:
-                await asyncio.sleep(2.5)
+            except Exception: await asyncio.sleep(2.5)
                 
     except Exception:
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
 
-# --- ۲. پردازش ساخت عکس (ارتقا یافته با پرامپت‌نویس هوشمند) ---
+# --- ۲. پردازش ساخت عکس ---
 async def process_image(client, chat_id, prompt):
     if not HF_TOKENS:
         await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
         return
 
     proc_msg = await send_with_keyboard(client, chat_id, "✨ در حال ترجمه و بهینه‌سازی پرامپت شما توسط جیمینای...\n(تبدیل به پرامپت حرفه‌ای)", False)
-    
-    # مرحله اول: ارسال پرامپت به جیمینای برای ترجمه و بهینه‌سازی
-    enhanced_prompt = prompt  # مقدار پیش‌فرض اگر جیمینای خطا داد
+    enhanced_prompt = prompt 
     if GEMINI_KEYS:
         keys_to_try_gemini = GEMINI_KEYS.copy()
         random.shuffle(keys_to_try_gemini)
-        
-        # پرامپت سیستم برای جیمینای
         gemini_sys_prompt = (
             "You are an expert AI image generation prompt engineer. "
             "Translate the following user input to English, and enhance it with high-quality, "
-            "highly detailed, 4k resolution, cinematic lighting, and visually striking descriptive keywords "
-            "suitable for Midjourney or Stable Diffusion. "
-            "Return ONLY the final English prompt string, with absolutely no extra text or explanations. "
-            f"User input: {prompt}"
+            "highly detailed, 4k resolution, cinematic lighting, and visually striking descriptive keywords. "
+            "Return ONLY the final English prompt string."
+            f"\nUser input: {prompt}"
         )
-        
         async with aiohttp.ClientSession() as session:
             for key in keys_to_try_gemini:
                 url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-                payload = {"contents": [{"parts":[{"text": gemini_sys_prompt}]}], "generationConfig": {"temperature": 0.7}}
+                payload = {"contents":[{"parts":[{"text": gemini_sys_prompt}]}], "generationConfig": {"temperature": 0.7}}
                 try:
                     async with session.post(url, json=payload, timeout=20) as response:
                         if response.status == 200:
                             data = await response.json()
                             enhanced_prompt = data["candidates"][0]["content"]["parts"][0]["text"].strip()
                             break
-                except Exception:
-                    continue
+                except Exception: continue
 
-    # پاک کردن پیام قبلی و اطلاع به کاربر درباره شروع طراحی
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
-            if isinstance(proc_msg, dict):
-                msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            if msg_id:
-                await client.delete_messages(chat_id,[msg_id])
+            if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+            if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
     
     short_preview = enhanced_prompt[:150] + "..." if len(enhanced_prompt) > 150 else enhanced_prompt
-    proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📝 پرامپت حرفه‌ای ساخته شده:\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
+    proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📝 پرامپت ساخته شده:\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
 
-    # مرحله دوم: ساخت عکس با پرامپت بهینه‌شده
     keys_to_try = HF_TOKENS.copy()
     random.shuffle(keys_to_try)
     generated_image = None
@@ -416,7 +391,6 @@ async def process_image(client, chat_id, prompt):
     for token in keys_to_try:
         try:
             hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
-            # اینجا پرامپت بهینه‌شده ارسال می‌شود
             generated_image = await hf_client.text_to_image(enhanced_prompt, model="Tongyi-MAI/Z-Image-Turbo")
             break 
         except Exception as e:
@@ -426,10 +400,8 @@ async def process_image(client, chat_id, prompt):
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
-            if isinstance(proc_msg, dict):
-                msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            if msg_id:
-                await client.delete_messages(chat_id,[msg_id])
+            if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+            if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
     if not generated_image:
@@ -440,30 +412,22 @@ async def process_image(client, chat_id, prompt):
         file_name = f"image_{random.randint(1000, 999999)}.jpg"
         rgb_im = generated_image.convert('RGB')
         rgb_im.save(file_name, format="JPEG", quality=100)
-        
         await asyncio.sleep(1)
-        caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n✨ ایده اولیه شما: {prompt}\n(مدل: Z-Image-Turbo)"
-        
+        caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n✨ ایده اولیه: {prompt}"
         upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
-        
         if upload_result is not True:
-            err_msg = str(upload_result)[:800]
-            await send_with_keyboard(client, chat_id, f"❌ تصویر در سرور ساخته شد، اما روبیکا در آپلود آن ارور داد.\n\n🔴 لاگ ارور برنامه‌نویس:\n`{err_msg}`", True)
-            
+            await send_with_keyboard(client, chat_id, f"❌ خطا در آپلود روبیکا:\n`{str(upload_result)[:800]}`", True)
         if os.path.exists(file_name): os.remove(file_name)
-            
     except Exception as e:
-        await send_with_keyboard(client, chat_id, f"❌ خطایی در فرآیند ذخیره عکس رخ داد:\n{str(e)[:150]}", True)
+        await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس:\n{str(e)[:150]}", True)
 
 
 # --- ۳. پردازش ساخت صدا ---
 async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
     try:
         proc_msg = await send_with_keyboard(client, chat_id, f"⏳ در حال ساخت صدا با «{speaker_name}»...\n(لطفاً صبور باشید)", False)
-
         payload = {"text": user_text, "speaker": speaker_id, "temperature": 1.5, "prompt": "", "use_live_model": True}
         headers = {"User-Agent": "Mozilla/5.0", "Content-Type": "application/json"}
-        
         audio_bytes = None
         last_error = "پاسخی دریافت نشد"
         workers = WORKER_URLS.copy()
@@ -478,85 +442,166 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
                             if 'audio' in content_type or response.content_length > 1000:
                                 audio_bytes = await response.read()
                                 break
-                            else: last_error = "فایل نامعتبر از سرور"
+                            else: last_error = "فایل نامعتبر"
                         else: last_error = f"ارور ({response.status})"
-                except asyncio.TimeoutError:
-                    last_error = "پایان زمان انتظار سرور."
-                    continue
-                except Exception as e:
-                    last_error = f"خطا در ارتباط: {str(e)}"
-                    continue
+                except Exception as e: last_error = f"خطا: {str(e)}"; continue
 
         try:
             if proc_msg:
                 msg_id = getattr(proc_msg, 'message_id', None)
-                if isinstance(proc_msg, dict):
-                    msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-                if msg_id:
-                    await client.delete_messages(chat_id,[msg_id])
+                if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+                if msg_id: await client.delete_messages(chat_id,[msg_id])
         except Exception: pass
 
         if audio_bytes:
-            file_name = f"audio_{random.randint(1000, 999999)}.wav"
+            file_name = f"audio_{random.randint(1000, 999999)}.mp3"
             with open(file_name, "wb") as f: f.write(audio_bytes)
-            
             await asyncio.sleep(1)
+            upload_result = await helper_upload_file(client, chat_id, file_name, "Voice", "✅ صدای شما آماده شد.")
+            if upload_result is not True: await send_with_keyboard(client, chat_id, f"❌ ارور آپلود:\n`{str(upload_result)[:800]}`", True)
+            if os.path.exists(file_name): os.remove(file_name)
+        else: await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
+    except Exception: traceback.print_exc()
+
+# --- بخش جدید: پردازش و ساخت پادکست (اتصال به سرور HF) ---
+async def process_podcast(client, chat_id, prompt):
+    proc_msg = await send_with_keyboard(client, chat_id, "📻 در حال بررسی موضوع و نگارش سناریوی پادکست توسط هوش مصنوعی...\n(لطفاً صبور باشید)", False)
+    
+    # آماده‌سازی دیتای گویندگان پیش‌فرض (همانند مرورگر)
+    available_speakers =[]
+    for num_key, (spk_name, spk_id) in SPEAKERS.items():
+        gender = "male" if "مرد" in spk_name else "female"
+        available_speakers.append({"id": spk_id, "name": spk_name.split(' (')[0], "gender": gender})
+
+    url_create = "https://ezmarynoori-podgen.hf.space/api/create-full-podcast"
+    payload_create = {
+        "prompt": prompt,
+        "available_speakers": available_speakers
+    }
+    
+    async with aiohttp.ClientSession() as session:
+        try:
+            # 1. درخواست نگارش سناریو به اسپیس ساخت پادکست
+            async with session.post(url_create, json=payload_create, timeout=60) as resp:
+                if resp.status != 202:
+                    await send_with_keyboard(client, chat_id, f"❌ خطا در سرور پادکست: وضعیت {resp.status}", True)
+                    return
+                data = await resp.json()
+                task_id = data.get("task_id")
+        except Exception as e:
+            await send_with_keyboard(client, chat_id, f"❌ خطا در اتصال به سرور پادکست:\n{str(e)}", True)
+            return
+
+        # 2. بررسی وضعیت (Polling) تا زمان آماده شدن سناریو
+        url_status = f"https://ezmarynoori-podgen.hf.space/api/podcast-status/{task_id}"
+        script_data = None
+        for _ in range(40): # تا 120 ثانیه صبر میکند
+            await asyncio.sleep(3)
+            try:
+                async with session.get(url_status) as resp:
+                    if resp.status == 200:
+                        status_data = await resp.json()
+                        if status_data.get("status") == "completed":
+                            script_data = status_data.get("data", {}).get("script",[])
+                            break
+                        elif status_data.get("status") == "failed":
+                            await send_with_keyboard(client, chat_id, "❌ متأسفانه هوش مصنوعی نتوانست برای این موضوع سناریو بنویسد.", True)
+                            return
+            except Exception: pass
             
-            upload_result = await helper_upload_file(client, chat_id, file_name, "Voice", "✅ پردازش صدای شما انجام شد.")
+        if not script_data:
+            await send_with_keyboard(client, chat_id, "❌ زمان انتظار برای نوشتن سناریو به پایان رسید.", True)
+            return
+
+        # آپدیت پیام به کاربر
+        try:
+            if proc_msg:
+                msg_id = getattr(proc_msg, 'message_id', None)
+                if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+                if msg_id: await client.delete_messages(chat_id, [msg_id])
+        except: pass
+        
+        proc_msg = await send_with_keyboard(client, chat_id, f"🎙 سناریو نوشته شد! در حال ریکورد و میکس دیالوگ‌های گویندگان (شامل {len(script_data)} نوبت صحبت)...\nاین مرحله زمان‌بر است...", False)
+
+        # 3. دریافت فایل صوتی هر دیالوگ و میکس آن‌ها
+        combined_audio = AudioSegment.empty()
+        url_generate = "https://ezmarynoori-podgen.hf.space/api/generate"
+        
+        for index, turn in enumerate(script_data):
+            payload_gen = {
+                "text": turn["dialogue"],
+                "speaker": turn["speaker_id"],
+                "temperature": 0.9
+            }
+            
+            chunk_audio_bytes = None
+            for attempt in range(3): # ۳ بار تلاش برای هر بخش
+                try:
+                    async with session.post(url_generate, json=payload_gen, timeout=120) as resp:
+                        if resp.status == 200:
+                            chunk_audio_bytes = await resp.read()
+                            break
+                except Exception:
+                    await asyncio.sleep(2)
             
-            if upload_result is not True:
-                err_msg = str(upload_result)[:800]
-                await send_with_keyboard(client, chat_id, f"❌ صدا ساخته شد، اما روبیکا در آپلود آن ارور داد.\n\n🔴 لاگ ارور برنامه‌نویس:\n`{err_msg}`", True)
+            if not chunk_audio_bytes:
+                await send_with_keyboard(client, chat_id, f"❌ خطا در تولید صدای بخش {index+1} از سرور. عملیات متوقف شد.", True)
+                return
                 
-            if os.path.exists(file_name): os.remove(file_name)
-        else:
-            await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
-    except Exception:
-        traceback.print_exc()
+            try:
+                # تبدیل بایت به AudioSegment با استفاده از pydub
+                audio_segment = AudioSegment.from_file(io.BytesIO(chunk_audio_bytes))
+                combined_audio += audio_segment
+            except Exception as e:
+                await send_with_keyboard(client, chat_id, f"❌ خطا در پردازش صدا (آیا pydub و ffmpeg نصب است؟):\n{str(e)}", True)
+                return
 
+        # 4. خروجی گرفتن و ارسال به کاربر
+        file_name = f"final_podcast_{random.randint(1000, 999999)}.mp3"
+        combined_audio.export(file_name, format="mp3")
+        
+        try:
+            if proc_msg:
+                msg_id = getattr(proc_msg, 'message_id', None)
+                if isinstance(proc_msg, dict): msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+                if msg_id: await client.delete_messages(chat_id, [msg_id])
+        except: pass
 
-# --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن (STT) با جیمینای ---
-async def process_stt(client, chat_id, audio_bytes, file_name):
-    if not GEMINI_KEYS:
-        await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
-        return
+        caption = f"🎧 پادکست شما با موفقیت میکس و ساخته شد!\n\n💡 موضوع شما: {prompt}"
+        upload_result = await helper_upload_file(client, chat_id, file_name, "Voice", caption)
+        
+        if upload_result is not True:
+            err_msg = str(upload_result)[:800]
+            await send_with_keyboard(client, chat_id, f"❌ پادکست با موفقیت در سرور ساخته شد اما روبیکا خطای آپلود داد.\n\n`{err_msg}`", True)
+        
+        if os.path.exists(file_name):
+            os.remove(file_name)
 
-    proc_msg = await send_with_keyboard(client, chat_id, "📝 در حال گوش دادن و پیاده‌سازی متن توسط جیمینای... (لطفاً صبور باشید)", False)
-    
+
+# --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن ---
+async def process_stt(client, chat_id, audio_bytes, file_name):
+    if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
+    proc_msg = await send_with_keyboard(client, chat_id, "📝 در حال گوش دادن و پیاده‌سازی متن...", False)
     base64_data = base64.b64encode(audio_bytes).decode('utf-8')
-    
     mime_type, _ = mimetypes.guess_type(file_name)
-    if not mime_type:
-        if file_name.endswith(('.ogg', '.oga')): mime_type = "audio/ogg"
-        elif file_name.endswith('.wav'): mime_type = "audio/wav"
-        elif file_name.endswith('.mp3'): mime_type = "audio/mp3"
-        elif file_name.endswith('.mp4'): mime_type = "video/mp4"
-        else: mime_type = "audio/ogg" 
+    if not mime_type: mime_type = "audio/ogg" 
     
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
     transcribed_text = None
-    
-    prompt = "لطفاً این فایل صوتی/تصویری را با دقت کامل گوش بده و صحبت‌های داخل آن را کلمه به کلمه به متن تبدیل کن. هیچ توضیح اضافه‌ای نده و فقط متن پیاده‌سازی شده را بفرست."
+    prompt = "لطفاً این فایل صوتی/تصویری را با دقت کامل گوش بده و صحبت‌های داخل آن را کلمه به کلمه به متن تبدیل کن. هیچ توضیح اضافه‌ای نده."
     
     async with aiohttp.ClientSession() as session:
         for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-            payload = {
-                "contents":[{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}],
-                "generationConfig": {"temperature": 0.2, "maxOutputTokens": 8192}
-            }
+            payload = {"contents":[{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.2}}
             try:
                 async with session.post(url, json=payload, timeout=60) as response:
                     if response.status == 200:
                         data = await response.json()
-                        try:
-                            transcribed_text = data["candidates"][0]["content"]["parts"][0]["text"]
-                            break
-                        except (KeyError, IndexError):
-                            continue 
-            except Exception:
-                continue
+                        transcribed_text = data["candidates"][0]["content"]["parts"][0]["text"]
+                        break
+            except Exception: continue
 
     try:
         if proc_msg:
@@ -565,31 +610,17 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
-    if transcribed_text:
-        await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
-    else:
-        await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.\n(احتمالاً فرمت فایل پشتیبانی نمی‌شود یا فایل شما صوتی نیست)", True)
+    if transcribed_text: await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
+    else: await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.", True)
 
 
 # --- ۵. پردازش تحلیل فایل مستقل ---
 async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
-    if not GEMINI_KEYS:
-        await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
-        return
-
-    proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال تحلیل و بررسی فایل با هوش مصنوعی...", False)
+    if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلید جیمینای تنظیم نیست.", False)
+    proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال تحلیل فایل...", False)
     base64_data = base64.b64encode(file_bytes).decode('utf-8')
-    
     mime_type, _ = mimetypes.guess_type(file_name)
-    if not mime_type:
-        if file_name.endswith(('.jpg', '.jpeg')): mime_type = "image/jpeg"
-        elif file_name.endswith('.png'): mime_type = "image/png"
-        elif file_name.endswith('.pdf'): mime_type = "application/pdf"
-        elif file_name.endswith('.mp4'): mime_type = "video/mp4"
-        elif file_name.endswith('.mp3'): mime_type = "audio/mp3"
-        elif file_name.endswith(('.ogg', '.oga')): mime_type = "audio/ogg"
-        elif file_name.endswith('.wav'): mime_type = "audio/wav"
-        else: mime_type = "image/jpeg"
+    if not mime_type: mime_type = "image/jpeg"
     
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
@@ -598,18 +629,13 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     async with aiohttp.ClientSession() as session:
         for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-            payload = {
-                "contents":[{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}],
-                "generationConfig": {"temperature": 0.6, "maxOutputTokens": 8192}
-            }
+            payload = {"contents":[{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}], "generationConfig": {"temperature": 0.6}}
             try:
                 async with session.post(url, json=payload, timeout=45) as response:
                     if response.status == 200:
                         data = await response.json()
-                        try:
-                            final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
-                            break
-                        except (KeyError, IndexError): continue 
+                        final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
+                        break
             except Exception: continue
 
     try:
@@ -619,10 +645,8 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
-    if final_answer:
-        await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
-    else:
-        await send_with_keyboard(client, chat_id, "❌ در حال حاضر پاسخی برای این فایل دریافت نشد (شاید فرمت فایل پشتیبانی نمی‌شود).", True)
+    if final_answer: await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
+    else: await send_with_keyboard(client, chat_id, "❌ پاسخی دریافت نشد.", True)
 
 
 # --- تنظیمات ربات روبیکا ---
@@ -672,27 +696,33 @@ else:
 
             if user_text_str in["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
                 user_states[chat_id] = {"mode": "chat", "text": "", "history":[]}
-                await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n(حالا ربات سابقه گفتگو را به یاد می‌آورد)\n\nهر سوالی دارید بفرستید تا جواب بدم:\n💡 نکته: در همین بخش هم می‌توانید برای من عکس یا فایل ارسال کنید تا آن را بررسی کنم.\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n\nهر سوالی دارید بفرستید تا جواب بدم:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/image", "🎨 عکس", "ساخت تصاویر با هوش مصنوعی🎨"]:
                 user_states[chat_id] = {"mode": "image_waiting_for_text", "text": "", "history":[]}
-                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\n💡 نکته: کافیست تصویر مورد نظرتان را به سادگی توصیف کنید، من خودم آن را به یک پرامپت انگلیسی و حرفه‌ای تبدیل می‌کنم و می‌سازم!\n\nمتن خود را ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\nمتن خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/tts", "🎙️ صدا", "تبدیل متن به صدا با هوش مصنوعی🎙️"]:
                 user_states[chat_id] = {"mode": "tts_waiting_for_text", "text": "", "history":[]}
-                await send_with_keyboard(client, chat_id, "🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.\n\nلطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید (حداکثر 2500 کاراکتر):\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.\n\nلطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                return
+            
+            # --- دکمه جدید برای پادکست ---
+            if user_text_str in["/podcast", "📻 پادکست", "ساخت پادکست با هوش مصنوعی 📻"]:
+                user_states[chat_id] = {"mode": "podcast_waiting_for_topic", "text": "", "history":[]}
+                await send_with_keyboard(client, chat_id, "📻 شما وارد بخش **ساخت پادکست** شدید.\n\nلطفاً موضوع پادکست خود را بفرستید.\nمثال: درباره تاریخچه پیدایش قهوه با ۳ گوینده یک پادکست جذاب بساز\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/file", "تحلیل فایل با هوش مصنوعی 📁"]:
                 user_states[chat_id] = {"mode": "file_waiting_for_file", "text": "", "history":[], "file_bytes": None, "file_name": None}
-                await send_with_keyboard(client, chat_id, "📁 شما وارد بخش **تحلیل فایل اختصاصی** شدید.\n\nلطفاً هر نوع فایلی (عکس، ویدیو، PDF، داکیومنت و...) را که می‌خواهید بررسی شود ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "📁 شما وارد بخش **تحلیل فایل اختصاصی** شدید.\n\nلطفاً فایل خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/stt", "تبدیل فایل صوتی به متن 📝"]:
                 user_states[chat_id] = {"mode": "stt_waiting_for_audio", "text": "", "history":[]}
-                await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را فوروارد یا ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             current_mode = user_states[chat_id].get("mode")
@@ -705,15 +735,12 @@ else:
             # --- مسیریابی ---
             elif current_mode == "chat":
                 if is_file:
-                    await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل برای چت هوشمند...", False)
+                    await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل...", False)
                     try:
                         file_bytes = await helper_download_file(client, msg_obj)
                         asyncio.create_task(process_gemini(client, chat_id, user_text_str, file_bytes=file_bytes, file_name=file_name))
-                    except Exception as dl_err:
-                        err_trace = traceback.format_exc()
-                        await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n\n🔴 لاگ برنامه‌نویس:\n`{err_trace[-300:]}`\n{str(dl_err)}", False)
-                elif user_text_str:
-                    asyncio.create_task(process_gemini(client, chat_id, user_text_str))
+                    except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
+                elif user_text_str: asyncio.create_task(process_gemini(client, chat_id, user_text_str))
                 return
 
             elif current_mode == "image_waiting_for_text":
@@ -723,12 +750,10 @@ else:
             elif current_mode == "tts_waiting_for_text":
                 if user_text_str:
                     if len(user_text_str) > 2500:
-                        await send_with_keyboard(client, chat_id, "⚠️ متن طولانی است. لطفاً متنی کوتاه‌تر از 2500 کاراکتر بفرستید.", False)
+                        await send_with_keyboard(client, chat_id, "⚠️ لطفاً متنی کوتاه‌تر از 2500 کاراکتر بفرستید.", False)
                         return
-                    
                     user_states[chat_id]["text"] = user_text_str
                     user_states[chat_id]["mode"] = "tts_waiting_for_speaker"
-                    
                     speakers_menu = """✅ متن شما ذخیره شد.
 لطفاً **شماره** گوینده مورد نظر خود را بفرستید:
 1. شهاب | 2. آوا | 3. نوید
@@ -755,36 +780,33 @@ else:
                     await send_with_keyboard(client, chat_id, "❌ شماره نامعتبر است! لطفاً فقط یک عدد بین 1 تا 30 بفرستید.", False)
                 return
 
+            # --- بخش مسیریابی برای قابلیت جدید (پادکست) ---
+            elif current_mode == "podcast_waiting_for_topic":
+                if user_text_str:
+                    asyncio.create_task(process_podcast(client, chat_id, user_text_str))
+                else:
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً موضوع پادکست خود را به صورت متنی بفرستید.", False)
+                return
+
             elif current_mode == "stt_waiting_for_audio":
-                if not is_file:
-                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً یک فایل (ویس، آهنگ، ویدیو و...) ارسال کنید.", False)
-                    return
-                
+                if not is_file: return await send_with_keyboard(client, chat_id, "⚠️ لطفاً یک فایل ارسال کنید.", False)
                 await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل...", False)
                 try:
                     audio_bytes = await helper_download_file(client, msg_obj)
-                    user_states[chat_id]["mode"] = "stt_waiting_for_audio" 
                     asyncio.create_task(process_stt(client, chat_id, audio_bytes, file_name))
-                except Exception as dl_err:
-                    err_trace = traceback.format_exc()
-                    await send_with_keyboard(client, chat_id, f"❌ خطا در دانلود فایل!\n\n🔴 لاگ برنامه‌نویس:\n`{err_trace[-300:]}`\n{str(dl_err)}", False)
+                except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دانلود فایل!\n{str(dl_err)}", False)
                 return
 
             elif current_mode == "file_waiting_for_file":
-                if not is_file:
-                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک فایل (تصویر، ویدیو، PDF و...) ارسال کنید.", False)
-                    return
-                
+                if not is_file: return await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک فایل ارسال کنید.", False)
                 await send_with_keyboard(client, chat_id, "📥 در حال دریافت فایل...", False)
                 try:
                     file_bytes = await helper_download_file(client, msg_obj)
                     user_states[chat_id]["file_bytes"] = file_bytes
                     user_states[chat_id]["file_name"] = file_name 
                     user_states[chat_id]["mode"] = "file_waiting_for_prompt"
-                    await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً به صورت متنی بگویید **چگونه تحلیل شود؟**", False)
-                except Exception as dl_err:
-                    err_trace = traceback.format_exc()
-                    await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n\n🔴 لاگ برنامه‌نویس:\n`{err_trace[-300:]}`\n{str(dl_err)}", False)
+                    await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بگویید **چگونه تحلیل شود؟**", False)
+                except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
                 return
 
             elif current_mode == "file_waiting_for_prompt":
@@ -795,12 +817,10 @@ else:
                     user_states[chat_id]["file_bytes"] = None
                     user_states[chat_id]["file_name"] = None
                     asyncio.create_task(process_file_analysis(client, chat_id, saved_bytes, saved_name, user_text_str))
-                else:
-                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را به صورت متنی بنویسید.", False)
+                else: await send_with_keyboard(client, chat_id, "⚠️ لطفاً درخواست خود را متنی بنویسید.", False)
                 return
 
-        except Exception:
-            traceback.print_exc()
+        except Exception: traceback.print_exc()
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()