Update main.py

main.py

@@ -298,8 +298,18 @@ async def helper_download_file(client, msg_obj):
 
 
 # --- تنظیمات کلیدها ---
-GEMINI_KEYS_STR = os.environ.get("GEMINI_API_KEYS", "")
-GEMINI_KEYS = [k.strip() for k in GEMINI_KEYS_STR.split(",") if k.strip()]
+GEMINI_KEYS_STR1 = os.environ.get("GEMINI_API_KEYS1", "")
+GEMINI_KEYS_STR2 = os.environ.get("GEMINI_API_KEYS2", "")
+
+_raw_keys = []
+if GEMINI_KEYS_STR1:
+    _raw_keys.extend(GEMINI_KEYS_STR1.split(","))
+if GEMINI_KEYS_STR2:
+    _raw_keys.extend(GEMINI_KEYS_STR2.split(","))
+
+# حذف کلیدهای تکراری و خالی
+GEMINI_KEYS = list(set([k.strip() for k in _raw_keys if k.strip()]))
+print(f"✅ تعداد {len(GEMINI_KEYS)} کلید جیمینای با موفقیت شناسایی شد.")
 
 HF_TOKENS_STR = os.environ.get("HF_TOKENS", "")
 HF_TOKENS = [k.strip() for k in HF_TOKENS_STR.split(",") if k.strip()]
@@ -440,6 +450,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
+    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی برای تست در هر پاسخ
     final_answer = None
     
     async with aiohttp.ClientSession() as session:
@@ -465,7 +476,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 
     if not final_answer:
         if history and history[-1]["role"] == "user": history.pop()
-        await send_with_keyboard(client, chat_id, "❌ متأسفانه پاسخی دریافت نشد. (شاید سایز فایل بیش از حد مجاز بوده است)", False)
+        await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", False)
         return
 
     # کسر اعتبار در دیتاست (اگر نامحدود نیست)
@@ -513,6 +524,7 @@ async def process_image(client, chat_id, 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] # انتخاب حداکثر ۲۰ کلید برای ترجمه پرامپت
         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:
@@ -832,6 +844,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
+    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی
     transcribed_text = None
     prompt = "لطفاً این فایل صوتی/تصویری را با دقت کامل گوش بده و صحبت‌های داخل آن را کلمه به کلمه به متن تبدیل کن. هیچ توضیح اضافه‌ای نده."
     
@@ -861,7 +874,7 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
             save_db(user_credits_db)
         await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
     else: 
-        await send_with_keyboard(client, chat_id, f"❌ تبدیل فایل به متن ناموفق بود.", True)
+        await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
 
 # --- ۵. پردازش تحلیل فایل مستقل ---
@@ -880,6 +893,7 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     
     keys_to_try = GEMINI_KEYS.copy()
     random.shuffle(keys_to_try)
+    keys_to_try = keys_to_try[:20] # انتخاب حداکثر ۲۰ کلید تصادفی
     final_answer = None
     
     async with aiohttp.ClientSession() as session:
@@ -908,7 +922,7 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
             save_db(user_credits_db)
         await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
     else: 
-        await send_with_keyboard(client, chat_id, "❌ پاسخی دریافت نشد.", True)
+        await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
 
 # --- تنظیمات ربات روبیکا ---