Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import random
+import asyncio
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
@@ -11,17 +12,15 @@ 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()]
 
-# نمایش پیام در لاگ هنگام شروع برنامه
 if GEMINI_KEYS:
-    print(f"تعداد {len(GEMINI_KEYS)} کلید جیمینای شناسایی شد")
+    print(f"تعداد {len(GEMINI_KEYS)} کلید جیمینای بارگذاری شد.")
 else:
-    print("هیچ کلید جیمینای شناسایی نشد. لطفاً متغیر ALL_GEMINI_API_KEYS را تنظیم کنید.")
+    print("هشدار: هیچ کلید جیمینای یافت نشد.")
 
 class IdeaRequest(BaseModel):
     idea: str
@@ -36,25 +35,26 @@ async def read_root():
 
 @app.post("/api/refine")
 async def refine_text(request: IdeaRequest):
+    # اگر کلاً کلیدی وجود ندارد، خطا بده
     if not GEMINI_KEYS:
-        raise HTTPException(status_code=500, detail="هیچ کلید API تنظیم نشده است.")
+        raise HTTPException(status_code=500, detail="کلید API تنظیم نشده است.")
 
-    # ساخت یک لیست کپی و مخلوط کردن آن برای انتخاب تصادفی
-    shuffled_keys = list(GEMINI_KEYS)
-    random.shuffle(shuffled_keys)
-
-    last_error = None
+    # 2. تنظیمات تلاش مجدد
+    # ما حداکثر 60 بار تلاش می‌کنیم (حدود 1 دقیقه در بدترین حالت)
+    # این کار باعث می‌شود اگر اکثر کلیدها 429 بودند، سرور بی‌نهایت درگیر نماند
+    max_retries = 60
     
-    # حلقه برای امتحان کردن کلیدها یکی پس از دیگری
-    for api_key in shuffled_keys:
+    last_error = None
+
+    for attempt in range(max_retries):
         try:
-            # تنظیم کلید برای این تلاش
-            genai.configure(api_key=api_key)
+            # 3. انتخاب کاملاً تصادفی یک کلید در هر بار تلاش
+            # کلید از لیست حذف نمی‌شود، فقط خوانده می‌شود
+            current_key = random.choice(GEMINI_KEYS)
             
-            # استفاده از مدل
+            genai.configure(api_key=current_key)
             model = genai.GenerativeModel('gemini-2.5-flash')
 
-            # پرامپت دقیق برای تشخیص زبان و جلوگیری از توضیحات اضافی
             prompt = f"""
             You are a professional songwriter and music producer.
             Task: Convert the user's input into a COMPLETE song structure and a music generation prompt.
@@ -62,7 +62,7 @@ async def refine_text(request: IdeaRequest):
             1. **Music Prompt (English):** Describe the mood, instruments, BPM, and style suitable for an AI music generator.
             
             2. **Lyrics (Match User's Language):**
-            - **Language Detection:** Detect the language of the "User Input" (e.g., English, Persian, Spanish, etc.).
+            - **Language Detection:** Detect the language of the "User Input".
             - **Write the lyrics entirely in that detected language.**
             - The song MUST be long enough for **3 to 5 minutes**.
             - If the language is Persian/Arabic, do **NOT** use diacritics (اعراب نگذار).
@@ -73,37 +73,45 @@ async def refine_text(request: IdeaRequest):
                 [Verse 2]
                 [Chorus]
                 [Bridge]
+                [Guitar Solo]
                 [Chorus]
                 [Outro]
             
             **CRITICAL RULES FOR LYRICS content:**
             - The "lyrics" field must contain **ONLY the words to be sung**.
-            - **ABSOLUTELY NO** descriptive text, stage directions, or instrumental notes inside the lyrics (e.g., DO NOT write "(Instrumental)", "(Fade out)", "(Repeat 2x)", or descriptions of the solo).
-            - For instrumental sections, if there are no words to sing, leave the content under that tag empty or just write the tag itself.
-            - Do NOT write things like "Guitar solo starts here" inside the lyrics value.
+            - **ABSOLUTELY NO** descriptive text inside the lyrics.
+            - For instrumental sections, leave the content empty.
 
             Output strictly in JSON format:
             {{
                 "music_prompt": "YOUR ENGLISH PROMPT HERE",
-                "lyrics": "YOUR FULL LYRICS HERE (ONLY SUNG WORDS IN DETECTED LANGUAGE)"
+                "lyrics": "YOUR FULL LYRICS HERE"
             }}
 
             User Input: {request.idea}
             """
             
-            # تلاش برای تولید محتوا
-            result = model.generate_content(prompt)
+            # 4. استفاده از متد Async (دست‌های زیاد)
+            # این خط کلیدی است: باعث می‌شود سرور منتظر نماند و همزمان درخواست بقیه را بگیرد
+            result = await model.generate_content_async(prompt)
+            
             clean_json = result.text.replace("```json", "").replace("```", "").strip()
             data = json.loads(clean_json)
             
-            # اگر موفقیت‌آمیز بود، خروجی را برگردان و از حلقه خارج شو
+            # اگر موفق شد، نتیجه را برگردان و تمام
             return JSONResponse(content=data)
 
         except Exception as e:
-            # اگر کلید فعلی خطا داد (مثلاً اعتبار تمام شده)، لاگ کن و برو سراغ کلید بعدی
-            print(f"Error with a key: {e}. Trying next key...")
             last_error = e
+            # خطای کوتاه در لاگ (برای جلوگیری از شلوغی)
+            # print(f"Attempt {attempt+1} failed with key ...{current_key[-4:]}: {str(e)[:50]}...")
+            
+            # 5. وقفه کوتاه بسیار مهم
+            # این وقفه (0.05 ثانیه) باعث می‌شود CPU درگیر حلقه‌های تند نشود
+            # و به "دست‌های دیگر" فرصت کار می‌دهد
+            await asyncio.sleep(0.05)
             continue
 
-    # اگر تمام کلیدها تست شدند و همه خطا دادند
-    return JSONResponse(content={"error": f"All keys failed. Last error: {str(last_error)}"}, status_code=500)
+    # اگر بعد از 60 تلاش شانسی موفق نشدیم
+    print("تمام تلاش‌ها ناموفق بود. ترافیک بالا یا اعتبار تمام .")
+    return JSONResponse(content={"error": "Server is busy, please try again."}, status_code=503)