Update app.py

app.py

@@ -108,7 +108,7 @@ def get_user_ip():
         return request.headers.getlist("X-Forwarded-For")[0].split(',')[0].strip()
     return request.remote_addr
 
-# --- TTS HELPER FUNCTIONS (Anti-DDoS & Jitter Optimized) ---
+# --- TTS HELPER FUNCTIONS (Optimized: Jitter + Safe Retries) ---
 def call_worker(index, chunk_payload):
     text_length = len(chunk_payload.get("text", ""))
     use_live = True if text_length <= 500 else False
@@ -126,16 +126,18 @@ def call_worker(index, chunk_payload):
         "fallback_to_live": True
     }
 
-    max_attempts = 15 
+    # افزایش سماجت به 40 بار تلاش
+    max_attempts = 40 
     
     for attempt in range(max_attempts):
         workers = list(WORKER_URLS)
-        random.shuffle(workers)
+        random.shuffle(workers) # کارگرها هر بار رندوم چیده میشوند
         
         for worker_url in workers:
             try:
-                logging.info(f"Chunk {index} -> Sending to {worker_url} (Attempt {attempt+1})")
-                response = requests.post(worker_url, json=worker_payload, timeout=120)
+                logging.info(f"Chunk {index} -> Sending to {worker_url} (Attempt {attempt+1}/{max_attempts})")
+                # افزایش تایم اوت به 150 ثانیه
+                response = requests.post(worker_url, json=worker_payload, timeout=150)
                 
                 if response.status_code == 200:
                     audio_data = io.BytesIO(response.content)
@@ -143,28 +145,26 @@ def call_worker(index, chunk_payload):
                     return index, audio_segment
                     
                 elif response.status_code == 429:
-                    logging.warning(f"Worker {worker_url} is busy (429). Trying next worker...")
-                    time.sleep(1 + random.uniform(0, 1.5)) # زمان‌بندی تصادفی برای جلوگیری از هم‌زمانی مطلق
+                    # 429 یعنی شلوغی، پس بدون معطلی میریم کارگر بعدی
                     continue
                     
                 else:
-                    logging.warning(f"Worker Error {worker_url}: {response.status_code}. Trying next worker...")
-                    time.sleep(1)
+                    # ارور 500 و بقیه، میریم کارگر بعدی
                     continue
                     
             except Exception as e:
-                logging.warning(f"Worker Connection Fail {worker_url}: {str(e)[:50]}. Trying next worker...")
-                time.sleep(1)
+                # تایم اوت یا خطای شبکه، میریم کارگر بعدی
                 continue
         
-        # استراحت تصاعدی + تصادفی در صورت مسدود شدن تمام کارگرها
-        sleep_time = 4 + (attempt * 2) + random.uniform(0, 2)
-        logging.warning(f"All workers failed for Chunk {index} on attempt {attempt+1}. Sleeping for {sleep_time:.2f}s...")
+        # اگر هر سه کارگر خطا دادند، اینجا Jitter (تاخیر تصادفی) اعمال می‌کنیم
+        # این کار باعث میشه Thread ها همزمان بیدار نشن
+        sleep_time = 4 + (attempt * 1.5) + random.uniform(0.5, 3.5)
+        logging.warning(f"All workers busy/failed for Chunk {index}. Sleeping for {sleep_time:.2f}s...")
         time.sleep(sleep_time)
     
     return index, None
 
-# --- AI PODCAST SCRIPT LOGIC ---
+# --- AI PODCAST SCRIPT LOGIC (Cohere Labs Space) ---
 def generate_podcast_in_background(task_id, system_prompt, safety_settings):
     try:
         AYA_SPACE_URL = "https://coherelabs-aya-expanse.hf.space/gradio_api"
@@ -235,7 +235,7 @@ def generate_podcast_in_background(task_id, system_prompt, safety_settings):
     except Exception as e:
         with tasks_lock: tasks[task_id].update({'status': 'failed', 'error': str(e)})
 
-# --- FULL AUTO PODCAST LOGIC ---
+# --- FULL AUTO PODCAST LOGIC (پردازش موازی مدیریت شده) ---
 def generate_full_podcast_audio_background(task_id, prompt, speakers):
     try:
         with tasks_lock:
@@ -309,7 +309,7 @@ Dialogue rules: No stage directions like [laugh], (sigh). Just spoken words."""
         with tasks_lock:
             tasks[task_id] = {'status': 'generating_audio', 'progress': f'در حال تولید همزمان صداها (0 از {total_turns} تکمیل شده)'}
         
-        # --- سیستم پردازش موازی محدود شده (جلوگیری از قفل شدن هاگینگ فیس) ---
+        # --- سیستم پردازش موازی بهینه‌شده ---
         audio_results = [None] * total_turns
         completed_count = 0
         
@@ -322,8 +322,9 @@ Dialogue rules: No stage directions like [laugh], (sigh). Just spoken words."""
                 raise ValueError(f"خطا در تولید صدای نوبت {index+1}")
             return idx, audio_seg
 
-        # محدود کردن به ۳ کارگر همزمان (دقیقاً برابر با تعداد اسپیس‌های TTS شما)
-        with concurrent.futures.ThreadPoolExecutor(max_workers=min(3, total_turns)) as executor:
+        # 🚨 کاهش فشار: اجرای درخواست‌ها تا سقف ۴ فایل موازی (جلوگیری از خفگی کارگرها)
+        max_concurrent_workers = min(4, total_turns)
+        with concurrent.futures.ThreadPoolExecutor(max_workers=max_concurrent_workers) as executor:
             future_to_index = {executor.submit(process_single_chunk, i, turn): i for i, turn in enumerate(script_turns)}
             
             for future in concurrent.futures.as_completed(future_to_index):
@@ -332,7 +333,7 @@ Dialogue rules: No stage directions like [laugh], (sigh). Just spoken words."""
                     audio_results[idx] = audio_seg
                     completed_count += 1
                     with tasks_lock:
-                        tasks[task_id]['progress'] = f'در حال تولید همزمان صداها ({completed_count} از {total_turns} تکمیل شده)'
+                        tasks[task_id]['progress'] = f'در حال تولید صداها ({completed_count} از {total_turns} تکمیل شده)'
                 except Exception as exc:
                     raise Exception(str(exc))