Update app.py

app.py

@@ -8,7 +8,6 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-# دریافت کلید از تنظیمات اسپیس
 api_key = os.getenv("GEMINI_API_KEY")
 
 app = FastAPI()
@@ -18,32 +17,49 @@ class IdeaRequest(BaseModel):
 
 if api_key:
     genai.configure(api_key=api_key)
+    # استفاده از مدل فلش برای سرعت و دقت بالا
     model = genai.GenerativeModel('gemini-2.5-flash')
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
-    with open("index.html", "r", encoding="utf-8") as f:
-        return f.read()
+    try:
+        with open("index.html", "r", encoding="utf-8") as f:
+            return f.read()
+    except FileNotFoundError:
+        return "<h1>Error: index.html not found</h1>"
 
 @app.post("/api/refine")
 async def refine_text(request: IdeaRequest):
     if not api_key:
-        raise HTTPException(status_code=500, detail="کلید API جمینای تنظیم نشده است.")
+        raise HTTPException(status_code=500, detail="کلید API تنظیم نشده است.")
 
+    # پرامپت دقیق برای تولید آهنگ طولانی و ساختار یافته
     prompt = f"""
-    You are an expert music producer AI. Convert the user's description into two parts:
-    1. A detailed English Audio Prompt (style, bpm, instruments, mood).
-    2. Lyrics based on the topic.
+    You are a professional songwriter and music producer.
+    Task: Convert the user's input into a COMPLETE song structure and a music generation prompt.
 
-    CRITICAL RULE FOR LYRICS:
-    - If the user wants Persian/Farsi lyrics, you MUST apply FULL DIACRITICS (Fathe, Kasra, Zamma, Sukun) to EVERY single word.
-    - Example: write "دِلَم گِرِفت" instead of "دلم گرفت". This is vital for the TTS engine.
-    - If the user provides lyrics, just add diacritics to them.
+    1. **Music Prompt (English):** Describe the mood, instruments, BPM, and style suitable for an AI music generator.
+    
+    2. **Lyrics (Persian/Farsi):**
+       - The song MUST be long enough for **3 to 5 minutes**.
+       - Do **NOT** use diacritics (اعراب نگذار). Write standard Persian.
+       - You MUST use the following structure tags exactly:
+         [Intro]
+         [Verse 1]
+         [Pre-Chorus]
+         [Chorus]
+         [Verse 2]
+         [Chorus]
+         [Bridge]
+         [Guitar Solo] (or Instrumental Break)
+         [Chorus]
+         [Outro]
+       - Ensure rhymes and rhythm are professional.
 
     Output strictly in JSON format:
     {{
-        "music_prompt": "string",
-        "lyrics": "string"
+        "music_prompt": "YOUR ENGLISH PROMPT HERE",
+        "lyrics": "YOUR FULL LYRICS HERE"
     }}
 
     User Input: {request.idea}
@@ -51,7 +67,6 @@ async def refine_text(request: IdeaRequest):
     
     try:
         result = model.generate_content(prompt)
-        # تمیزکاری جیسون
         clean_json = result.text.replace("```json", "").replace("```", "").strip()
         data = json.loads(clean_json)
         return JSONResponse(content=data)