Update app.py

app.py

@@ -8,6 +8,8 @@ import time
 import zipfile
 from google import genai
 from google.genai import types
+import threading # اضافه شد: برای KEY_LOCK و ریسه
+import logging # اضافه شد: برای لاگ استاندارد
 
 try:
     from pydub import AudioSegment
@@ -15,37 +17,50 @@ try:
 except ImportError:
     PYDUB_AVAILABLE = False
 
-# --- START: منطق چرخش API Key ---
-GEMINI_API_KEYS = []
-i = 1
-while os.environ.get(f"GEMINI_API_KEY_{i}"):
-  GEMINI_API_KEYS.append(os.environ.get(f"GEMINI_API_KEY_{i}"))
-  i += 1
+# --- START: پیکربندی لاگینگ ---
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
+# --- END: پیکربندی لاگینگ ---
 
-NUM_API_KEYS = len(GEMINI_API_KEYS)
-CURRENT_KEY_INDEX = 0
+# --- START: منطق جدید مدیریت API Key (مشابه کد قبلی شما) ---
+ALL_API_KEYS: list[str] = []
+NEXT_KEY_INDEX: int = 0
+KEY_LOCK: threading.Lock = threading.Lock() # برای اطمینان از ایمنی تردها هنگام به روز رسانی ایندکس
 
-def _log(message):
-    """تابع ساده شده برای لاگ کردن پیام‌ها به کنسول."""
-    print(f"[لاگ آلفا TTS] {message}")
+def _init_api_keys():
+    """
+    کلیدهای API را از یک متغیر محیطی واحد شناسایی و مرتب می‌کند.
+    این تابع باید یک بار هنگام شروع برنامه اجرا شود.
+    """
+    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 یافت نشد!")
+        logging.warning("   لطفاً Secret را به عنوان یک رشته با کاما جدا شده (مثال: key1,key2,key3) در تنظیمات Space خود اضافه کنید.")
 
-if NUM_API_KEYS == 0:
-  _log("⛔️ خطای حیاتی: هیچ Secret با نام GEMINI_API_KEY_n یافت نشد!")
-  _log("   لطفاً Secret ها را مانند GEMINI_API_KEY_1, GEMINI_API_KEY_2, ... در تنظیمات Space خود اضافه کنید.")
-else:
-  _log(f"✅ تعداد {NUM_API_KEYS} کلید API جیمینای بارگذاری شد.")
+# فراخوانی تابع شناسایی کلیدها در ابتدای برنامه
+_init_api_keys()
 
 def get_next_api_key():
-  global CURRENT_KEY_INDEX
-  if NUM_API_KEYS == 0:
-    _log("⚠️ تلاش برای گرفتن کلید API در حالی که هیچ کلیدی بارگذاری نشده است.")
-    return None, None
-  
-  key_to_use = GEMINI_API_KEYS[CURRENT_KEY_INDEX % NUM_API_KEYS]
-  key_display_index = (CURRENT_KEY_INDEX % NUM_API_KEYS) + 1
-  CURRENT_KEY_INDEX += 1
-  return key_to_use, key_display_index
-# --- END: منطق چرخش API Key ---
+    """
+    کلید API بعدی را به صورت چرخشی برمی‌گرداند.
+    """
+    global NEXT_KEY_INDEX, ALL_API_KEYS, KEY_LOCK
+    
+    with KEY_LOCK: # اطمینان از اینکه تنها یک ترد در هر زمان به ایندکس دسترسی دارد
+        if not ALL_API_KEYS:
+            return None, None
+
+        key_to_use = ALL_API_KEYS[NEXT_KEY_INDEX % len(ALL_API_KEYS)]
+        key_display_index = (NEXT_KEY_INDEX % len(ALL_API_KEYS)) + 1
+        NEXT_KEY_INDEX += 1
+        return key_to_use, key_display_index
+# --- END: منطق جدید مدیریت API Key ---
 
 SPEAKER_VOICES = [
     "Achird", "Zubenelgenubi", "Vindemiatrix", "Sadachbia", "Sadaltager",
@@ -64,7 +79,7 @@ def save_binary_file(file_name, data):
         with open(file_name, "wb") as f: f.write(data)
         return file_name
     except Exception as e:
-        _log(f"❌ خطا در ذخیره فایل {file_name}: {e}")
+        logging.error(f"❌ خطا در ذخیره فایل {file_name}: {e}")
         return None
 
 def convert_to_wav(audio_data: bytes, mime_type: str) -> bytes:
@@ -106,31 +121,40 @@ def smart_text_split(text, max_size=3800):
     return final_chunks
 
 def merge_audio_files_func(file_paths, output_path):
-    if not PYDUB_AVAILABLE: _log("⚠️ pydub برای ادغام در دسترس نیست."); return False
+    if not PYDUB_AVAILABLE: logging.warning("⚠️ pydub برای ادغام در دسترس نیست."); return False
     try:
         combined = AudioSegment.empty()
         for i, fp in enumerate(file_paths):
             if os.path.exists(fp): combined += AudioSegment.from_file(fp) + (AudioSegment.silent(duration=150) if i < len(file_paths) - 1 else AudioSegment.empty())
-            else: _log(f"⚠️ فایل برای ادغام پیدا نشد: {fp}")
+            else: logging.warning(f"⚠️ فایل برای ادغام پیدا نشد: {fp}")
         combined.export(output_path, format="wav")
         return True
-    except Exception as e: _log(f"❌ خطا در ادغام فایل‌های صوتی: {e}"); return False
+    except Exception as e: logging.error(f"❌ خطا در ادغام فایل‌های صوتی: {e}"); return False
 
-# --- START: منطق جدید تولید صدا با قابلیت تلاش مجدد ---
+# --- START: منطق تولید صدا با قابلیت تلاش مجدد با کلیدهای چرخشی ---
 
 def generate_audio_chunk_with_retry(chunk_text, prompt_text, voice, temp):
     """
     یک قطعه صوتی را با قابلیت تلاش مجدد با کلیدهای مختلف API تولید می‌کند.
-    اگر یک کلید ناموفق بود، به طور خودکار کلید بعدی را امتحان می‌کند.
+    اگر یک کلید ناموفق بود، به طور خودکار کلید بعدی را امتحان می‌کند تا تمام کلیدها بررسی شوند.
     """
-    # تلاش برای تولید صدا به تعداد کلیدهای موجود
-    for i in range(NUM_API_KEYS):
+    # تلاش برای تولید صدا به تعداد کلیدهای موجود در ALL_API_KEYS
+    # از len(ALL_API_KEYS) به جای NUM_API_KEYS استفاده میکنیم چون NUM_API_KEYS حذف شده است
+    if not ALL_API_KEYS:
+        logging.error("❌ هیچ کلید API برای تولید صدا در دسترس نیست.")
+        return None
+
+    # این حلقه به تعداد کل کلیدهای موجود در ALL_API_KEYS تلاش می‌کند.
+    # در هر دور، یک کلید جدید و چرخشی از get_next_api_key دریافت می‌شود.
+    for _ in range(len(ALL_API_KEYS)): 
         selected_api_key, key_idx_display = get_next_api_key()
+        
+        # اگر به دلیلی get_next_api_key، کلیدی برنگرداند (نبود کلید یا خطای داخلی)
         if not selected_api_key:
-            _log("⚠️ هیچ کلید API برای تلاش باقی نمانده است.")
-            continue
+            logging.warning("⚠️ get_next_api_key هیچ کلیدی برنگرداند. تلاش‌های باقیمانده نادیده گرفته می‌شوند.")
+            break 
 
-        _log(f"⚙️ تلاش برای تولید قطعه با کلید API شماره {key_idx_display} (...{selected_api_key[-4:]})")
+        logging.info(f"⚙️ تلاش برای تولید قطعه با کلید API شماره {key_idx_display} (...{selected_api_key[-4:]})")
 
         try:
             client = genai.Client(api_key=selected_api_key)
@@ -143,35 +167,35 @@ def generate_audio_chunk_with_retry(chunk_text, prompt_text, voice, temp):
             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:
-                _log(f"✅ قطعه با موفقیت توسط کلید شماره {key_idx_display} تولید شد.")
+                logging.info(f"✅ قطعه با موفقیت توسط کلید شماره {key_idx_display} تولید شد.")
                 return response.candidates[0].content.parts[0].inline_data
             else:
-                _log(f"⚠️ پاسخ API برای قطعه با کلید شماره {key_idx_display} بدون داده صوتی بود. تلاش با کلید بعدی...")
+                logging.warning(f"⚠️ پاسخ API برای قطعه با کلید شماره {key_idx_display} بدون داده صوتی بود. تلاش با کلید بعدی...")
 
         except Exception as e:
-            _log(f"❌ خطا در تولید قطعه با کلید شماره {key_idx_display}: {e}. تلاش با کلید بعدی...")
+            logging.error(f"❌ خطا در تولید قطعه با کلید شماره {key_idx_display}: {e}. تلاش با کلید بعدی...")
     
-    _log("❌ تمام کلیدهای API امتحان شدند اما هیچ‌کدام موفق به تولید قطعه نشدند.")
+    logging.error("❌ تمام کلیدهای API امتحان شدند اما هیچ‌کدام موفق به تولید قطعه نشدند.")
     return None
 
 def core_generate_audio(text_input, prompt_input, selected_voice, temperature_val):
-    _log("🚀 شروع فرآیند تولید صدا با قابلیت تعویض کلید خودکار...")
+    logging.info("🚀 شروع فرآیند تولید صدا با قابلیت تعویض کلید خودکار...")
 
     output_base_name = DEFAULT_OUTPUT_FILENAME_BASE
     max_chunk, sleep_time = DEFAULT_MAX_CHUNK_SIZE, DEFAULT_SLEEP_BETWEEN_REQUESTS
 
     if not text_input or not text_input.strip():
-        _log("❌ متن ورودی خالی است.")
+        logging.error("❌ متن ورودی خالی است.")
         return None
 
     text_chunks = smart_text_split(text_input, max_chunk)
     if not text_chunks:
-        _log("❌ متن قابل پردازش به قطعات کوچکتر نیست.")
+        logging.error("❌ متن قابل پردازش به قطعات کوچکتر نیست.")
         return None
 
     generated_files = []
     for i, chunk in enumerate(text_chunks):
-        _log(f"🔊 پردازش قطعه {i+1}/{len(text_chunks)}...")
+        logging.info(f"🔊 پردازش قطعه {i+1}/{len(text_chunks)}...")
         
         inline_data = generate_audio_chunk_with_retry(chunk, prompt_input, selected_voice, temperature_val)
 
@@ -187,17 +211,17 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
             if fpath: 
                 generated_files.append(fpath)
             else:
-                 _log(f"❌ موفق به ذخیره فایل برای قطعه {i+1} نشدیم. این قطعه نادیده گرفته می‌شود.")
+                 logging.error(f"❌ موفق به ذخیره فایل برای قطعه {i+1} نشدیم. این قطعه نادیده گرفته می‌شود.")
                  continue
         else:
-            _log(f"🛑 فرآیند متوقف شد زیرا تولید قطعه {i+1} با تمام کلیدهای موجود ناموفق بود.")
-            break 
+            logging.error(f"🛑 فرآیند متوقف شد زیرا تولید قطعه {i+1} با تمام کلیدهای موجود ناموفق بود.")
+            break # اگر یک قطعه با هیچ کلیدی تولید نشد، کل فرآیند را متوقف می کنیم.
 
         if i < len(text_chunks) - 1 and len(text_chunks) > 1: 
             time.sleep(sleep_time)
 
     if not generated_files:
-        _log(f"❌ هیچ فایل صوتی تولید نشد.")
+        logging.error(f"❌ هیچ فایل صوتی تولید نشد.")
         return None
     
     final_audio_file = None
@@ -218,7 +242,7 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
                         os.rename(generated_files[0], renamed_first_chunk)
                         final_audio_file = renamed_first_chunk
                     except Exception as e_rename:
-                         _log(f"خطا در تغییر نام فایل اولین قطعه (پس از ادغام ناموفق): {e_rename}")
+                         logging.error(f"خطا در تغییر نام فایل اولین قطعه (پس از ادغام ناموفق): {e_rename}")
                          final_audio_file = generated_files[0]
             
             for fp_cleanup in generated_files:
@@ -228,7 +252,7 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
                 except: pass
 
         else: 
-            _log("⚠️ pydub در دسترس نیست. اولین قطعه صوتی ارائه می‌شود.")
+            logging.warning("⚠️ pydub در دسترس نیست. اولین قطعه صوتی ارائه می‌شود.")
             if generated_files:
                 try:
                     target_ext = os.path.splitext(generated_files[0])[1]
@@ -240,7 +264,7 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
                         try: os.remove(generated_files[i_gf])
                         except: pass
                 except Exception as e_rename_single:
-                    _log(f"خطا در تغییر نام فایل اولین قطعه (بدون pydub): {e_rename_single}")
+                    logging.error(f"خطا در تغییر نام فایل اولین قطعه (بدون pydub): {e_rename_single}")
                     final_audio_file = generated_files[0]
 
     elif len(generated_files) == 1:
@@ -251,16 +275,16 @@ def core_generate_audio(text_input, prompt_input, selected_voice, temperature_va
             os.rename(generated_files[0], final_single_fn)
             final_audio_file = final_single_fn
         except Exception as e_rename_single_final:
-            _log(f"خطا در تغییر نام فایل تکی نهایی: {e_rename_single_final}")
+            logging.error(f"خطا در تغییر نام فایل تکی نهایی: {e_rename_single_final}")
             final_audio_file = generated_files[0]
     
     if final_audio_file and os.path.exists(final_audio_file):
-        _log(f"✅ فایل صوتی نهایی با موفقیت تولید شد: {os.path.basename(final_audio_file)}")
+        logging.info(f"✅ فایل صوتی نهایی با موفقیت تولید شد: {os.path.basename(final_audio_file)}")
     elif final_audio_file:
-        _log(f"⚠️ فایل نهایی '{final_audio_file}' پس از پردازش وجود ندارد!")
+        logging.warning(f"⚠️ فایل نهایی '{final_audio_file}' پس از پردازش وجود ندارد!")
         return None
     else:
-        _log(f"❓ وضعیت نامشخص برای فایل نهایی.")
+        logging.error(f"❓ وضعیت نامشخص برای فایل نهایی.")
         return None
         
     return final_audio_file
@@ -273,17 +297,27 @@ def gradio_tts_interface(use_file_input, uploaded_file, text_to_speak, speech_pr
         if uploaded_file:
             try:
                 with open(uploaded_file.name, 'r', encoding='utf-8') as f: actual_text = f.read().strip()
-                if not actual_text: _log("❌ فایل آپلود شده خالی است یا خوانده نشد."); return None
-            except Exception as e: _log(f"❌ خطا در خواندن فایل آپلود شده: {e}"); return None
-        else: _log("❌ گزینه استفاده از فایل انتخاب شده اما فایلی آپلود نشده."); return None
+                if not actual_text: logging.error("❌ فایل آپلود شده خالی است یا خوانده نشد."); return None
+            except Exception as e: logging.error(f"❌ خطا در خواندن فایل آپلود شده: {e}"); return None
+        else: logging.warning("❌ گزینه استفاده از فایل انتخاب شده اما فایلی آپلود نشده."); return None
     else:
         actual_text = text_to_speak
-        if not actual_text or not actual_text.strip(): _log("❌ متن ورودی برای تبدیل خالی است."); return None
+        if not actual_text or not actual_text.strip(): logging.warning("❌ متن ورودی برای تبدیل خالی است."); return None
     
     final_path = core_generate_audio(actual_text, speech_prompt, speaker_voice, temperature)
     return final_path
 
-# --- CSS ---
+# --- تابع جدید برای ریست خودکار هر 24 ساعت ---
+def auto_restart_service():
+    RESTART_INTERVAL_SECONDS = 24 * 60 * 60
+    logging.info(f"سرویس برای ری‌استارت خودکار پس از {RESTART_INTERVAL_SECONDS / 3600:.0f} ساعت زمان‌بندی شده است.")
+    time.sleep(RESTART_INTERVAL_SECONDS)
+    logging.info(f"زمان ری‌استارت خودکار فرا رسیده است. برنامه برای ری‌استارت خارج می‌شود...")
+    os._exit(1)
+# --- END: تابع ری‌استارت خودکار ---
+
+
+# --- CSS و Gradio UI (بدون تغییر) ---
 custom_css_inspired_by_image = f"""
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700;800&display=swap');
 :root {{
@@ -401,7 +435,10 @@ with gr.Blocks(theme=gr.themes.Base(font=[gr.themes.GoogleFont("Vazirmatn")]), c
     gr.Markdown("<p class='app-footer-final'>Alpha Language Learning © 2024</p>")
 
 if __name__ == "__main__":
-    if NUM_API_KEYS > 0 :
-        demo.launch()
+    threading.Thread(target=auto_restart_service, daemon=True, name="AutoRestartThread").start()
+    
+    # استفاده از len(ALL_API_KEYS) به جای NUM_API_KEYS قدیمی
+    if len(ALL_API_KEYS) > 0 :
+        demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))
     else:
-        _log("🔴 برنامه به دلیل عدم وجود کلید API جیمینای اجرا نشد. لطفاً Secrets را بررسی کنید.")
+        logging.critical("🔴 برنامه به دلیل عدم وجود کلید API جیمینای اجرا نشد. لطفاً Secrets را بررسی کنید.")