Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import json
 import random
 import asyncio
+import re
 from concurrent.futures import ThreadPoolExecutor
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
@@ -13,17 +14,30 @@ load_dotenv()
 
 app = FastAPI()
 
-# 1. بارگذاری تمام کلیدها در حافظه
+# --- بخش اصلاح شده برای بارگذاری تمیز کلیدها ---
 all_keys_str = os.getenv("ALL_GEMINI_API_KEYS", "")
-# تمیز کردن لیست کلیدها
-GEMINI_KEYS = [k.strip() for k in all_keys_str.split(",") if k.strip()]
+
+# استفاده از Regex برای جدا کردن کلیدها بر اساس کاما، خط جدید یا فاصله
+# این کار باعث می‌شود اگر کلیدها پشت سر هم با اینتر جدا شده باشند هم درست کار کند
+raw_keys = re.split(r'[,\s\n]+', all_keys_str)
+
+GEMINI_KEYS = []
+for k in raw_keys:
+    # حذف کوتیشن‌های احتمالی (' یا ") و کاراکترهای کنترلی مثل \r
+    clean_key = k.strip().replace('"', '').replace("'", "").replace("\r", "").replace("\n", "")
+    
+    # فیلتر کردن کلیدهای خراب (کلیدهای جمینای معمولاً طولانی هستند)
+    if len(clean_key) > 30:
+        GEMINI_KEYS.append(clean_key)
 
 if GEMINI_KEYS:
-    print(f"تعداد {len(GEMINI_KEYS)} کلید جیمینای بارگذاری شد.")
+    print(f"تعداد {len(GEMINI_KEYS)} کلید سالم و تمیز بارگذاری شد.")
+    # نمایش ۵ کاراکتر اول یکی از کلیدها جهت اطمینان در لاگ
+    print(f"Sample loaded key start: {GEMINI_KEYS[0][:5]}...")
 else:
-    print("هشدار: هیچ کلید جیمینای یافت نشد.")
+    print("هشدار جدی: هیچ کلید معتبری یافت نشد. لطفاً متغیر ALL_GEMINI_API_KEYS را بررسی کنید.")
 
-# 2. ایجاد ThreadPool برای اجرای موازی درخواست‌ها (دست‌های زیاد)
+# تنظیم ThreadPool برای درخواست‌های موازی
 executor = ThreadPoolExecutor(max_workers=10)
 
 class IdeaRequest(BaseModel):
@@ -39,33 +53,33 @@ async def read_root():
 
 def sync_generate_content(prompt):
     """
-    این تابع همگام است و در یک ترد جداگانه اجرا می‌شود تا سرور اصلی را قفل نکند.
-    تلاش می‌کند تا 100 بار با کلیدهای مختلف محتوا تولید کند.
+    این تابع همگام است و تلاش می‌کند با کلیدهای مختلف محتوا تولید کند.
     """
-    max_retries = 100  # درخواست شما: 100 بار تلاش تصادفی
+    max_retries = 100
     
+    # اگر کلیدی موجود نیست، فوراً خارج شو
+    if not GEMINI_KEYS:
+        print("No keys available to use.")
+        return None
+
     for _ in range(max_retries):
         try:
-            # انتخاب کاملاً تصادفی یک کلید در هر بار تلاش
             current_key = random.choice(GEMINI_KEYS)
             
             genai.configure(api_key=current_key)
-            model = genai.GenerativeModel('gemini-2.5-flash')
+            model = genai.GenerativeModel('gemini-2.5-flash') # استفاده از مدل جدیدتر و سریعتر
             
-            # تولید محتوا (این عملیات ممکن است طول بکشد)
             response = model.generate_content(prompt)
             
             clean_json = response.text.replace("```json", "").replace("```", "").strip()
             data = json.loads(clean_json)
             
-            return data  # موفقیت! خروجی را برگردان
+            return data
 
         except Exception as e:
-            # خطا را نادیده بگیر و دوباره تلاش کن (حلقه بعدی)
-            # print(f"Key failed: {str(e)[:50]}...")
+            # خطا را لاگ نمی‌کنیم تا لاگ‌ها شلوغ نشود، فقط ریترای می‌کنیم
             continue
             
-    # اگر بعد از 100 تلاش نشد
     return None
 
 @app.post("/api/refine")
@@ -109,13 +123,10 @@ async def refine_text(request: IdeaRequest):
     User Input: {request.idea}
     """
     
-    # 3. اجرای درخواست سنگین در پس‌زمینه (Non-blocking)
-    # این باعث می‌شود سرور اصلی آزاد باشد و بتواند درخواست‌های جدید بگیرد
     loop = asyncio.get_running_loop()
     result = await loop.run_in_executor(executor, sync_generate_content, prompt)
     
     if result:
         return JSONResponse(content=result)
     else:
-        # اگر بعد از 100 تلاش هیچ کلیدی کار نکرد
-        return JSONResponse(content={"error": "Server busy (100 retries failed). Please try again later."}, status_code=503)
+        return JSONResponse(content={"error": "Server busy (All keys limit reached or invalid). Please try again later."}, status_code=503)