Update main.py

main.py

@@ -311,6 +311,32 @@ if GEMINI_KEYS_STR2:
 GEMINI_KEYS = list(set([k.strip() for k in _raw_keys if k.strip()]))
 print(f"✅ تعداد {len(GEMINI_KEYS)} کلید جیمینای با موفقیت شناسایی شد.")
 
+# --- 🚀 سیستم چرخشی پیشرفته (Round-Robin) برای کلیدهای جیمینای ---
+current_gemini_key_index = 0
+gemini_key_lock = threading.Lock()
+
+def get_next_gemini_keys(count=100):
+    """
+    این تابع تضمین می‌کند که به ترتیب از کلیدها استفاده شود.
+    مثلاً درخواست اول کلیدهای ۱ تا ۱۰۰ را می‌گیرد.
+    درخواست دوم کلیدهای ۱۰۱ تا ۲۰۰ را می‌گیرد.
+    و وقتی به آخر لیست برسد، دوباره از اول شروع می‌کند.
+    """
+    global current_gemini_key_index
+    with gemini_key_lock:
+        total_keys = len(GEMINI_KEYS)
+        if total_keys == 0:
+            return []
+            
+        actual_count = min(count, total_keys)
+        selected_keys = []
+        
+        for _ in range(actual_count):
+            selected_keys.append(GEMINI_KEYS[current_gemini_key_index])
+            current_gemini_key_index = (current_gemini_key_index + 1) % total_keys
+            
+        return selected_keys
+
 HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS = [k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
 
@@ -448,9 +474,8 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         history = history[-40:]
         if history[0]["role"] == "model": history = history[1:]
 
-    keys_to_try = GEMINI_KEYS.copy()
-    random.shuffle(keys_to_try)
-    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی برای تست در هر پاسخ
+    # دریافت ۱۰۰ کلید به صورت چرخشی
+    keys_to_try = get_next_gemini_keys(100)
     final_answer = None
     
     async with aiohttp.ClientSession() as session:
@@ -522,9 +547,8 @@ async def process_image(client, chat_id, prompt):
     proc_msg = await send_with_keyboard(client, chat_id, "✨ در حال ترجمه و بهینه‌سازی پرامپت شما توسط جیمینای...\n(تبدیل به پرامپت حرفه‌ای)", False)
     enhanced_prompt = prompt 
     if GEMINI_KEYS:
-        keys_to_try_gemini = GEMINI_KEYS.copy()
-        random.shuffle(keys_to_try_gemini)
-        keys_to_try_gemini = keys_to_try_gemini[:20] # انتخاب حداکثر ۲۰ کلید برای ترجمه پرامپت
+        # دریافت ۱۰۰ کلید به صورت چرخشی
+        keys_to_try_gemini = get_next_gemini_keys(100)
         gemini_sys_prompt = f"You are an expert AI image generation prompt engineer. Translate the following user input to English, and enhance it with high-quality, highly detailed, 4k resolution, cinematic lighting, and visually striking descriptive keywords. Return ONLY the final English prompt string.\nUser input: {prompt}"
         async with aiohttp.ClientSession() as session:
             for key in keys_to_try_gemini:
@@ -842,9 +866,8 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     mime_type, _ = mimetypes.guess_type(file_name)
     if not mime_type: mime_type = "audio/ogg" 
     
-    keys_to_try = GEMINI_KEYS.copy()
-    random.shuffle(keys_to_try)
-    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی
+    # دریافت ۱۰۰ کلید به صورت چرخشی
+    keys_to_try = get_next_gemini_keys(100)
     transcribed_text = None
     prompt = "لطفاً این فایل صوتی/تصویری را با دقت کامل گوش بده و صحبت‌های داخل آن را کلمه به کلمه به متن تبدیل کن. هیچ توضیح اضافه‌ای نده."
     
@@ -891,9 +914,8 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     mime_type, _ = mimetypes.guess_type(file_name)
     if not mime_type: mime_type = "image/jpeg"
     
-    keys_to_try = GEMINI_KEYS.copy()
-    random.shuffle(keys_to_try)
-    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی
+    # دریافت ۱۰۰ کلید به صورت چرخشی
+    keys_to_try = get_next_gemini_keys(100)
     final_answer = None
     
     async with aiohttp.ClientSession() as session: