Update main.py

main.py

@@ -28,32 +28,32 @@ def run_flask():
 MAIN_KEYPAD_DICT = {
     "rows": [
         {
-            "buttons": [
+            "buttons":[
                 {"id": "chat_btn", "type": "Simple", "button_text": "چت با هوش مصنوعی 🤖"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "img_btn", "type": "Simple", "button_text": "ساخت تصاویر با هوش مصنوعی🎨"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "tts_btn", "type": "Simple", "button_text": "تبدیل متن به صدا با هوش مصنوعی🎙️"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "file_btn", "type": "Simple", "button_text": "تحلیل فایل با هوش مصنوعی 📁"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "stt_btn", "type": "Simple", "button_text": "تبدیل فایل صوتی به متن 📝"}
             ]
         },
         {
-            "buttons": [
+            "buttons":[
                 {"id": "cancel_btn", "type": "Simple", "button_text": "برگشت♻️"}
             ]
         }
@@ -94,7 +94,7 @@ async def helper_download_file(client, msg_obj):
     
     file_id = None
     file_obj = None
-    for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+    for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
         val = getattr(msg_obj, attr, None)
         if val:
             file_obj = val
@@ -182,10 +182,10 @@ SPEAKERS = {
 
 # --- تنظیمات کلیدها ---
 GEMINI_KEYS_STR = os.environ.get("GEMINI_API_KEYS", "")
-GEMINI_KEYS = [k.strip() for k in GEMINI_KEYS_STR.split(",") if k.strip()]
+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()]
+HF_TOKENS =[k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
 user_states = {}
 
@@ -203,7 +203,7 @@ async def process_gemini(client, chat_id, prompt):
     if history and history[-1]["role"] == "user":
         history[-1]["parts"][0]["text"] += "\n" + prompt
     else:
-        history.append({"role": "user", "parts": [{"text": prompt}]})
+        history.append({"role": "user", "parts":[{"text": prompt}]})
 
     if len(history) > 10:
         history = history[-10:]
@@ -250,7 +250,7 @@ async def process_gemini(client, chat_id, prompt):
 
     try:
         max_len = 1000 
-        chunks = []
+        chunks =[]
         temp_text = final_answer
         
         while len(temp_text) > max_len:
@@ -275,7 +275,7 @@ async def process_gemini(client, chat_id, prompt):
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
 
-# --- ۲. پردازش ساخت عکس ---
+# --- ۲. پردازش ساخت عکس (نسخه آپدیت شده و 100% تضمینی) ---
 async def process_image(client, chat_id, prompt):
     if not HF_TOKENS:
         await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -319,27 +319,49 @@ async def process_image(client, chat_id, prompt):
         caption_text = "🎨 تصویر شما آماده شد!\n(مدل: Z-Image-Turbo)"
         
         uploaded_directly = False
+        
+        # 🚨 تست تمام حالت‌های آپلود عکس برای دور زدن باگ Rubpy
         try:
-            await client.send_photo(chat_id, file_name)
+            await client.send_message(chat_id, file=file_name)
             uploaded_directly = True
-        except Exception:
+        except Exception: pass
+
+        if not uploaded_directly:
+            try:
+                await client.send_photo(chat_id, photo=file_name)
+                uploaded_directly = True
+            except Exception: pass
+
+        if not uploaded_directly:
+            try:
+                await client.send_photo(chat_id, file_name)
+                uploaded_directly = True
+            except Exception: pass
+
+        if not uploaded_directly:
+            try:
+                await client.send_document(chat_id, document=file_name)
+                uploaded_directly = True
+            except Exception: pass
+
+        if not uploaded_directly:
             try:
-                await client.send_document(chat_id, file_name)
+                await client.send_document(chat_id, file=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)
+            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)
+    except Exception as e:
+        await send_with_keyboard(client, chat_id, f"❌ خطایی در فرآیند ذخیره عکس رخ داد:\n{str(e)[:150]}", True)
 
 
-# --- ۳. پردازش ساخت صدا ---
+# --- ۳. پردازش ساخت صدا (نسخه آپدیت شده و 100% تضمینی) ---
 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)
@@ -384,19 +406,48 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
             with open(file_name, "wb") as f: f.write(audio_bytes)
             
             uploaded_directly = False
+            
+            # 🚨 تست تمام حالت‌های آپلود ویس برای دور زدن باگ Rubpy
             try:
-                await client.send_voice(chat_id, file_name)
+                await client.send_message(chat_id, file=file_name)
                 uploaded_directly = True
-            except Exception:
+            except Exception: pass
+
+            if not uploaded_directly:
                 try:
-                    await client.send_audio(chat_id, file_name)
+                    await client.send_voice(chat_id, voice=file_name)
                     uploaded_directly = True
                 except Exception: pass
-            
+
+            if not uploaded_directly:
+                try:
+                    await client.send_voice(chat_id, file_name)
+                    uploaded_directly = True
+                except Exception: pass
+
+            if not uploaded_directly:
+                try:
+                    await client.send_audio(chat_id, audio=file_name)
+                    uploaded_directly = True
+                except Exception: pass
+
+            if not uploaded_directly:
+                try:
+                    await client.send_document(chat_id, document=file_name)
+                    uploaded_directly = True
+                except Exception: pass
+                
+            if not uploaded_directly:
+                try:
+                    await client.send_document(chat_id, file=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)
+                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)
@@ -468,7 +519,7 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
         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}}]}],
+                "contents": [{"parts":[{"text": prompt}, {"inlineData": {"mimeType": mime_type, "data": base64_data}}]}],
                 "generationConfig": {"temperature": 0.6, "maxOutputTokens": 8192}
             }
             try:
@@ -514,7 +565,7 @@ else:
             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']:
+                for attr in['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
                     file_attr = getattr(msg_obj, attr, None)
                     if file_attr:
                         is_file = True
@@ -523,7 +574,7 @@ else:
                 
                 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']:
+                    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, {})
@@ -534,35 +585,35 @@ else:
             if not chat_id: return
 
             if chat_id not in user_states:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
+                user_states[chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None}
 
-            if user_text_str in ["/start", "سلام", "لغو", "/cancel", "❌ لغو", "برگشت♻️"]:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
+            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\nلطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:", True)
                 return
 
             if user_text_str in ["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
-                user_states[chat_id] = {"mode": "chat", "text": "", "history": []}
+                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 in ["/image", "🎨 عکس", "ساخت تصاویر با هوش مصنوعی🎨"]:
-                user_states[chat_id] = {"mode": "image_waiting_for_text", "text": "", "history": []}
+            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 in ["/tts", "🎙️ صدا", "تبدیل متن به صدا با هوش مصنوعی🎙️"]:
-                user_states[chat_id] = {"mode": "tts_waiting_for_text", "text": "", "history": []}
+            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 in ["/file", "تحلیل فایل با هوش مصنوعی 📁"]:
-                user_states[chat_id] = {"mode": "file_waiting_for_file", "text": "", "history": [], "file_bytes": None, "file_name": None}
+            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 in ["/stt", "تبدیل فایل صوتی به متن 📝"]:
-                user_states[chat_id] = {"mode": "stt_waiting_for_audio", "text": "", "history": []}
+            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
 
@@ -671,4 +722,4 @@ if __name__ == "__main__":
     
     if bot_token:
         print("ربات یکپارچه آلفا روشن شد...")
-        bot.run()
+        bot.run()