Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 import random
 import asyncio
 import re
-import fcntl  # کتابخانه قفل کردن فایل برای دقت ۱۰۰ درصد
+import fcntl
 import time
 from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor
@@ -19,7 +19,7 @@ app = FastAPI()
 
 # --- تنظیمات محدودیت ---
 MAX_FREE_DAILY_REQUESTS = 5
-USAGE_DB_FILE = "usage_data.json"  # فایل ذخیره سوابق
+USAGE_DB_FILE = "usage_data.json"
 
 # --- بخش بارگذاری کلیدها ---
 all_keys_str = os.getenv("ALL_GEMINI_API_KEYS", "")
@@ -42,6 +42,7 @@ class IdeaRequest(BaseModel):
     idea: str
     fingerprint: str
     is_premium: bool = False
+    is_instrumental: bool = False  # فیلد جدید برای حالت بی‌کلام
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
@@ -52,7 +53,7 @@ async def read_root():
         return "<h1>Error: index.html not found</h1>"
 
 def sync_generate_content(prompt):
-    max_retries = 1000
+    max_retries = 3
     if not GEMINI_KEYS:
         return None
 
@@ -69,62 +70,46 @@ def sync_generate_content(prompt):
     return None
 
 def check_and_increment_usage(ip: str, fingerprint: str):
-    """
-    این تابع با استفاده از مکانیزم قفل (Lock) فایل، دسترسی همزمان را مدیریت می‌کند.
-    این روش دقیق‌ترین راه برای اعمال محدودیت در محیط‌های چند-رشته‌ای است.
-    """
     today = datetime.now().strftime("%Y-%m-%d")
-    
-    # تلاش برای ایجاد فایل اگر وجود ندارد (با مدیریت خطا برای جلوگیری از تداخل)
     if not os.path.exists(USAGE_DB_FILE):
         try:
             with open(USAGE_DB_FILE, 'w') as f:
                 json.dump({}, f)
         except OSError:
-            pass # ممکن است یک پردازش دیگر همزمان فایل را ساخته باشد
+            pass
 
-    # باز کردن فایل با قابلیت خواندن و نوشتن
     try:
         f = open(USAGE_DB_FILE, 'r+')
     except FileNotFoundError:
-        # اگر فایل در لحظه باز کردن حذف شده باشد دوباره می‌سازیم
         f = open(USAGE_DB_FILE, 'w+')
         json.dump({}, f)
         f.seek(0)
 
     with f:
         try:
-            # اعمال قفل انحصاری: بقیه درخواست‌ها منتظر می‌مانند تا این یکی تمام شود
             fcntl.flock(f, fcntl.LOCK_EX)
-            
             try:
                 content = f.read()
                 data = json.loads(content) if content else {}
             except json.JSONDecodeError:
                 data = {}
 
-            # آماده‌سازی کلیدها
             ip_key = f"ip:{ip}"
             fp_key = f"fp:{fingerprint}"
             
-            # --- منطق بررسی IP ---
             ip_record = data.get(ip_key, {"date": today, "count": 0})
-            if ip_record["date"] != today: # اگر روز جدید است، ریست کن
+            if ip_record["date"] != today:
                 ip_record = {"date": today, "count": 0}
             
-            # --- منطق بررسی اثر انگشت ---
             fp_record = data.get(fp_key, {"date": today, "count": 0})
-            if fp_record["date"] != today: # اگر روز جدید است، ریست کن
+            if fp_record["date"] != today:
                 fp_record = {"date": today, "count": 0}
 
-            # --- شرط مسدودسازی ---
-            # اگر هر کدام از سقف رد شده باشند، اجازه نده
             if ip_record["count"] >= MAX_FREE_DAILY_REQUESTS:
                 return False
             if fingerprint and fp_record["count"] >= MAX_FREE_DAILY_REQUESTS:
                 return False
 
-            # --- ثبت استفاده ---
             ip_record["count"] += 1
             data[ip_key] = ip_record
             
@@ -132,16 +117,13 @@ def check_and_increment_usage(ip: str, fingerprint: str):
                 fp_record["count"] += 1
                 data[fp_key] = fp_record
 
-            # بازنویسی فایل با داده‌های جدید
             f.seek(0)
             f.truncate()
             json.dump(data, f)
-            f.flush() # اطمینان از نوشته شدن روی دیسک
-            
+            f.flush()
             return True
 
         finally:
-            # آزاد کردن قفل
             fcntl.flock(f, fcntl.LOCK_UN)
 
 @app.post("/api/refine")
@@ -149,54 +131,47 @@ async def refine_text(request: IdeaRequest, req: Request):
     if not GEMINI_KEYS:
         raise HTTPException(status_code=500, detail="کلید API تنظیم نشده است.")
 
-    # --- بررسی دقیق محدودیت روزانه ---
     if not request.is_premium:
         client_ip = req.client.host
         client_fp = request.fingerprint
-        
-        # فراخوانی تابع با قفل فایل
         try:
             allowed = check_and_increment_usage(client_ip, client_fp)
-        except Exception as e:
-            print(f"Error checking limit: {e}")
-            # در صورت بروز خطای سیستمی، برای اطمینان کاربر را بلاک می‌کنیم یا باز می‌گذاریم؟
-            # اینجا برای امنیت بیشتر، اجازه نمی‌دهیم.
+        except Exception:
             return JSONResponse(content={"error": "System Error checking limits"}, status_code=500)
         
         if not allowed:
-            # کد 429 به معنی Too Many Requests
             return JSONResponse(content={"error": "LIMIT_REACHED"}, status_code=429)
-    # ---------------------------
+
+    # لاجیک جدید برای حالت بی‌کلام
+    instruction = ""
+    if request.is_instrumental:
+        instruction = """
+        **INSTRUMENTAL MODE ACTIVE:**
+        - The user wants an instrumental song (NO LYRICS).
+        - **lyrics field MUST be an empty string ""**.
+        - Focus entirely on the `music_prompt` to describe the instruments and mood vividly.
+        """
+    else:
+        instruction = """
+        - **Lyrics (Match User's Language):**
+        - Detect the language of the "User Input".
+        - Write the lyrics entirely in that detected language.
+        - Structure tags: [Verse 1], [Chorus], etc.
+        - "lyrics" field must contain ONLY words to be sung.
+        """
 
     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.
+    
+    {instruction}
 
     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**.
-    - 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 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"
+        "lyrics": "YOUR FULL LYRICS HERE (OR EMPTY STRING IF INSTRUMENTAL)"
     }}
 
     User Input: {request.idea}