Update main.py

main.py

@@ -5,6 +5,7 @@ import aiohttp
 import traceback
 import asyncio
 import base64
+import mimetypes
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
@@ -17,7 +18,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو با تحلیل فایل و STT) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو با پشتیبانی کامل تمام فایل‌ها) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -64,11 +65,9 @@ MAIN_KEYPAD_DICT = {
 # --- تابع ارسال منحصراً برای کیبورد پایین صفحه ---
 async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
     if not use_keyboard:
-        # ارسال پیام متنی خالص بدون دستکاری کیبورد
         msg = await client.send_message(chat_id, text)
         return msg
 
-    # ارسال تهاجمی: دور زدن محدودیت‌های rubpy
     try:
         payload = {
             "chat_id": chat_id,
@@ -85,7 +84,6 @@ async def send_with_keyboard(client, chat_id, text, use_keyboard=True):
         except Exception as standard_err:
             pass
 
-    # اگر تمامی روش‌ها شکست خورد، فقط پیام ارسال شود
     msg = await client.send_message(chat_id, text)
     return msg
 
@@ -126,16 +124,14 @@ HF_TOKENS = [k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 user_states = {}
 
 
-# --- ۱. پردازش چت متنی در پس‌زمینه (با حذف پیام پردازش) ---
+# --- ۱. پردازش چت متنی در پس‌زمینه ---
 async def process_gemini(client, chat_id, 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)
     
-    # استخراج تاریخچه مکالمات کاربر
     history = user_states[chat_id].get("history", [])
 
     if history and history[-1]["role"] == "user":
@@ -170,7 +166,6 @@ async def process_gemini(client, chat_id, prompt):
             except Exception as e:
                 continue
     
-    # 🔴 پاک کردن پیام موقت "در حال پردازش..."
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -178,8 +173,7 @@ async def process_gemini(client, chat_id, prompt):
                 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 as e:
-        print(f"Failed to delete processing msg: {e}")
+    except Exception: pass
 
     if not final_answer:
         if history and history[-1]["role"] == "user":
@@ -215,14 +209,14 @@ async def process_gemini(client, chat_id, prompt):
             try:
                 await send_with_keyboard(client, chat_id, chunk, False)
                 await asyncio.sleep(2.5) 
-            except Exception as send_err:
+            except Exception:
                 await asyncio.sleep(2.5)
                 
-    except Exception as e:
+    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)
@@ -245,7 +239,6 @@ async def process_image(client, chat_id, prompt):
             last_error_log = str(e)
             continue
             
-    # پاک کردن پیام موقت
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -285,11 +278,11 @@ async def process_image(client, chat_id, prompt):
         if os.path.exists(file_name):
             os.remove(file_name)
             
-    except Exception as e:
+    except Exception:
         await send_with_keyboard(client, chat_id, "❌ خطایی در فرآیند ذخیره عکس رخ داد.", True)
 
 
-# --- ۳. پردازش ساخت صدا در پس‌زمینه (TTS) ---
+# --- ۳. پردازش ساخت صدا ---
 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)
@@ -324,7 +317,6 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
                     last_error = f"خطا در ارتباط: {str(e)}"
                     continue
 
-        # پاک کردن پیام موقت
         try:
             if proc_msg:
                 msg_id = getattr(proc_msg, 'message_id', None)
@@ -361,11 +353,11 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
         else:
             await send_with_keyboard(client, chat_id, f"❌ سرورها درگیر هستند.\nدلیل: {last_error}", True)
 
-    except Exception as e:
+    except Exception:
         traceback.print_exc()
 
 
-# --- ۴. پردازش تبدیل فایل صوتی به متن (STT) ---
+# --- ۴. پردازش تبدیل فایل صوتی/تصویری به متن (STT) ---
 async def process_stt(client, chat_id, audio_bytes):
     if not HF_TOKENS:
         await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
@@ -381,7 +373,6 @@ async def process_stt(client, chat_id, audio_bytes):
     
     for token in keys_to_try:
         try:
-            # استفاده از مدل قدرتمند ویسپر برای تبدیل صدا به متن (پشتیبانی عالی از فارسی)
             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
@@ -390,7 +381,6 @@ async def process_stt(client, chat_id, audio_bytes):
             last_error_log = str(e)
             continue
 
-    # پاک کردن پیام موقت
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -403,20 +393,31 @@ async def process_stt(client, chat_id, audio_bytes):
     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\n⚠️ دلیل: {last_error_log[:150]}", 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, prompt):
+# --- ۵. پردازش تحلیل هر نوع فایل با جیمینای ---
+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)
+    proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال تحلیل و بررسی فایل با هوش مصنوعی...", False)
     
-    # تبدیل فایل به base64 برای ارسال به جیمینای
     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'): mime_type = "audio/ogg"
+        elif file_name.endswith('.wav'): mime_type = "audio/wav"
+        else: mime_type = "image/jpeg"
+    
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
     final_answer = None
@@ -428,7 +429,7 @@ async def process_file_analysis(client, chat_id, file_bytes, prompt):
                 "contents": [{
                     "parts": [
                         {"text": prompt},
-                        {"inlineData": {"mimeType": "image/jpeg", "data": base64_data}}
+                        {"inlineData": {"mimeType": mime_type, "data": base64_data}}
                     ]
                 }],
                 "generationConfig": {"temperature": 0.6, "maxOutputTokens": 8192}
@@ -444,10 +445,9 @@ async def process_file_analysis(client, chat_id, file_bytes, prompt):
                             continue 
                     else:
                         continue 
-            except Exception as e:
+            except Exception:
                 continue
 
-    # پاک کردن پیام موقت
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -460,7 +460,7 @@ async def process_file_analysis(client, chat_id, file_bytes, prompt):
     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)
 
 
 # --- تنظیمات ربات روبیکا ---
@@ -474,27 +474,45 @@ else:
     @bot.on_update(filters.private)
     async def main_handler(client, update):
         try:
-            # استخراج اطلاعات پیام
             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 = "unknown_file.jpg"
             msg_obj = getattr(update, "message", None) or getattr(update, "new_message", None)
+            
             if msg_obj:
-                if getattr(msg_obj, "file_inline", None) or getattr(msg_obj, "photo", None) or getattr(msg_obj, "voice", None) or getattr(msg_obj, "audio", None) or getattr(msg_obj, "document", None):
-                    is_file = True
+                # بررسی متغیرهای مختلف (کلاس)
+                for attr in ['file', 'file_inline', 'photo', 'voice', 'audio', 'document', 'video']:
+                    file_attr = getattr(msg_obj, attr, None)
+                    if file_attr:
+                        is_file = True
+                        if hasattr(file_attr, 'file_name'):
+                            file_name = file_attr.file_name
+                        break
+                
+                # بررسی متغیرهای مختلف (دیکشنری) - برای پشتیبانی از نسخه‌های مختلف rubpy
+                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
 
             if chat_id not in user_states:
-                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": 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}
+                user_states[chat_id] = {"mode": None, "text": "", "history": [], "file_bytes": None, "file_name": None}
                 menu = (
                     "سلام! به ربات هوشمند آلفا خوش آمدید 🤖\n\n"
                     "لطفاً برای شروع، از کیبورد پایین یکی از بخش‌ها را انتخاب کنید:"
@@ -518,20 +536,20 @@ else:
                 return
 
             if user_text_str in ["/file", "تحلیل فایل با هوش مصنوعی 📁"]:
-                user_states[chat_id] = {"mode": "file_waiting_for_file", "text": "", "history": [], "file_bytes": None}
-                await send_with_keyboard(client, chat_id, "📁 شما وارد بخش **تحلیل فایل** شدید.\n\nلطفاً فایل (ترجیحاً عکس) خود را ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                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": []}
-                await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل صوتی (آهنگ یا ویس) خود را فوروارد یا ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را فوروارد یا ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             current_mode = user_states[chat_id].get("mode")
 
             if current_mode is None:
                 if is_file:
-                    pass # اگر حالت مشخص نیست فایل را نادیده بگیر
+                    pass 
                 elif user_text_str:
                     await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا از کیبورد پایین، بخش مورد نظرتان را انتخاب کنید:", True)
                 return
@@ -586,45 +604,48 @@ else:
             # --- بخش جدید: STT (تبدیل صدا به متن) ---
             elif current_mode == "stt_waiting_for_audio":
                 if not is_file:
-                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً فقط یک فایل صوتی (ویس یا موزیک) ارسال کنید.", False)
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً یک فایل (ویس، آهنگ، ویدیو و...) ارسال کنید.", False)
                     return
                 
-                await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل صوتی...", False)
+                await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل...", False)
                 try:
                     audio_bytes = await client.download(msg_obj)
-                    user_states[chat_id]["mode"] = "stt_waiting_for_audio" # آماده برای فایل بعدی
+                    user_states[chat_id]["mode"] = "stt_waiting_for_audio" 
                     asyncio.create_task(process_stt(client, chat_id, audio_bytes))
-                except Exception as e:
-                    await send_with_keyboard(client, chat_id, f"❌ خطا در دانلود فایل! لطفاً دوباره تلاش کنید.", False)
+                except Exception:
+                    await send_with_keyboard(client, chat_id, "❌ خطا در دانلود فایل! لطفاً دوباره تلاش کنید.", False)
                 return
 
             # --- بخش جدید: تحلیل فایل با هوش مصنوعی ---
             elif current_mode == "file_waiting_for_file":
                 if not is_file:
-                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک عکس/فایل ارسال کنید.", False)
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک فایل (تصویر، ویدیو، PDF و...) ارسال کنید.", False)
                     return
                 
                 await send_with_keyboard(client, chat_id, "📥 در حال دریافت فایل...", False)
                 try:
                     file_bytes = await client.download(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 e:
-                    await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل! لطفاً دوباره تلاش کنید.", False)
+                    await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً به صورت متنی بگویید **چگونه تحلیل شود؟**\n(مثلاً: متن این فایل را استخراج کن، یا بگو محتوای آن چیست؟)", False)
+                except Exception:
+                    await send_with_keyboard(client, chat_id, "❌ خطا در دریافت فایل! لطفاً دوباره تلاش کنید.", False)
                 return
 
             elif current_mode == "file_waiting_for_prompt":
                 if user_text_str:
                     saved_bytes = user_states[chat_id].get("file_bytes")
-                    user_states[chat_id]["mode"] = "file_waiting_for_file" # آماده برای فایل بعدی
+                    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
-                    asyncio.create_task(process_file_analysis(client, chat_id, saved_bytes, user_text_str))
+                    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 as e:
+        except Exception:
             traceback.print_exc()
 
 if __name__ == "__main__":