Update app.py

app.py

@@ -38,6 +38,7 @@ app = Flask(__name__)
 
 # ================== بخش پیکربندی Gemini ==================
 
+# مدل 2.5 فلش بهترین مدل برای پردازش ویدیو است
 GEMINI_MODEL_NAME = "gemini-2.5-flash" 
 ALL_KEYS_STR = os.environ.get("ALL_GEMINI_API_KEYS", "")
 GEMINI_API_KEYS = [key.strip() for key in ALL_KEYS_STR.split(',') if key.strip()]
@@ -58,10 +59,9 @@ 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 
+# افزایش تایم‌اوت برای آپلود ویدیو ضروری است
+STREAM_CONNECT_TIMEOUT = 20
+STREAM_READ_TIMEOUT = 120 
 
 # ================== پایان بخش پیکربندی ====================
 
@@ -76,11 +76,11 @@ def chat():
         return Response(f"data: {json.dumps(error_payload)}\n\n", status=500, mimetype='text/event-stream')
 
     data = request.json
-    system_instruction = "تو چت بات هوش مصنوعی آلفا هستی و توسط برنامه هوش مصنوعی آلفا توسعه داده شدی. کمی با کاربران باحال و دوستانه صحبت کن و از ایموجی‌ها استفاده کن. همیشه پاسخ‌هایت را به زبان فارسی و یا هر زبانی که کاربر صحبت میکنه ارائه بده."
+    system_instruction = "تو چت بات هوش مصنوعی آلفا هستی. تو توانایی تحلیل دقیق ویدیو، صدا و تصویر را داری. اگر کاربر ویدیویی فرستاد، تمام فریم‌ها و صدای آن را بررسی کن و به سوالات پاسخ بده. همیشه پاسخ‌هایت را به زبان فارسی و یا هر زبانی که کاربر صحبت میکنه ارائه بده."
     
     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")
@@ -109,6 +109,11 @@ def chat():
                         processed_parts.append({"text": "[خطا: امکان پردازش فایل Word وجود نداشت.]"})
                 
                 else:
+                    # *** تغییر اصلی برای ویدیو اینجاست ***
+                    # اگر فایل ویدیویی بود، یک راهنمایی متنی اضافه می‌کنیم تا مدل گیج نشود
+                    if mime_type.startswith("video/"):
+                        processed_parts.append({"text": "این فایل یک ویدیو است. لطفا فریم‌ها، حرکات و صدای داخل ویدیو را با دقت تحلیل کن."})
+                    
                     processed_parts.append({"inline_data": {"mime_type": part["mimeType"], "data": part["base64Data"]}})
         
         if processed_parts:
@@ -121,7 +126,7 @@ def chat():
         return Response("data: [DONE]\n\n", mimetype='text/event-stream')
         
     def stream_response():
-        # تعداد دفعات تلاش کلی: به اندازه 2 برابر تعداد کلیدها تلاش میکنیم تا مطمئن شویم
+        # چرخش بین کلیدها در صورت بروز خطا
         max_attempts = len(GEMINI_API_KEYS) * 2 
         attempts = 0
         last_error = None
@@ -130,8 +135,7 @@ def chat():
             attempts += 1
             try:
                 api_key, key_index = get_next_key_with_index()
-                # logging.info(f"تلاش شماره {attempts}: استفاده از کلید ایندکس {key_index}")
-
+                
                 api_endpoint = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:streamGenerateContent?key={api_key}&alt=sse"
                 
                 payload = {
@@ -148,15 +152,15 @@ def chat():
                         "include_thoughts": True
                     }
 
-                # ارسال درخواست به گوگل
+                # درخواست به گوگل با تایم‌اوت کنترل شده
                 with requests.post(api_endpoint, json=payload, stream=True, timeout=(STREAM_CONNECT_TIMEOUT, STREAM_READ_TIMEOUT)) as response:
-                    # اگر خطای 429 (محدودیت نرخ) یا 5xx (خطای سرور گوگل) باشد
+                    
+                    # مدیریت خطاهای خاص گوگل (429, 500, 403) برای سوئیچ روی کلید بعدی
                     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 # برو به کلید بعدی
+                        continue
                     
-                    # اگر خطای 403 (Forbidden/Invalid Key) باشد
                     if response.status_code == 403:
                         logging.warning(f"کلید {key_index} نامعتبر است (403). رفتن به کلید بعدی...")
                         last_error = "Invalid API Key (403)"
@@ -164,9 +168,6 @@ def chat():
 
                     response.raise_for_status()
                     
-                    # اگر موفق شدیم وصل بشیم، شروع به استریم میکنیم
-                    # logging.info(f"اتصال موفق با کلید {key_index}. شروع استریم...")
-                    
                     data_received = False
                     for line in response.iter_lines():
                         if line:
@@ -191,38 +192,30 @@ def chat():
                                 except (json.JSONDecodeError, IndexError, KeyError):
                                     continue
                     
-                    # اگر دیتا با موفقیت دریافت شد و حلقه تمام شد، از تابع خارج میشویم
                     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)}. تلاش مجدد با کلید بعدی...")
+                logging.warning(f"خطای شبکه با کلید {key_index}: {str(e)}. تلاش مجدد...")
                 last_error = str(e)
-                continue # برو به کلید بعدی
+                continue
             
             except Exception as e:
                 logging.error(f"خطای غیرمنتظره با کلید {key_index}: {e}")
                 last_error = str(e)
-                continue # برای اطمینان، حتی با خطای عجیب هم سعی میکنیم با کلید بعدی ادامه بدیم
+                continue
         
-        # اگر همه تلاش‌ها شکست خورد
-        logging.critical("تمام کلیدها امتحان شدند و پاسخگو نبودند.")
-        # اینجا دیگر چاره‌ای نیست جز نمایش خطا، اما چون کاربر گفته خطا نبینه،
-        # می‌توانیم یک پیام مودبانه بفرستیم یا هیچ چیزی نفرستیم (که البته باعث میشه کاربر فکر کنه هنوز داره فکر میکنه)
-        # بهترین کار ارسال یک پیام خطای نرم است.
-        error_message = "سیستم در حال حاضر شلوغ است. لطفاً چند لحظه دیگر دوباره تلاش کنید."
-        error_payload = {"choices": [{"delta": {"content": error_message}}]} # به جای ارور، به عنوان متن چت میفرستیم
+        # اگر هیچ کلیدی کار نکرد، پیام خطای ملایم به کاربر بده (نه ارور قرمز)
+        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 ---