Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - نسخه نهایی با اجرای همزمان واقعی برای حداکثر پایداری
+# app.py - نسخه نهایی با انتخاب تصادفی کلیدها و تلاش ۵۰ باره
 
 import os
 import sys
@@ -11,6 +11,7 @@ import shutil
 import logging
 import mimetypes
 import threading
+import random  # اضافه شد برای انتخاب تصادفی
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from google import genai
@@ -25,42 +26,40 @@ except ImportError:
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
 
+# --- تنظیمات مدیریت کلیدها ---
 GEMINI_CLIENTS_CACHE = {} 
 CLIENT_CACHE_LOCK = threading.Lock() 
 
 ALL_API_KEYS: list[str] = []
-NEXT_KEY_INDEX: int = 0
-KEY_LOCK: threading.Lock = threading.Lock()
 
 def _init_api_keys():
     global ALL_API_KEYS
     all_keys_string = os.environ.get("ALL_GEMINI_API_KEYS")
     if all_keys_string:
+        # حذف فاصله‌های اضافی و جدا کردن با کاما
         ALL_API_KEYS = [key.strip() for key in all_keys_string.split(',') if key.strip()]
     logging.info(f"✅ تعداد {len(ALL_API_KEYS)} کلید API جیمینای بارگذاری شد.")
     if not ALL_API_KEYS:
         logging.warning("⛔️ هشدار: هیچ Secret با نام ALL_GEMINI_API_KEYS یافت نشد!")
 
-def get_next_api_key_and_client():
-    global NEXT_KEY_INDEX
-    with KEY_LOCK:
-        if not ALL_API_KEYS:
-            return None, None, -1
-        
-        current_index = NEXT_KEY_INDEX % len(ALL_API_KEYS)
-        key_to_use = ALL_API_KEYS[current_index]
-        key_display_index = current_index + 1
-        NEXT_KEY_INDEX += 1
-
+# --- تغییر: تابع انتخاب تصادفی کلید ---
+def get_random_api_key_and_client():
+    if not ALL_API_KEYS:
+        return None, None
+    
+    # انتخاب کاملاً تصادفی یک کلید از لیست
+    key_to_use = random.choice(ALL_API_KEYS)
+    
     with CLIENT_CACHE_LOCK:
         if key_to_use in GEMINI_CLIENTS_CACHE:
             client = GEMINI_CLIENTS_CACHE[key_to_use]
         else:
-            logging.info(f"Creating new Gemini client for key ending in ...{key_to_use[-4:]}")
+            # ساخت کلاینت جدید اگر در کش نباشد
+            # logging.info(f"Creating new Gemini client for key ending in ...{key_to_use[-4:]}")
             client = genai.Client(api_key=key_to_use)
             GEMINI_CLIENTS_CACHE[key_to_use] = client
             
-    return key_to_use, client, key_display_index
+    return key_to_use, client
 
 FIXED_MODEL_NAME = "gemini-2.5-flash-native-audio-preview-09-2025"
 DEFAULT_MAX_CHUNK_SIZE = 3800
@@ -123,33 +122,50 @@ def merge_audio_files_func(file_paths, output_path):
         return True
     except Exception as e: logging.error(f"❌ خطا در ادغام فایل‌های صوتی: {e}"); return False
 
+# --- تغییر: منطق تلاش مجدد تصادفی تا ۵۰ بار ---
 def generate_audio_chunk_with_retry(chunk_text, prompt_text, voice, temp, session_id):
     if not ALL_API_KEYS: raise Exception("هیچ کلید API برای پردازش در دسترس نیست.")
     
-    for _ in range(len(ALL_API_KEYS)):
-        selected_api_key, client, key_idx_display = get_next_api_key_and_client()
+    MAX_RETRIES = 50  # درخواست شما: تا ۵۰ بار تلاش تصادفی
+    
+    for attempt in range(MAX_RETRIES):
+        selected_api_key, client = get_random_api_key_and_client()
+        
         if not client:
+            logging.error(f"[{session_id}] کلاینت یافت نشد، تلاش متوقف شد.")
             break
 
-        logging.info(f"[{session_id}] ⚙️ تلاش برای تولید قطعه با کلید API شماره {key_idx_display}")
+        # لاگ کوتاه برای نشان دادن شماره تلاش
+        # logging.info(f"[{session_id}] ⚙️ تلاش {attempt+1}/{MAX_RETRIES} با کلید تصادفی ...{selected_api_key[-4:]}")
+        
         try:
             final_text = f'{chunk_text}({prompt_text})' if prompt_text and prompt_text.strip() else chunk_text
             contents = [types.Content(role="user", parts=[types.Part.from_text(text=final_text)])]
             config = types.GenerateContentConfig(temperature=temp, response_modalities=["audio"],
                 speech_config=types.SpeechConfig(voice_config=types.VoiceConfig(
                     prebuilt_voice_config=types.PrebuiltVoiceConfig(voice_name=voice))))
+            
             response = client.models.generate_content(model=FIXED_MODEL_NAME, contents=contents, config=config)
             
             if response.candidates and response.candidates[0].content and response.candidates[0].content.parts and response.candidates[0].content.parts[0].inline_data:
-                logging.info(f"[{session_id}] ✅ قطعه با موفقیت توسط کلید شماره {key_idx_display} تولید شد.")
+                logging.info(f"[{session_id}] ✅ قطعه با موفقیت در تلاش {attempt+1} تولید شد.")
                 return response.candidates[0].content.parts[0].inline_data
+                
         except Exception as e:
-            logging.error(f"[{session_id}] ❌ خطا در تولید قطعه با کلید شماره {key_idx_display}: {e}.")
-            if "authentication" in str(e).lower():
+            error_msg = str(e).lower()
+            logging.warning(f"[{session_id}] ⚠️ خطا در تلاش {attempt+1} (کلید ...{selected_api_key[-4:]}): {e}")
+            
+            # اگر خطا مربوط به اعتبار یا بن شدن کلید بود، از کش حذفش می‌کنیم تا کلاینت رفرش شود
+            # اما چون انتخاب تصادفی است، دفعه بعد ممکن است دوباره انتخاب شود که مشکلی نیست چون شاید خطای لحظه‌ای بوده
+            if "authentication" in error_msg or "key" in error_msg or "403" in error_msg:
                 with CLIENT_CACHE_LOCK:
                     if selected_api_key in GEMINI_CLIENTS_CACHE:
                         del GEMINI_CLIENTS_CACHE[selected_api_key]
-                        logging.warning(f"Client for key ...{selected_api_key[-4:]} removed from cache due to auth error.")
+                        logging.warning(f"Client for key ...{selected_api_key[-4:]} removed from cache due to fatal error.")
+            
+            # وقفه کوتاه قبل از تلاش تصادفی بعدی
+            time.sleep(1)
+            
     return None
 
 def core_generate_audio(text_input, prompt_input, selected_voice, temperature_val, session_id):
@@ -174,7 +190,7 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
                 fpath = save_binary_file(f"{output_base_name}_part{i+1:03d}{ext}", data_buffer)
                 if fpath: generated_files.append(fpath)
             else:
-                raise Exception(f"فرآیند متوقف شد زیرا تولید قطعه {i+1}  با تمام سرورهای موجود ناموفق بود.امروز تعداد فایل های فوق العاده زیادی با این ربات ساخته شده فردا باید مجدداً تولید کنید  ")
+                raise Exception(f"تولید قطعه {i+1} پس از ۵۰ تلاش ناموفق بود. احتمالاً تمام کلیدها مشکل دارند یا سرور شلوغ است.")
             if i < len(text_chunks) - 1 and len(text_chunks) > 1: time.sleep(DEFAULT_SLEEP_BETWEEN_REQUESTS)
         if not generated_files: raise Exception("هیچ فایل صوتی تولید نشد.")
         final_output_path = f"output_{session_id}.wav"
@@ -203,12 +219,8 @@ class TTSRequest(BaseModel):
     speaker: str
     temperature: float
 
-# --- START: تغییر اصلی برای اجرای همزمان واقعی ---
-# کلمه کلیدی async از تعریف تابع حذف شده است.
-# این به FastAPI می‌گوید که این تابع سنگین را در یک thread جداگانه اجرا کند.
 @app.post("/generate")
 def generate_audio_endpoint(request: TTSRequest):
-# --- END: تغییر اصلی ---
     session_id = str(uuid.uuid4())[:8]
     logging.info(f"[{session_id}] 🏁 درخواست جدید API در این Worker دریافت شد.")
     try: