Update app.py

app.py

@@ -57,13 +57,13 @@ def get_next_key_with_index():
         key_index_counter = (key_index_counter + 1) % len(GEMINI_API_KEYS)
         return key, current_index
 
-# *** START: MODIFIED - افزودن ثابت‌های جدید برای مهلت زمانی ***
-# مهلت زمانی (به ثانیه) برای شروع دریافت پاسخ از سرور
-STREAM_START_TIMEOUT = 4
-# مهلت زمانی (به ثانیه) برای دریافت هر قطعه جدید از داده در حین استریم
-# اگر در این مدت داده جدیدی نرسد، اتصال قطع و با کلید بعدی تلاش می‌شود
-STREAM_READ_TIMEOUT = 15
-# *** END: MODIFIED ***
+# *** START: OPTIMIZED TIMEOUT SETTINGS ***
+# مهلت زمانی برای برقراری اتصال اولیه با سرور گوگل (اگر اینترنت قطع باشد یا سرور دان باشد سریع سوئیچ میکند)
+STREAM_CONNECT_TIMEOUT = 10 
+# مهلت زمانی برای انتظار پردازش و دریافت پاسخ.
+# برای فایل‌ها بسیار مهم است که این عدد بالا باشد (مثلا 100 ثانیه) تا وسط پردازش فایل قطع نکند.
+STREAM_READ_TIMEOUT = 100
+# *** END: OPTIMIZED TIMEOUT SETTINGS ***
 
 # ================== پایان بخش پیکربندی ====================
 
@@ -82,7 +82,7 @@ def chat():
     
     show_thoughts = data.get("show_thoughts", False)
 
-    # بخش پردازش پیام‌ها و فایل DOCX (بدون تغییر باقی می‌ماند)
+    # بخش پردازش پیام‌ها و فایل DOCX
     gemini_messages = []
     for msg in data.get("messages", []):
         role = "model" if msg.get("role") == "assistant" else msg.get("role")
@@ -124,10 +124,14 @@ def chat():
         
     def stream_response():
         last_error = None
-        for _ in range(len(GEMINI_API_KEYS)):
+        # تلاش برای همه کلیدها تا زمانی که یکی جواب دهد
+        # ما یک کپی از لیست کلیدها میگیریم تا مطمئن شویم حلقه به اندازه تعداد کلیدها اجرا میشود
+        attempts = len(GEMINI_API_KEYS)
+        
+        for i in range(attempts):
             try:
                 api_key, key_index = get_next_key_with_index()
-                logging.info(f"تلاش برای ارسال درخواست با کلید شماره {key_index + 1}...")
+                logging.info(f"تلاش {i+1} از {attempts}: ارسال درخواست با کلید شماره {key_index + 1}...")
 
                 api_endpoint = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:streamGenerateContent?key={api_key}&alt=sse"
                 
@@ -145,19 +149,29 @@ 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"
-                        continue
+                # استفاده از تنظیمات Timeout بهینه شده
+                # connect: زمان اتصال به سرور (سریع)
+                # read: زمان انتظار برای پردازش فایل و پاسخ (طولانی)
+                with requests.post(api_endpoint, json=payload, stream=True, timeout=(STREAM_CONNECT_TIMEOUT, STREAM_READ_TIMEOUT)) as response:
+                    
+                    # بررسی خطاهای HTTP
+                    if response.status_code != 200:
+                        logging.warning(f"خطای سرور با کلید {key_index + 1}: کد {response.status_code} - متن: {response.text}")
+                        # اگر محدودیت درخواست یا خطای سرور بود، برو کلید بعدی
+                        if response.status_code in [429, 500, 502, 503, 504]:
+                            last_error = f"HTTP {response.status_code}"
+                            continue # پرش سریع به کلید بعدی
+                        elif response.status_code == 403:
+                            last_error = "Invalid API Key"
+                            continue # پرش سریع
+                        else:
+                            response.raise_for_status() # برای سایر خطاها
                     
-                    response.raise_for_status()
-                    logging.info(f"اتصال با کلید شماره {key_index + 1} موفقیت‌آمیز بود. در حال استریم پاسخ...")
+                    logging.info(f"اتصال با کلید شماره {key_index + 1} برقرار شد. در حال دریافت داده...")
                     
-                    # بخش استریم پاسخ (بدون تغییر)
+                    # اگر به اینجا رسیدیم یعنی اتصال موفق بوده و استریم شروع شده
+                    # حالا داده‌ها را به کلاینت می‌فرستیم
+                    has_sent_data = False
                     for line in response.iter_lines():
                         if line:
                             decoded_line = line.decode('utf-8')
@@ -169,6 +183,8 @@ def chat():
                                     for part in parts:
                                         if "text" not in part or not part["text"]:
                                             continue
+                                        
+                                        has_sent_data = True
                                         is_a_thought = part.get("thought") is True
                                         if show_thoughts and is_a_thought:
                                             thought_payload = {"type": "thought", "content": part["text"]}
@@ -179,37 +195,42 @@ def chat():
                                 except (json.JSONDecodeError, IndexError, KeyError):
                                     continue
                     
-                    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). در حال تلاش با کلید بعدی...")
-                    last_error = f"ReadTimeout after {STREAM_READ_TIMEOUT} seconds of inactivity"
-                else:
-                    logging.warning(f"مهلت زمانی برای اتصال با کلید شماره {key_index + 1} به پایان رسید (ConnectTimeout). در حال تلاش با کلید بعدی...")
-                    last_error = f"ConnectTimeout after {STREAM_START_TIMEOUT} seconds"
+                    # اگر حلقه تمام شد و دیتایی ارسال شد، یعنی موفقیت آمیز بوده
+                    if has_sent_data:
+                        logging.info(f"پاسخ کامل با کلید {key_index + 1} ارسال شد.")
+                        return # خروج از کل تابع و پایان کار
+                    else:
+                        # اگر هیچ دیتایی نیامد اما کد 200 بود (مثلا پاسخ خالی)، باز هم موفقیت حساب میشه
+                        return 
+
+            except requests.exceptions.ConnectTimeout:
+                logging.warning(f"Timeout در اتصال با کلید {key_index + 1}. تلاش فوری با کلید بعدی...")
+                last_error = "Connection Timeout"
+                continue # تلاش با کلید بعدی
+
+            except requests.exceptions.ReadTimeout:
+                # اگر ReadTimeout رخ داد یعنی گوگل خیلی طولش داده، اما معمولا با 100 ثانیه رخ نمیده.
+                # اگر رخ داد یعنی سرور پاسخگو نیست، پس کلید بعدی.
+                logging.warning(f"ReadTimeout (عدم پاسخگویی طولانی) با کلید {key_index + 1}. تلاش با کلید بعدی...")
+                last_error = "Read Timeout"
                 continue
-            # *** END: MODIFIED ***
-
-            except requests.exceptions.HTTPError as e:
-                if e.response.status_code == 403:
-                     logging.warning(f"کلید شماره {key_index + 1} نامعتبر است (Permission Denied). در حال تلاش با کلید بعدی...")
-                     last_error = e
-                     continue
-                else:
-                     logging.error(f"خطای HTTP پیش‌بینی نشده در حین تلاش با کلید {key_index + 1}: {e}")
-                     last_error = e
-                     break
 
+            except requests.exceptions.ConnectionError:
+                 logging.warning(f"خطای شبکه (ConnectionError) با کلید {key_index + 1}. تلاش با کلید بعدی...")
+                 last_error = "Network Error"
+                 continue
+            
             except Exception as e:
-                logging.error(f"خطای پیش‌بینی نشده در حین تلاش با کلید {key_index + 1}: {e}")
-                last_error = e
-                break 
+                logging.error(f"خطای پیش‌بی��ی نشده با کلید {key_index + 1}: {e}")
+                last_error = str(e)
+                # برای خطاهای ناشناخته هم به کلید بعدی شانس میدهیم
+                continue
+
+        # اگر حلقه تمام شد و هیچ کلیدی موفق نشد:
+        error_message = "متاسفانه سرور پاسخگو نیست. لطفا دقایقی دیگر تلاش کنید."
+        if last_error:
+            logging.critical(f"تمام تلاش‌ها ناموفق بود. آخرین خطا: {last_error}")
         
-        error_message = f"متاسفانه تمام کلیدهای API موجود نامعتبر هستند یا سهمیه آنها به پایان رسیده است. لطفا بعدا تلاش کنید.\n\nآخرین خطا: {str(last_error)}"
-        logging.critical("هیچ‌کدام از کلیدهای Gemini کار نکردند.")
         error_payload = {"type": "error", "message": error_message}
         yield f"data: {json.dumps(error_payload)}\n\n"
 
@@ -217,7 +238,7 @@ def chat():
 
 if __name__ == '__main__':
     if GEMINI_API_KEYS:
-        logging.info(f"سیستم در حالت توسعه شروع به کار کرد. تعداد {len(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 ---