Update app.py

app.py

@@ -11,7 +11,7 @@ from flask import Flask, render_template, request, Response
 import requests
 import docx
 
-# ================== بخش تنظیمات لاگ‌نویسی (بدون تغییر) ==================
+# ================== بخش تنظیمات لاگ‌نویسی ==================
 
 class NoGrpcFilter(logging.Filter):
     def filter(self, record):
@@ -36,7 +36,7 @@ def setup_logging():
 setup_logging()
 app = Flask(__name__)
 
-# ================== بخش پیکربندی Gemini (با تغییر) ==================
+# ================== بخش پیکربندی Gemini ==================
 
 GEMINI_MODEL_NAME = "gemini-2.5-flash" 
 ALL_KEYS_STR = os.environ.get("ALL_GEMINI_API_KEYS", "")
@@ -58,6 +58,11 @@ def get_next_key_with_index():
         key_index_counter = (key_index_counter + 1) % len(GEMINI_API_KEYS)
         return key, current_index
 
+# تنظیمات تایم‌اوت برای درخواست‌ها به گوگل
+# تایم‌اوت اول برای اتصال، دومی برای خواندن داده. مقادیر افزایش یافته تا پایداری بیشتر شود.
+STREAM_CONNECT_TIMEOUT = 10
+STREAM_READ_TIMEOUT = 40 
+
 # ================== پایان بخش پیکربندی ====================
 
 @app.route('/')
@@ -67,8 +72,8 @@ def index():
 @app.route('/chat', methods=['POST'])
 def chat():
     if not GEMINI_API_KEYS:
-        # اگر هیچ کلیدی نباشد چاره‌ای جز خطا نیست، اما این حالت نادری است
-        return Response("data: [DONE]\n\n", mimetype='text/event-stream')
+        error_payload = {"type": "error", "message": "خطای سرور: هیچ کلید API پیکربندی نشده است."}
+        return Response(f"data: {json.dumps(error_payload)}\n\n", status=500, mimetype='text/event-stream')
 
     data = request.json
     system_instruction = "تو چت بات هوش مصنوعی آلفا هستی و توسط برنامه هوش مصنوعی آلفا توسعه داده شدی. کمی با کاربران باحال و دوستانه صحبت کن و از ایموجی‌ها استفاده کن. همیشه پاسخ‌هایت را به زبان فارسی و یا هر زبانی که کاربر صحبت میکنه ارائه بده."
@@ -116,14 +121,16 @@ def chat():
         return Response("data: [DONE]\n\n", mimetype='text/event-stream')
         
     def stream_response():
-        # تلاش برای تعداد زیادی بار (عملا تا وقتی یک کلید سالم پیدا شود)
-        # ضرب در 2 یعنی هر کلید را دو بار شانس می دهیم اگر شبکه قطع و وصل شد
-        max_attempts = len(GEMINI_API_KEYS) * 3 
+        # تعداد دفعات تلاش کلی: به اندازه 2 برابر تعداد کلیدها تلاش میکنیم تا مطمئن شویم
+        max_attempts = len(GEMINI_API_KEYS) * 2 
+        attempts = 0
+        last_error = None
         
-        for attempt in range(max_attempts):
+        while attempts < max_attempts:
+            attempts += 1
             try:
                 api_key, key_index = get_next_key_with_index()
-                # logging.info(f"تلاش با کلید شماره {key_index + 1}...")
+                # logging.info(f"تلاش شماره {attempts}: استفاده از کلید ایندکس {key_index}")
 
                 api_endpoint = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:streamGenerateContent?key={api_key}&alt=sse"
                 
@@ -141,17 +148,26 @@ def chat():
                         "include_thoughts": True
                     }
 
-                # *** نکته مهم: Timeout بسیار کوتاه برای اتصال، تا سریع برود کلید بعدی ***
-                # connect=4: اگر 4 ثانیه وصل نشد، ولش کن برو بعدی
-                # read=20: اگر وسط کار 20 ثانیه دیتا نیامد، ولش کن برو بعدی
-                with requests.post(api_endpoint, json=payload, stream=True, timeout=(4, 20)) as response:
+                # ارسال درخواست به گوگل
+                with requests.post(api_endpoint, json=payload, stream=True, timeout=(STREAM_CONNECT_TIMEOUT, STREAM_READ_TIMEOUT)) as response:
+                    # اگر خطای 429 (محدودیت نرخ) یا 5xx (خطای سرور گوگل) باشد
+                    if response.status_code == 429 or response.status_code >= 500:
+                        logging.warning(f"خطای {response.status_code} با کلید {key_index}. رفتن به کلید بعدی...")
+                        last_error = f"Status Code: {response.status_code}"
+                        continue # برو به کلید بعدی
+                    
+                    # اگر خطای 403 (Forbidden/Invalid Key) باشد
+                    if response.status_code == 403:
+                        logging.warning(f"کلید {key_index} نامعتبر است (403). رفتن به کلید بعدی...")
+                        last_error = "Invalid API Key (403)"
+                        continue
+
+                    response.raise_for_status()
                     
-                    # اگر ارور 429 (محدودیت) یا هر ارور دیگری داد، سریع برو بعدی
-                    if response.status_code != 200:
-                        logging.warning(f"کلید {key_index + 1} با وضعیت {response.status_code} پاسخ نداد. تلاش با کلید بعدی...")
-                        continue 
+                    # اگر موفق شدیم وصل بشیم، شروع به استریم میکنیم
+                    # logging.info(f"اتصال موفق با کلید {key_index}. شروع استریم...")
                     
-                    # اگر موفق شد، شروع به ارسال کن و از تابع خارج شو (return)
+                    data_received = False
                     for line in response.iter_lines():
                         if line:
                             decoded_line = line.decode('utf-8')
@@ -163,6 +179,8 @@ def chat():
                                     for part in parts:
                                         if "text" not in part or not part["text"]:
                                             continue
+                                        
+                                        data_received = True
                                         is_a_thought = part.get("thought") is True
                                         if show_thoughts and is_a_thought:
                                             thought_payload = {"type": "thought", "content": part["text"]}
@@ -173,24 +191,38 @@ def chat():
                                 except (json.JSONDecodeError, IndexError, KeyError):
                                     continue
                     
-                    # اگر استریم بدون خطا تمام شد، کار تمام است
-                    return
-
+                    # اگر دیتا با موفقیت دریافت شد و حلقه تمام شد، از تابع خارج میشویم
+                    if data_received:
+                        return
+                    else:
+                        # اگر وصل شدیم اما دیتایی نیامد (پاسخ خالی)، شاید مشکلی بوده، امتحان مجدد با کلید بعدی
+                        last_error = "Empty response"
+                        continue
+
+            except (requests.exceptions.Timeout, requests.exceptions.ConnectionError, requests.exceptions.ChunkedEncodingError) as e:
+                logging.warning(f"خطای شبکه/تایم‌اوت با کلید {key_index}: {str(e)}. تلاش مجدد با کلید بعدی...")
+                last_error = str(e)
+                continue # برو به کلید بعدی
+            
             except Exception as e:
-                # هر نوع خطایی (تایم اوت، شبکه و ...) رخ داد، فقط لاگ کن و برو کلید بعدی
-                # هیچ خطایی به کاربر نشان نده
-                logging.warning(f"خطا در کلید {key_index + 1}: {e} - رفتن به کلید بعدی")
-                continue 
+                logging.error(f"خطای غیرمنتظره با کلید {key_index}: {e}")
+                last_error = str(e)
+                continue # برای اطمینان، حتی با خطای عجیب هم سعی میکنیم با کلید بعدی ادامه بدیم
         
-        # اگر همه کلیدها تست شدند و نشد (خیلی بعید)، یک پیام خالی بفرست که ارور نده
-        logging.critical("هیچ کلیدی کار نکرد.")
-        # به جای ارور، یک پیام ساده میفرستیم که برنامه کرش نکند
-        end_payload = {"choices": [{"delta": {"content": "..."}}]} 
-        yield f"data: {json.dumps(end_payload)}\n\n"
+        # اگر همه تلاش‌ها شکست خورد
+        logging.critical("تمام کلیدها امتحان شدند و پاسخگو نبودند.")
+        # اینجا دیگر چاره‌ای نیست جز نمایش خطا، اما چون کاربر گفته خطا نبینه،
+        # می‌توانیم یک پیام مودبانه بفرستیم یا هیچ چیزی نفرستیم (که البته باعث میشه کاربر فکر کنه هنوز داره فکر میکنه)
+        # بهترین کار ارسال یک پیام خطای نرم است.
+        error_message = "سیستم در حال حاضر شلوغ است. لطفاً چند لحظه دیگر دوباره تلاش کنید."
+        error_payload = {"choices": [{"delta": {"content": error_message}}]} # به جای ارور، به عنوان متن چت میفرستیم
+        yield f"data: {json.dumps(error_payload)}\n\n"
 
     return Response(stream_response(), mimetype='text/event-stream')
 
 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 ---