Update app.py

app.py

@@ -1,96 +1,59 @@
 import os
-from fastapi import FastAPI, Request, HTTPException
+import json
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
-from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 import google.generativeai as genai
 from dotenv import load_dotenv
 
-# بارگذاری متغیرهای محیطی
 load_dotenv()
 
-# تنظیم API Key
+# دریافت کلید از تنظیمات اسپیس
 api_key = os.getenv("GEMINI_API_KEY")
 
-# ساخت اپلیکیشن FastAPI
 app = FastAPI()
 
-# مدل داده‌ای که از سمت HTML دریافت می‌شود
-class UserRequest(BaseModel):
-    description: str
+class IdeaRequest(BaseModel):
+    idea: str
 
-# تنظیمات جمینای
 if api_key:
     genai.configure(api_key=api_key)
-    generation_config = {
-        "temperature": 0.7,
-        "top_p": 0.95,
-        "top_k": 40,
-        "max_output_tokens": 8192,
-    }
-    model = genai.GenerativeModel(
-        model_name="gemini-2.5-flash",
-        generation_config=generation_config,
-    )
+    model = genai.GenerativeModel('gemini-2.5-flash')
 
-# 1. مسیر اصلی که فایل HTML را نمایش می‌دهد
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
-    try:
-        with open("index.html", "r", encoding="utf-8") as f:
-            return f.read()
-    except FileNotFoundError:
-        return "<h1>Error: index.html not found!</h1>"
+    with open("index.html", "r", encoding="utf-8") as f:
+        return f.read()
 
-# 2. مسیر API برای پردازش درخواست
-@app.post("/api/generate")
-async def generate_lyrics(request: UserRequest):
+@app.post("/api/refine")
+async def refine_text(request: IdeaRequest):
     if not api_key:
-        raise HTTPException(status_code=500, detail="Gemini API Key not set in Space secrets.")
+        raise HTTPException(status_code=500, detail="کلید API جمینای تنظیم نشده است.")
 
-    user_input = request.description
+    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.
 
-    # پرامپت سیستم دقیقاً مشابه قبل برای اعراب‌گذاری
-    system_instruction = """
-    You are a professional music producer and songwriter assistant.
-    The user will provide a description or a topic for a song.
-    
-    You must generate two things:
-    1. A detailed 'Audio Prompt' in English describing the style, instruments, mood, and bpm.
-    2. The 'Lyrics' for the song based on the user's topic.
+    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.
 
-    CRITICAL RULE FOR PERSIAN LYRICS:
-    - If the lyrics are in PERSIAN (Farsi), you MUST apply full diacritics (Vowel signs: Fathe, Zamma, Kasra, Sukun) to EVERY SINGLE WORD.
-    - Example: Instead of "دلم گرفت", write "دِلَم گِرِفت".
-    - This is strictly required for the TTS model.
+    Output strictly in JSON format:
+    {{
+        "music_prompt": "string",
+        "lyrics": "string"
+    }}
 
-    OUTPUT FORMAT (Return JSON structure only inside text, no markdown code blocks):
-    {
-        "music_prompt": "YOUR ENGLISH PROMPT HERE",
-        "lyrics": "YOUR LYRICS HERE"
-    }
+    User Input: {request.idea}
     """
-
-    full_prompt = f"{system_instruction}\n\nUser Input: {user_input}\nResponse must be valid JSON."
-
+    
     try:
-        response = model.generate_content(full_prompt)
-        text_response = response.text
-        
-        # تمیز کردن خروجی برای تبدیل به JSON (حذف احتمالی ```json)
-        clean_text = text_response.replace("```json", "").replace("```", "").strip()
-        
-        # تلاش برای پارس کردن جیسون، اگر جمینای متن خالی فرستاد مدیریت شود
-        import json
-        try:
-            data = json.loads(clean_text)
-            return JSONResponse(content=data)
-        except json.JSONDecodeError:
-            # اگر فرمت جیسون نبود، متن خام را برمی‌گردانیم
-            return JSONResponse(content={
-                "music_prompt": "Error parsing JSON from AI",
-                "lyrics": clean_text
-            })
-
+        result = model.generate_content(prompt)
+        # تمیزکاری جیسون
+        clean_json = result.text.replace("```json", "").replace("```", "").strip()
+        data = json.loads(clean_json)
+        return JSONResponse(content=data)
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return JSONResponse(content={"error": str(e)}, status_code=500)