Update app.py

app.py

@@ -1,28 +1,11 @@
 import os
 import json
-import random
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
-import google.generativeai as genai
-from dotenv import load_dotenv
-
-load_dotenv()
 
 app = FastAPI()
 
-# بارگذاری و پردازش کلیدها از متغیر محیطی
-# کلیدها باید با کاما (,) جدا شده باشند
-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("هیچ کلید جیمینای شناسایی نشد. لطفاً متغیر ALL_GEMINI_API_KEYS را تنظیم کنید.")
-
 class IdeaRequest(BaseModel):
     idea: str
 
@@ -36,74 +19,27 @@ 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 تنظیم نشده است.")
-
-    # ساخت یک لیست کپی و مخلوط کردن آن برای انتخاب تصادفی
-    shuffled_keys = list(GEMINI_KEYS)
-    random.shuffle(shuffled_keys)
-
-    last_error = None
+    # چون جیمینای نداریم، فرض می‌کنیم ورودی کاربر همان متن آهنگ است
+    user_input = request.idea.strip()
     
-    # حلقه برای امتحان کردن کلیدها یکی پس از دیگری
-    for api_key in shuffled_keys:
-        try:
-            # تنظیم کلید برای این تلاش
-            genai.configure(api_key=api_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]
-            
-            **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}
-            """
-            
-            # تلاش برای تولید محتوا
-            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:
-            # اگر کلید فعلی خطا داد (مثلاً اعتبار تمام شده)، لاگ کن و برو سراغ کلید بعدی
-            print(f"Error with a key: {e}. Trying next key...")
-            last_error = e
-            continue
-
-    # اگر تمام کلیدها تست شدند و همه خطا دادند
-    return JSONResponse(content={"error": f"All keys failed. Last error: {str(last_error)}"}, status_code=500)
+    # تشخیص ساده برای اینکه آیا ورودی فارسی است یا انگلیسی (برای جهت متن)
+    is_persian = any("\u0600" <= c <= "\u06FF" for c in user_input)
+    
+    # ساختاردهی دستی (چون هوش مصنوعی نداریم که ورس و کورس بسازد)
+    # کل متن کاربر را در یک بخش [Verse] قرار می‌دهیم
+    lyrics = f"""[Verse]
+{user_input}
+    """
+    
+    # پرامپت موسیقی پیش‌فرض (چون نمی‌توانیم سبک را از متن استخراج کنیم)
+    # اگر متن فارسی بود، سبک پاپ ایرانی در نظر می‌گیریم
+    if is_persian:
+        music_prompt = "Persian Pop song, emotional, clear vocals, high quality, 120 bpm"
+    else:
+        music_prompt = "Pop song, catchy melody, clear vocals, high quality, 120 bpm"
+
+    # خروجی استاندارد که فرانت‌اند انتظار دارد
+    return JSONResponse(content={
+        "music_prompt": music_prompt,
+        "lyrics": lyrics
+    })