Update app.py

app.py

@@ -82,7 +82,6 @@ def chat():
     
     show_thoughts = data.get("show_thoughts", False)
 
-    # بخش پردازش پیام‌ها و فایل DOCX (بدون تغییر باقی می‌ماند)
     gemini_messages = []
     for msg in data.get("messages", []):
         role = "model" if msg.get("role") == "assistant" else msg.get("role")
@@ -95,6 +94,8 @@ def chat():
             if part.get("base64Data") and part.get("mimeType"):
                 mime_type = part["mimeType"]
                 
+                # *** START: MODIFIED - منطق جدید برای تفکیک فایل‌ها ***
+                # 1. فایل Word را به متن تبدیل می‌کنیم (چون Gemini مستقیماً آن را قبول نمی‌کند)
                 if mime_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
                     try:
                         decoded_data = base64.b64decode(part["base64Data"])
@@ -110,8 +111,15 @@ def chat():
                         logging.error(f"خطا در پردازش فایل DOCX: {e}")
                         processed_parts.append({"text": "[خطا: امکان پردازش فایل Word وجود نداشت.]"})
                 
+                # 2. فایل PDF را مستقیماً ارسال می‌کنیم (چون Gemini آن را تحلیل می‌کند)
+                elif mime_type == "application/pdf":
+                    logging.info("فایل PDF برای تحلیل مستقیم به Gemini ارسال می‌شود.")
+                    processed_parts.append({"inline_data": {"mime_type": part["mimeType"], "data": part["base64Data"]}})
+
+                # 3. سایر فایل‌ها (مثل تصاویر) را نیز مستقیماً ارسال می‌کنیم
                 else:
                     processed_parts.append({"inline_data": {"mime_type": part["mimeType"], "data": part["base64Data"]}})
+                # *** END: MODIFIED ***
         
         if processed_parts:
             if gemini_messages and gemini_messages[-1]["role"] == role:
@@ -145,10 +153,7 @@ def chat():
                         "include_thoughts": True
                     }
 
-                # *** START: MODIFIED - استفاده از Timeout تفکیک شده ***
-                # timeout اول برای اتصال اولیه، timeout دوم برای فاصله بین دریافت داده‌ها
                 with requests.post(api_endpoint, json=payload, stream=True, timeout=(STREAM_START_TIMEOUT, STREAM_READ_TIMEOUT)) as response:
-                # *** END: MODIFIED ***
                     if response.status_code == 429:
                         logging.warning(f"کلید شماره {key_index + 1} سهمیه آن تمام شده است. در حال تلاش با کلید بعدی...")
                         last_error = "Rate limit exceeded"
@@ -157,7 +162,6 @@ def chat():
                     response.raise_for_status()
                     logging.info(f"اتصال با کلید شماره {key_index + 1} موفقیت‌آمیز بود. در حال استریم پاسخ...")
                     
-                    # بخش استریم پاسخ (بدون تغییر)
                     for line in response.iter_lines():
                         if line:
                             decoded_line = line.decode('utf-8')
@@ -182,7 +186,6 @@ def chat():
                     logging.info(f"استریم با کلید شماره {key_index + 1} به پایان رسید.")
                     return
 
-            # *** START: MODIFIED - افزودن ReadTimeout به مدیریت خطا ***
             except (requests.exceptions.Timeout, requests.exceptions.ReadTimeout) as e:
                 if isinstance(e, requests.exceptions.ReadTimeout):
                     logging.warning(f"استریم با کلید شماره {key_index + 1} به دلیل عدم دریافت داده جدید متوقف شد (ReadTimeout). در حال تلاش با کلید بعدی...")
@@ -191,7 +194,6 @@ def chat():
                     logging.warning(f"مهلت زمانی برای اتصال با کلید شماره {key_index + 1} به پایان رسید (ConnectTimeout). در حال تلاش با کلید بعدی...")
                     last_error = f"ConnectTimeout after {STREAM_START_TIMEOUT} seconds"
                 continue
-            # *** END: MODIFIED ***
 
             except requests.exceptions.HTTPError as e:
                 if e.response.status_code == 403:
@@ -219,5 +221,3 @@ if __name__ == '__main__':
     if GEMINI_API_KEYS:
         logging.info(f"سیستم در حالت توسعه شروع به کار کرد. تعداد {len(GEMINI_API_KEYS)} کلید شناسایی شد.")
     app.run(debug=True, host='0.0.0.0', port=os.environ.get("PORT", 7860))
-
-# --- END OF FILE app.py ---