Update app.py

app.py

@@ -1,8 +1,6 @@
 import os
 import json
 import random
-import asyncio
-from concurrent.futures import ThreadPoolExecutor
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
@@ -13,18 +11,17 @@ 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("هشدار: هیچ کلید جیمینای یافت نشد.")
-
-# 2. ایجاد ThreadPool برای اجرای موازی درخواست‌ها (دست‌های زیاد)
-executor = ThreadPoolExecutor(max_workers=10)
+    print("هیچ کلید جیمینای شناسایی نشد. لطفاً متغیر ALL_GEMINI_API_KEYS را تنظیم کنید.")
 
 class IdeaRequest(BaseModel):
     idea: str
@@ -37,85 +34,76 @@ async def read_root():
     except FileNotFoundError:
         return "<h1>Error: index.html not found</h1>"
 
-def sync_generate_content(prompt):
-    """
-    این تابع همگام است و در یک ترد جداگانه اجرا می‌شود تا سرور اصلی را قفل نکند.
-    تلاش می‌کند تا 100 بار با کلیدهای مختلف محتوا تولید کند.
-    """
-    max_retries = 100  # درخواست شما: 100 بار تلاش تصادفی
+@app.post("/api/refine")
+async def refine_text(request: IdeaRequest):
+    if not GEMINI_KEYS:
+        raise HTTPException(status_code=500, detail="هیچ کلید API تنظیم نشده است.")
+
+    # ساخت یک لیست کپی و مخلوط کردن آن برای انتخاب تصادفی
+    shuffled_keys = list(GEMINI_KEYS)
+    random.shuffle(shuffled_keys)
+
+    last_error = None
     
-    for _ in range(max_retries):
+    # حلقه برای امتحان کردن کلیدها یکی پس از دیگری
+    for api_key in shuffled_keys:
         try:
-            # انتخاب کاملاً تصادفی یک کلید در هر بار تلاش
-            current_key = random.choice(GEMINI_KEYS)
+            # تنظیم کلید برای این تلاش
+            genai.configure(api_key=api_key)
             
-            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.
+
+            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.).
+            - **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 (اعراب نگذار).
+            - You MUST use the following structure tags exactly:
+                [Verse 1]
+                [Pre-Chorus]
+                [Chorus]
+                [Verse 2]
+                [Chorus]
+                [Bridge]
+                [Chorus]
+                [Outro]
             
-            # تولید محتوا (این عملیات ممکن است طول بکشد)
-            response = model.generate_content(prompt)
+            **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.
+
+            Output strictly in JSON format:
+            {{
+                "music_prompt": "YOUR ENGLISH PROMPT HERE",
+                "lyrics": "YOUR FULL LYRICS HERE (ONLY SUNG WORDS IN DETECTED LANGUAGE)"
+            }}
+
+            User Input: {request.idea}
+            """
             
-            clean_json = response.text.replace("```json", "").replace("```", "").strip()
+            # تلاش برای تولید محتوا
+            result = model.generate_content(prompt)
+            clean_json = result.text.replace("```json", "").replace("```", "").strip()
             data = json.loads(clean_json)
             
-            return data  # موفقیت! خروجی را برگردان
+            # اگر موفقیت‌آمیز بود، خروجی را برگردان و از حلقه خارج شو
+            return JSONResponse(content=data)
 
         except Exception as e:
-            # خطا را نادیده بگیر و دوباره تلاش کن (حلقه بعدی)
-            # print(f"Key failed: {str(e)[:50]}...")
+            # اگر کلید فعلی خطا داد (مثلاً اعتبار تمام شده)، لاگ کن و برو سراغ کلید بعدی
+            print(f"Error with a key: {e}. Trying next key...")
+            last_error = e
             continue
-            
-    # اگر بعد از 100 تلاش نشد
-    return None
 
-@app.post("/api/refine")
-async def refine_text(request: IdeaRequest):
-    if not GEMINI_KEYS:
-        raise HTTPException(status_code=500, detail="کلید API تنظیم نشده است.")
-
-    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.
-
-    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".
-    - **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 (اعراب نگذار).
-    - You MUST use the following structure tags exactly:
-        [Verse 1]
-        [Pre-Chorus]
-        [Chorus]
-        [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 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"
-    }}
-
-    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": f"All keys failed. Last error: {str(last_error)}"}, status_code=500)