Update main.py

main.py

@@ -18,13 +18,13 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو نهایی) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + دانلود فایل ۱۰۰٪ تضمینی) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
 
 
-# --- ساختار کاملاً خالص کیبورد پایین صفحه (Chat Keypad) ---
+# --- ساختار کیبورد پایین صفحه (Chat Keypad) ---
 MAIN_KEYPAD_DICT = {
     "rows": [
         {
@@ -62,7 +62,7 @@ MAIN_KEYPAD_DICT = {
 }
 
 
-# --- تابع ارسال منحصراً برای کیبورد پایین صفحه (هک مستقیم API) ---
+# --- تابع ارسال منحصراً برای کیبورد پایین صفحه ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     if not use_keyboard:
         msg = await client.send_message(chat_id, text)
@@ -88,9 +88,10 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     return msg
 
 
-# --- تابع هوشمند دانلود فایل ---
+# --- 🚨 تابع هوشمند دانلود فایل (آپدیت و رفع باگ مسیر ذخیره) 🚨 ---
 async def helper_download_file(client, msg_obj):
     errors = []
+    
     file_id = None
     file_obj = None
     for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
@@ -101,17 +102,22 @@ async def helper_download_file(client, msg_obj):
             elif isinstance(val, dict) and 'file_id' in val: file_id = val['file_id']
             if file_id: break
 
+    # ساخت یک نام رندوم تا کتابخانه rubpy ارور NoneType برای نام فایل ندهد
     temp_name = f"temp_dl_{random.randint(1000, 999999)}.tmp"
 
     if hasattr(client, "download_file"):
+        
+        # تلاش ۱: ارسال مستقیم آبجکت پیام و نام فایل
         try:
             await client.download_file(msg_obj, temp_name)
             if os.path.exists(temp_name):
                 with open(temp_name, 'rb') as f: data = f.read()
                 os.remove(temp_name)
                 return data
-        except Exception as e: errors.append(f"تلاش ۱: {e}")
+        except Exception as e:
+            errors.append(f"تلاش ۱ (msg_obj): {e}")
 
+        # تلاش ۲: ارسال آبجکت فایل اختصاصی
         if file_obj:
             try:
                 await client.download_file(file_obj, temp_name)
@@ -119,8 +125,10 @@ async def helper_download_file(client, msg_obj):
                     with open(temp_name, 'rb') as f: data = f.read()
                     os.remove(temp_name)
                     return data
-            except Exception as e: errors.append(f"تلاش ۲: {e}")
+            except Exception as e:
+                errors.append(f"تلاش ۲ (file_obj): {e}")
 
+        # تلاش ۳: ارسال آیدی با آرگومان file_name
         if file_id:
             try:
                 await client.download_file(file_id, file_name=temp_name)
@@ -128,12 +136,25 @@ async def helper_download_file(client, msg_obj):
                     with open(temp_name, 'rb') as f: data = f.read()
                     os.remove(temp_name)
                     return data
-            except Exception as e: errors.append(f"تلاش ۳: {e}")
+            except Exception as e:
+                errors.append(f"تلاش ۳ (file_id, file_name): {e}")
 
-    raise Exception("دانلود ناموفق:\n" + "\n".join(errors))
+        # تلاش ۴: ارسال آیدی با آرگومان save_as
+        if file_id:
+            try:
+                await client.download_file(file_id, save_as=temp_name)
+                if os.path.exists(temp_name):
+                    with open(temp_name, 'rb') as f: data = f.read()
+                    os.remove(temp_name)
+                    return data
+            except Exception as e:
+                errors.append(f"تلاش ۴ (file_id, save_as): {e}")
+
+    # در صورت شکست تمام حالت‌ها
+    raise Exception("دانلود ناموفق. لاگ:\n" + "\n".join(errors))
 
 
-# --- لیست سرورهای تولید صدا ---
+# --- لیست تمام سرورهای تولید صدای شما ---
 WORKER_URLS =[
     "https://hamed744-ttspro.hf.space/generate",
     "https://hamed744-ttspro2.hf.space/generate",
@@ -147,16 +168,16 @@ WORKER_URLS =[
 ]
 
 SPEAKERS = {
-    "1": ("شهاب", "Charon"), "2": ("آوا", "Zephyr"), "3": ("نوید", "Achird"), 
-    "4": ("آرمان", "Zubenelgenubi"), "5": ("مهسا", "Vindemiatrix"), "6": ("دانا", "Rasalgethi"),
-    "7": ("سامان", "Sadachbia"), "8": ("آرش", "Sadaltager"), "9": ("شبنم", "Sulafat"),
-    "10": ("سحر", "Laomedeia"), "11": ("مریم", "Achernar"), "12": ("بهرام", "Alnilam"),
-    "13": ("نیکان", "Schedar"), "14": ("فرناز", "Gacrux"), "15": ("سارا", "Pulcherrima"),
-    "16": ("مانی", "Umbriel"), "17": ("آرتین", "Algieba"), "18": ("دلنواز", "Despina"),
-    "19": ("روژان", "Erinome"), "20": ("امید", "Algenib"), "21": ("بردیا", "Orus"),
-    "22": ("ترانه", "Aoede"), "23": ("نیکو", "Callirrhoe"), "24": ("هستی", "Autonoe"),
-    "25": ("کامیار", "Enceladus"), "26": ("کیانوش", "Iapetus"), "27": ("پویا", "Puck"),
-    "28": ("مهتاب", "Kore"), "29": ("سام", "Fenrir"), "30": ("لیدا", "Leda")
+    "1": ("شهاب (مرد)", "Charon"), "2": ("آوا (زن)", "Zephyr"), "3": ("نوید (مرد)", "Achird"), 
+    "4": ("آرمان (مرد)", "Zubenelgenubi"), "5": ("مهسا (زن)", "Vindemiatrix"), "6": ("دانا (مرد)", "Rasalgethi"),
+    "7": ("سامان (مرد)", "Sadachbia"), "8": ("آرش (مرد)", "Sadaltager"), "9": ("شبنم (زن)", "Sulafat"),
+    "10": ("سحر (زن)", "Laomedeia"), "11": ("مریم (زن)", "Achernar"), "12": ("بهرام (مرد)", "Alnilam"),
+    "13": ("نیکان (مرد)", "Schedar"), "14": ("فرناز (زن)", "Gacrux"), "15": ("سارا (زن)", "Pulcherrima"),
+    "16": ("مانی (مرد)", "Umbriel"), "17": ("آرتین (مرد)", "Algieba"), "18": ("دلنواز (زن)", "Despina"),
+    "19": ("روژان (زن)", "Erinome"), "20": ("امید (مرد)", "Algenib"), "21": ("بردیا (مرد)", "Orus"),
+    "22": ("ترانه (زن)", "Aoede"), "23": ("نیکو (زن)", "Callirrhoe"), "24": ("هستی (زن)", "Autonoe"),
+    "25": ("کامیار (مرد)", "Enceladus"), "26": ("کیانوش (مرد)", "Iapetus"), "27": ("پویا (مرد)", "Puck"),
+    "28": ("مهتاب (زن)", "Kore"), "29": ("سام (مرد)", "Fenrir"), "30": ("لیدا (زن)", "Leda")
 }
 
 # --- تنظیمات کلیدها ---
@@ -172,198 +193,312 @@ user_states = {}
 # --- ۱. پردازش چت متنی ---
 async def process_gemini(client, chat_id, prompt):
     if not GEMINI_KEYS:
-        await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
+        await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
         return
 
     proc_msg = await send_with_keyboard(client, chat_id, "🧠 در حال پردازش...", False)
+    
     history = user_states[chat_id].get("history", [])
-    history.append({"role": "user", "parts": [{"text": prompt}]})
 
-    if len(history) > 10: history = history[-10:]
-    if history[0]["role"] == "model": history = history[1:]
+    if history and history[-1]["role"] == "user":
+        history[-1]["parts"][0]["text"] += "\n" + prompt
+    else:
+        history.append({"role": "user", "parts": [{"text": prompt}]})
+
+    if len(history) > 10:
+        history = history[-10:]
+        if history[0]["role"] == "model":
+            history = history[1:]
 
+    keys_to_try = GEMINI_KEYS.copy()
+    random.shuffle(keys_to_try)
     final_answer = None
+    
     async with aiohttp.ClientSession() as session:
-        for key in random.sample(GEMINI_KEYS, len(GEMINI_KEYS)):
+        for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
+            payload = {"contents": history, "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
             try:
-                async with session.post(url, json={"contents": history, "generationConfig": {"temperature": 0.7}}, timeout=45) as response:
+                async with session.post(url, json=payload, timeout=45) as response:
                     if response.status == 200:
                         data = await response.json()
-                        final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
-                        break
-            except Exception: continue
+                        try:
+                            final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
+                            break
+                        except (KeyError, IndexError):
+                            continue 
+            except Exception:
+                continue
     
-    if proc_msg:
-        try:
-            msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            await client.delete_messages(chat_id, [msg_id])
-        except Exception: pass
+    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 Exception: pass
 
     if not final_answer:
-        await send_with_keyboard(client, chat_id, "❌ خطایی رخ داد. دوباره تلاش کنید.", False)
+        if history and history[-1]["role"] == "user":
+            history.pop()
+        await send_with_keyboard(client, chat_id, "❌ متأسفانه پاسخی دریافت نشد. لطفاً دوباره تلاش کنید.", False)
         return
 
-    user_states[chat_id]["history"].append({"role": "model", "parts": [{"text": final_answer}]})
-    await send_with_keyboard(client, chat_id, final_answer, False)
+    history.append({"role": "model", "parts": [{"text": final_answer}]})
+    user_states[chat_id]["history"] = history
+
+    try:
+        max_len = 1000 
+        chunks = []
+        temp_text = final_answer
+        
+        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
+            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⏳ *(ادامه در پیام بعدی)...* 👇"
+            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 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)
+        await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
         return
 
-    proc_msg = await send_with_keyboard(client, chat_id, "🎨 در حال طراحی عکس...", False)
+    proc_msg = await send_with_keyboard(client, chat_id, "🎨 در حال طراحی عکس...\n(ممکن است چند ثانیه زمان ببرد)", False)
+    
+    keys_to_try = HF_TOKENS.copy()
+    random.shuffle(keys_to_try)
     generated_image = None
-    for token in random.sample(HF_TOKENS, len(HF_TOKENS)):
+    last_error_log = "هیچ اتصالی برقرار نشد."
+    
+    for token in keys_to_try:
         try:
             hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
             generated_image = await hf_client.text_to_image(prompt, model="Tongyi-MAI/Z-Image-Turbo")
             break 
-        except Exception: continue
+        except Exception as e:
+            last_error_log = str(e)
+            continue
             
-    if proc_msg:
-        try:
-            msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            await client.delete_messages(chat_id, [msg_id])
-        except Exception: pass
+    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 Exception: pass
 
     if not generated_image:
-        await send_with_keyboard(client, chat_id, "❌ عکس ساخته نشد.", True)
+        await send_with_keyboard(client, chat_id, f"❌ عکس ساخته نشد.\n\n⚠️ خطا:\n{last_error_log[:200]}", True)
         return
 
-    file_name = f"img_{random.randint(100,999)}.jpg"
-    generated_image.convert('RGB').save(file_name, format="JPEG", quality=100)
     try:
-        await client.send_photo(chat_id, file_name)
-        await send_with_keyboard(client, chat_id, "🎨 تصویر شما آماده شد!", True)
+        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 = "🎨 تصویر شما آماده شد!\n(مدل: Z-Image-Turbo)"
+        
+        uploaded_directly = False
+        try:
+            await client.send_photo(chat_id, file_name)
+            uploaded_directly = True
+        except Exception:
+            try:
+                await client.send_document(chat_id, file_name)
+                uploaded_directly = True
+            except Exception: pass
+
+        if uploaded_directly:
+            await send_with_keyboard(client, chat_id, caption_text, True)
+        else:
+            await send_with_keyboard(client, chat_id, "❌ تصویر ساخته شد اما روبیکا اجازه آپلود نداد!", True)
+        
+        if os.path.exists(file_name): os.remove(file_name)
+            
     except Exception:
-        await send_with_keyboard(client, chat_id, "❌ خطای آپلود در روبیکا.", True)
-    if os.path.exists(file_name): os.remove(file_name)
+        await send_with_keyboard(client, chat_id, "❌ خطایی در فرآیند ذخیره عکس رخ داد.", True)
 
 
 # --- ۳. پرداز�� ساخت صدا ---
 async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
-    proc_msg = await send_with_keyboard(client, chat_id, f"⏳ ساخت صدا با «{speaker_name}»...", False)
-    audio_bytes = None
-    async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=600)) as session:
-        for worker_url in random.sample(WORKER_URLS, 3):
-            try:
-                async with session.post(worker_url, json={"text": user_text, "speaker": speaker_id, "use_live_model": True}) as response:
-                    if response.status == 200:
-                        audio_bytes = await response.read()
-                        break
-            except Exception: continue
+    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()
+        random.shuffle(workers)
+        
+        async with aiohttp.ClientSession(headers=headers, timeout=aiohttp.ClientTimeout(total=600)) as session:
+            for worker_url in workers[:3]:
+                try:
+                    async with session.post(worker_url, json=payload) as response:
+                        if response.status == 200:
+                            content_type = response.headers.get('Content-Type', '')
+                            if 'audio' in content_type or response.content_length > 1000:
+                                audio_bytes = await response.read()
+                                break
+                            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
 
-    if proc_msg:
         try:
-            msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            await client.delete_messages(chat_id, [msg_id])
+            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 Exception: pass
 
-    if audio_bytes:
-        file_name = f"voice_{random.randint(100,999)}.wav"
-        with open(file_name, "wb") as f: f.write(audio_bytes)
-        try:
-            await client.send_voice(chat_id, file_name)
-            await send_with_keyboard(client, chat_id, "✅ صدا آماده شد.", True)
-        except Exception:
-            await send_with_keyboard(client, chat_id, "❌ خطای آپلود صدا.", True)
-        if os.path.exists(file_name): os.remove(file_name)
-    else:
-        await send_with_keyboard(client, chat_id, "❌ سرورهای صدا در دسترس نیستند.", True)
+        if audio_bytes:
+            file_name = f"audio_{random.randint(1000, 999999)}.wav"
+            with open(file_name, "wb") as f: f.write(audio_bytes)
+            
+            uploaded_directly = False
+            try:
+                await client.send_voice(chat_id, file_name)
+                uploaded_directly = True
+            except Exception:
+                try:
+                    await client.send_audio(chat_id, file_name)
+                    uploaded_directly = True
+                except Exception: pass
+            
+            if uploaded_directly:
+                await send_with_keyboard(client, chat_id, "✅ پردازش صدا انجام شد.", True)
+            else:
+                await send_with_keyboard(client, chat_id, f"❌ صدا ساخته شد اما روبیکا اجازه آپلود نداد!", 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()
 
 
-# --- ۴. پردازش تبدیل صدا به متن (STT) 🚨 اصلاح شده برای مشکل Content-Type 🚨 ---
-async def process_stt(client, chat_id, audio_data, original_name):
+# --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن (STT) ---
+async def process_stt(client, chat_id, audio_bytes):
     if not HF_TOKENS:
-        await send_with_keyboard(client, chat_id, "❌ توکن هاگینگ فیس نیست.", False)
+        await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
         return
 
-    proc_msg = await send_with_keyboard(client, chat_id, "📝 در حال استخراج متن از فایل...", False)
-    
-    # تشخیص پسوند
-    ext = ".ogg"
-    if original_name and "." in original_name:
-        ext = "." + original_name.split(".")[-1].lower()
-
-    temp_file = f"stt_{random.randint(100,999)}{ext}"
-    with open(temp_file, "wb") as f: f.write(audio_data)
+    proc_msg = await send_with_keyboard(client, chat_id, "📝 در حال گوش دادن و پیاده‌سازی متن... (لطفاً صبور باشید)", False)
     
+    keys_to_try = HF_TOKENS.copy()
+    random.shuffle(keys_to_try)
     transcribed_text = None
-    for token in random.sample(HF_TOKENS, len(HF_TOKENS)):
+    last_error_log = "ناموفق"
+    
+    for token in keys_to_try:
         try:
-            # 💡 تغییر اصلی: ارسال فایل با تعیین Content-Type دقیق برای جلوگیری از ارور None
-            mime, _ = mimetypes.guess_type(temp_file)
-            if not mime: mime = "audio/ogg"
-            
-            async with aiohttp.ClientSession() as session:
-                headers = {"Authorization": f"Bearer {token}", "Content-Type": mime}
-                # استفاده از API مستقیم به جای کلاینت برای کنترل کامل هدرها
-                api_url = "https://api-inference.huggingface.co/models/openai/whisper-large-v3-turbo"
-                with open(temp_file, "rb") as f:
-                    data = f.read()
-                async with session.post(api_url, headers=headers, data=data) as resp:
-                    if resp.status == 200:
-                        res_json = await resp.json()
-                        transcribed_text = res_json.get("text")
-                        break
-        except Exception: continue
+            hf_client = AsyncInferenceClient(provider="hf-inference", api_key=token)
+            response = await hf_client.automatic_speech_recognition(audio_bytes, model="openai/whisper-large-v3-turbo")
+            transcribed_text = response.text
+            break 
+        except Exception as e:
+            last_error_log = str(e)
+            continue
 
-    if os.path.exists(temp_file): os.remove(temp_file)
-    if proc_msg:
-        try:
-            msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            await client.delete_messages(chat_id, [msg_id])
-        except Exception: pass
+    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 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, "❌ تبدیل به متن انجام نشد (فرمت فایل یا سرور مشکل داشت).", True)
+        await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.\n\n⚠️ دلیل: {last_error_log[:150]}", 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, "❌ کلید جیمینای نیست.", False)
+        await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
         return
 
-    proc_msg = 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: mime_type = "image/jpeg"
+    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'): mime_type = "audio/ogg"
+        elif file_name.endswith('.wav'): mime_type = "audio/wav"
+        else: mime_type = "image/jpeg"
     
-    base64_data = base64.b64encode(file_bytes).decode('utf-8')
+    keys_to_try = GEMINI_KEYS.copy()
+    random.shuffle(keys_to_try)
     final_answer = None
+    
     async with aiohttp.ClientSession() as session:
-        for key in random.sample(GEMINI_KEYS, len(GEMINI_KEYS)):
+        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}}]}]}
+            payload = {
+                "contents": [{"parts": [{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}],
+                "generationConfig": {"temperature": 0.6, "maxOutputTokens": 8192}
+            }
             try:
-                async with session.post(url, json=payload, timeout=45) as resp:
-                    if resp.status == 200:
-                        data = await resp.json()
-                        final_answer = data["candidates"][0]["content"]["parts"][0]["text"]
-                        break
+                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 
             except Exception: continue
 
-    if proc_msg:
-        try:
-            msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
-            await client.delete_messages(chat_id, [msg_id])
-        except Exception: pass
+    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 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)
+        await send_with_keyboard(client, chat_id, "❌ در حال حاضر پاسخی برای این فایل دریافت نشد (شاید فرمت فایل پشتیبانی نمی‌شود).", True)
 
 
-# --- تنظیمات اصلی ربات ---
+# --- تنظیمات ربات روبیکا ---
 bot_token = os.environ.get("RUBIKA_AUTH", "").strip()
+
 if not bot_token:
-    print("توکن وارد نشده!")
+    print("خطا: توکن ربات روبیکا وارد نشده است!")
 else:
     bot = BotClient(bot_token)
 
@@ -373,16 +508,27 @@ else:
             user_text = getattr(update, "text", "") or getattr(getattr(update, "message", None), "text", "") or getattr(getattr(update, "new_message", None), "text", "")
             user_text_str = str(user_text).strip() if user_text else ""
             
+            # تشخیص فایل
             is_file = False
-            file_name = "file.ogg"
+            file_name = "unknown_file.jpg"
             msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
+            
             if msg_obj:
                 for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
                     file_attr = getattr(msg_obj, attr, None)
                     if file_attr:
                         is_file = True
-                        file_name = getattr(file_attr, 'file_name', "file.ogg")
+                        if hasattr(file_attr, 'file_name'): file_name = file_attr.file_name
                         break
+                
+                if not is_file and hasattr(msg_obj, 'to_dict'):
+                    msg_dict = msg_obj.to_dict()
+                    for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+                        if attr in msg_dict:
+                            is_file = True
+                            file_attr = msg_dict.get(attr, {})
+                            if isinstance(file_attr, dict) and 'file_name' in file_attr: file_name = file_attr['file_name']
+                            break
 
             chat_id = getattr(update, "chat_id", None) or getattr(update, "author_guid", None) or getattr(update, "object_guid", None)
             if not chat_id: return
@@ -392,65 +538,137 @@ else:
 
             if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
                 user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
-                await send_with_keyboard(client, chat_id, "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\nیکی از بخش‌ها را انتخاب کنید:", True)
+                await send_with_keyboard(client, chat_id, "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
                 return
 
-            if user_text_str == "چت با هوش مصنوعی 🤖":
-                user_states[chat_id] = {"mode": "chat", "history": []}
-                await send_with_keyboard(client, chat_id, "💬 وارد بخش چت شدید. بپرسید:", True)
+            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(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str == "ساخت تصاویر با هوش مصنوعی🎨":
-                user_states[chat_id]["mode"] = "image_waiting"
-                await send_with_keyboard(client, chat_id, "🎨 توصیف عکس را بفرستید:", True)
+            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(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str == "تبدیل متن به صدا با هوش مصنوعی🎙️":
-                user_states[chat_id]["mode"] = "tts_waiting"
-                await send_with_keyboard(client, chat_id, "🎙️ متن را بفرستید (حداکثر ۲۵۰۰ کاراکتر):", True)
+            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)
                 return
 
-            if user_text_str == "تحلیل فایل با هوش مصنوعی 📁":
-                user_states[chat_id] = {"mode": "file_waiting", "file_bytes": None}
-                await send_with_keyboard(client, chat_id, "📁 فایل را بفرستید:", True)
+            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)
                 return
 
-            if user_text_str == "تبدیل فایل صوتی به متن 📝":
-                user_states[chat_id]["mode"] = "stt_waiting"
-                await send_with_keyboard(client, chat_id, "📝 فایل صوتی یا ویدیو را بفرستید:", True)
+            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)
                 return
 
             current_mode = user_states[chat_id].get("mode")
 
-            if current_mode == "chat" and user_text_str:
-                asyncio.create_task(process_gemini(client, chat_id, user_text_str))
-            elif current_mode == "image_waiting" and user_text_str:
-                asyncio.create_task(process_image(client, chat_id, user_text_str))
-            elif current_mode == "tts_waiting" and user_text_str:
-                user_states[chat_id]["text"] = user_text_str
-                user_states[chat_id]["mode"] = "tts_speaker"
-                await send_with_keyboard(client, chat_id, "شماره گوینده (۱ تا ۳۰) را بفرستید:", False)
-            elif current_mode == "tts_speaker" and user_text_str.isdigit():
-                s_name, s_id = SPEAKERS.get(user_text_str, ("شهاب", "Charon"))
-                saved_txt = user_states[chat_id]["text"]
-                user_states[chat_id]["mode"] = "tts_waiting"
-                asyncio.create_task(process_tts(client, chat_id, saved_txt, s_id, s_name))
-            elif current_mode == "stt_waiting" and is_file:
-                audio_bytes = await helper_download_file(client, msg_obj)
-                asyncio.create_task(process_stt(client, chat_id, audio_bytes, file_name))
-            elif current_mode == "file_waiting" and is_file:
-                user_states[chat_id]["file_bytes"] = await helper_download_file(client, msg_obj)
-                user_states[chat_id]["file_name"] = file_name
-                user_states[chat_id]["mode"] = "file_prompt"
-                await send_with_keyboard(client, chat_id, "حالا بگویید چگونه تحلیل شود؟", False)
-            elif current_mode == "file_prompt" and user_text_str:
-                f_bytes = user_states[chat_id]["file_bytes"]
-                f_name = user_states[chat_id]["file_name"]
-                user_states[chat_id]["mode"] = "file_waiting"
-                asyncio.create_task(process_file_analysis(client, chat_id, f_bytes, f_name, user_text_str))
-
-        except Exception: traceback.print_exc()
+            if current_mode is None:
+                if is_file: pass 
+                elif user_text_str: await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا از کیبورد پایین، بخش مورد نظرتان را انتخاب کنید:", True)
+                return
+
+            # --- مسیریابی ---
+            elif current_mode == "chat":
+                if user_text_str: asyncio.create_task(process_gemini(client, chat_id, user_text_str))
+                return
+
+            elif current_mode == "image_waiting_for_text":
+                if user_text_str: asyncio.create_task(process_image(client, chat_id, user_text_str))
+                return
+
+            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)
+                        return
+                    
+                    user_states[chat_id]["text"] = user_text_str
+                    user_states[chat_id]["mode"] = "tts_waiting_for_speaker"
+                    
+                    speakers_menu = """✅ متن شما ذخیره شد.
+لطفاً **شماره** گوینده مورد نظر خود را بفرستید:
+1. شهاب | 2. آوا | 3. نوید
+4. آرمان | 5. مهسا | 6. دانا
+7. سامان | 8. آرش | 9. شبنم
+10. سحر | 11. مریم | 12. بهرام
+13. نیکان| 14. فرناز | 15. سارا
+16. مانی | 17. آرتین | 18. دلنواز
+19. روژان | 20. امید | 21. بردیا
+22. ترانه | 23. نیکو | 24. هستی
+25. کامیار| 26. کیانوش| 27. پویا
+28. مهتاب | 29. سام | 30. لیدا"""
+                    await send_with_keyboard(client, chat_id, speakers_menu, False)
+                return
+
+            elif current_mode == "tts_waiting_for_speaker":
+                if user_text_str.isdigit() and user_text_str in SPEAKERS:
+                    speaker_name, speaker_id = SPEAKERS[user_text_str]
+                    saved_text = user_states[chat_id]["text"]
+                    user_states[chat_id]["mode"] = "tts_waiting_for_text"
+                    user_states[chat_id]["text"] = ""
+                    asyncio.create_task(process_tts(client, chat_id, saved_text, speaker_id, speaker_name))
+                else:
+                    await send_with_keyboard(client, chat_id, "❌ شماره نامعتبر است! لطفاً فقط یک عدد بین 1 تا 30 بفرستید.", False)
+                return
+
+            # --- بخش STT (صدا به متن) ---
+            elif current_mode == "stt_waiting_for_audio":
+                if not is_file:
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً یک فایل (ویس، آهنگ، ویدیو و...) ارسال کنید.", False)
+                    return
+                
+                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))
+                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)
+                return
+
+            # --- بخش تحلیل فایل ---
+            elif current_mode == "file_waiting_for_file":
+                if not is_file:
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک فایل (تصویر، ویدیو، PDF و...) ارسال کنید.", False)
+                    return
+                
+                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)
+                return
+
+            elif current_mode == "file_waiting_for_prompt":
+                if user_text_str:
+                    saved_bytes = user_states[chat_id].get("file_bytes")
+                    saved_name = user_states[chat_id].get("file_name", "file.jpeg")
+                    user_states[chat_id]["mode"] = "file_waiting_for_file" 
+                    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)
+                return
+
+        except Exception:
+            traceback.print_exc()
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
-    if bot_token: bot.run()
+    
+    if bot_token:
+        print("ربات یکپارچه آلفا روشن شد...")
+        bot.run()