Update app.py

app.py

@@ -1,11 +1,31 @@
 import os
 import json
-from fastapi import FastAPI
+import random
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
+import google.generativeai as genai
+from dotenv import load_dotenv
+
+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)} کلید جیمینای بارگذاری شد.")
+else:
+    print("هشدار: هیچ کلید جیمینای یافت نشد.")
+
+# 2. ایجاد ThreadPool برای اجرای موازی درخواست‌ها (دست‌های زیاد)
+executor = ThreadPoolExecutor(max_workers=10)
+
 class IdeaRequest(BaseModel):
     idea: str
 
@@ -17,29 +37,85 @@ async def read_root():
     except FileNotFoundError:
         return "<h1>Error: index.html not found</h1>"
 
+def sync_generate_content(prompt):
+    """
+    این تابع همگام است و در یک ترد جداگانه اجرا می‌شود تا سرور اصلی را قفل نکند.
+    تلاش می‌کند تا 100 بار با کلیدهای مختلف محتوا تولید کند.
+    """
+    max_retries = 100  # درخواست شما: 100 بار تلاش تصادفی
+    
+    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')
+            
+            # تولید محتوا (این عملیات ممکن است طول بکشد)
+            response = model.generate_content(prompt)
+            
+            clean_json = response.text.replace("```json", "").replace("```", "").strip()
+            data = json.loads(clean_json)
+            
+            return data  # موفقیت! خروجی را برگردان
+
+        except Exception as e:
+            # خطا را نادیده بگیر و دوباره تلاش کن (حلقه بعدی)
+            # print(f"Key failed: {str(e)[:50]}...")
+            continue
+            
+    # اگر بعد از 100 تلاش نشد
+    return None
+
 @app.post("/api/refine")
 async def refine_text(request: IdeaRequest):
-    # چون جیمینای نداریم، فرض می‌کنیم ورودی کاربر همان متن آهنگ است
-    user_input = request.idea.strip()
+    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.
     
-    # تشخیص ساده برای اینکه آیا ورودی فارسی است یا انگلیسی (برای جهت متن)
-    is_persian = any("\u0600" <= c <= "\u06FF" for c in user_input)
+    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]
     
-    # ساختاردهی دستی (چون هوش مصنوعی نداریم که ورس و کورس بسازد)
-    # کل متن کاربر را در یک بخش [Verse] قرار می‌دهیم
-    lyrics = f"""[Verse]
-{user_input}
+    **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}
     """
     
-    # پرامپت موسیقی پیش‌فرض (چون نمی‌توانیم سبک را از متن استخراج کنیم)
-    # اگر متن فارسی بود، سبک پاپ ایرانی در نظر می‌گیریم
-    if is_persian:
-        music_prompt = "Persian Pop song, emotional, clear vocals, high quality, 120 bpm"
+    # 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:
-        music_prompt = "Pop song, catchy melody, clear vocals, high quality, 120 bpm"
-
-    # خروجی استاندارد که فرانت‌اند انتظار دارد
-    return JSONResponse(content={
-        "music_prompt": music_prompt,
-        "lyrics": lyrics
-    })
+        # اگر بعد از 100 تلاش هیچ کلیدی کار نکرد
+        return JSONResponse(content={"error": "Server busy (100 retries failed). Please try again later."}, status_code=503)