Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - نسخه کامل و بهینه‌سازی شده برای پایداری بالا
+# app.py - نسخه کامل و بهینه‌سازی شده برای پایداری بالا (با اصلاح نهایی)
 
 import os
 import sys
@@ -25,12 +25,9 @@ except ImportError:
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
 
-# --- START: بخش جدید برای مدیریت بهینه Client ها ---
-GEMINI_CLIENTS_CACHE = {} # یک دیکشنری برای کش کردن client ها
-CLIENT_CACHE_LOCK = threading.Lock() # یک قفل برای دسترسی امن به کش
-# --- END: بخش جدید ---
+GEMINI_CLIENTS_CACHE = {} 
+CLIENT_CACHE_LOCK = threading.Lock() 
 
-# --- منطق مدیریت API Key ---
 ALL_API_KEYS: list[str] = []
 NEXT_KEY_INDEX: int = 0
 KEY_LOCK: threading.Lock = threading.Lock()
@@ -48,6 +45,10 @@ def get_next_api_key_and_client():
     """
     تغییر اصلی: این تابع هم کلید و هم یک client کش‌شده یا جدید را برمی‌گرداند.
     """
+    # --- START: خط اصلاح شده اینجاست ---
+    global NEXT_KEY_INDEX 
+    # --- END: خط اصلاح شده ---
+
     with KEY_LOCK:
         if not ALL_API_KEYS:
             return None, None, -1
@@ -57,13 +58,10 @@ def get_next_api_key_and_client():
         key_display_index = current_index + 1
         NEXT_KEY_INDEX += 1
 
-    # حالا که کلید را داریم، client مربوط به آن را از کش دریافت می‌کنیم
     with CLIENT_CACHE_LOCK:
         if key_to_use in GEMINI_CLIENTS_CACHE:
-            # اگر client در کش بود، همان را برمی‌گردانیم
             client = GEMINI_CLIENTS_CACHE[key_to_use]
         else:
-            # اگر نبود، یکی جدید می‌سازیم و در کش ذخیره می‌کنیم
             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
@@ -75,7 +73,6 @@ FIXED_MODEL_NAME = "gemini-2.5-flash-preview-tts"
 DEFAULT_MAX_CHUNK_SIZE = 3800
 DEFAULT_SLEEP_BETWEEN_REQUESTS = 8
 
-# ... (توابع کمکی دیگر مثل save_binary_file, convert_to_wav و ... بدون تغییر اینجا قرار می‌گیرند) ...
 def save_binary_file(file_name, data):
     try:
         with open(file_name, "wb") as f: f.write(data)
@@ -128,22 +125,17 @@ def merge_audio_files_func(file_paths, output_path):
         combined.export(output_path, format="wav")
         return True
     except Exception as e: logging.error(f"❌ خطا در ادغام فایل‌های صوتی: {e}"); return False
-# ... (پایان توابع کمکی) ...
-
 
-# --- منطق اصلی تولید صدا (با استفاده از Client کش شده) ---
 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)):
-        # تغییر اصلی: دریافت همزمان کلید و client
         selected_api_key, client, key_idx_display = get_next_api_key_and_client()
         if not client:
             break
 
         logging.info(f"[{session_id}] ⚙️ تلاش برای تولید قطعه با کلید API شماره {key_idx_display}")
         try:
-            # دیگر نیازی به ساخت client جدید نیست!
             final_text = f'"{prompt_text}"\n{chunk_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"],
@@ -156,17 +148,14 @@ def generate_audio_chunk_with_retry(chunk_text, prompt_text, voice, temp, sessio
                 return response.candidates[0].content.parts[0].inline_data
         except Exception as e:
             logging.error(f"[{session_id}] ❌ خطا در تولید قطعه با کلید شماره {key_idx_display}: {e}.")
-            # اگر خطا مربوط به احراز هویت بود، client را از کش حذف می‌کنیم تا دفعه بعد دوباره ساخته شود
             if "authentication" in str(e).lower():
                 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.")
-
     return None
 
 def core_generate_audio(text_input, prompt_input, selected_voice, temperature_val, session_id):
-    # این تابع بدون تغییر باقی می‌ماند
     logging.info(f"[{session_id}] 🚀 شروع فرآیند تولید صدا.")
     temp_dir = f"temp_{session_id}"
     os.makedirs(temp_dir, exist_ok=True)
@@ -207,10 +196,8 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
         if os.path.exists(temp_dir):
             shutil.rmtree(temp_dir)
 
-# --- اجرای کدهای اولیه برنامه ---
 _init_api_keys()
 
-# --- تعریف اپلیکیشن FastAPI ---
 app = FastAPI(title="Alpha TTS Worker API")
 
 class TTSRequest(BaseModel):