Update app.py

app.py

@@ -3,8 +3,9 @@ import json
 import random
 import asyncio
 import re
+from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
 import google.generativeai as genai
@@ -14,11 +15,15 @@ load_dotenv()
 
 app = FastAPI()
 
+# --- حافظه موقت برای محدودیت روزانه ---
+# ساختار: { "identifier_string": { "date": "YYYY-MM-DD", "count": int } }
+DAILY_LIMITS = {}
+MAX_FREE_DAILY_REQUESTS = 5
+
 # --- بخش اصلاح شده برای بارگذاری تمیز کلیدها ---
 all_keys_str = os.getenv("ALL_GEMINI_API_KEYS", "")
 
 # استفاده از Regex برای جدا کردن کلیدها بر اساس کاما، خط جدید یا فاصله
-# این کار باعث می‌شود اگر کلیدها پشت سر هم با اینتر جدا شده باشند هم درست کار کند
 raw_keys = re.split(r'[,\s\n]+', all_keys_str)
 
 GEMINI_KEYS = []
@@ -42,6 +47,8 @@ executor = ThreadPoolExecutor(max_workers=10)
 
 class IdeaRequest(BaseModel):
     idea: str
+    fingerprint: str  # شناسه مرورگر کاربر
+    is_premium: bool = False # وضعیت اشتراک
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root():
@@ -67,7 +74,7 @@ def sync_generate_content(prompt):
             current_key = random.choice(GEMINI_KEYS)
             
             genai.configure(api_key=current_key)
-            model = genai.GenerativeModel('gemini-2.5-flash') # استفاده از مدل جدیدتر و سریعتر
+            model = genai.GenerativeModel('gemini-2.0-flash') # استفاده از مدل جدیدتر و سریعتر
             
             response = model.generate_content(prompt)
             
@@ -82,11 +89,57 @@ def sync_generate_content(prompt):
             
     return None
 
+def check_and_update_limit(identifier: str):
+    """
+    بررسی محدودیت روزانه برای یک شناسه (آی‌پی یا فینگرپرینت).
+    اگر محدودیت رد شده باشد False برمی‌گرداند.
+    """
+    today = datetime.now().strftime("%Y-%m-%d")
+    
+    if identifier not in DAILY_LIMITS:
+        DAILY_LIMITS[identifier] = {"date": today, "count": 1}
+        return True
+    
+    record = DAILY_LIMITS[identifier]
+    
+    # اگر تاریخ رکورد قدیمی است، ریست کن
+    if record["date"] != today:
+        DAILY_LIMITS[identifier] = {"date": today, "count": 1}
+        return True
+    
+    # بررسی تعداد
+    if record["count"] >= MAX_FREE_DAILY_REQUESTS:
+        return False
+    
+    # افزایش شمارنده
+    DAILY_LIMITS[identifier]["count"] += 1
+    return True
+
 @app.post("/api/refine")
-async def refine_text(request: IdeaRequest):
+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_fingerprint = request.fingerprint
+
+        # بررسی آی‌پی
+        if not check_and_update_limit(client_ip):
+             return JSONResponse(content={"error": "LIMIT_REACHED"}, status_code=429)
+        
+        # بررسی فینگرپرینت (اگر متفاوت از آی‌پی باشد)
+        # نکته: ما هر دو را جداگانه می‌شماریم تا دور زدن سخت‌تر شود
+        # اما چون تابع check_and_update_limit شمارنده را زیاد می‌کند، 
+        # اگر کاربر با یک آی‌پی و فینگرپرینت بیاید، عملاً دوبار چک می‌شود که مشکلی نیست،
+        # ولی برای دقت بیشتر، اگر یکی از این دو بسته شده باشد، کل درخواست رد می‌شود.
+        # اینجا صرفاً اگر لیمیت آی‌پی پر نشده بود، فینگرپرینت را هم چک و آپدیت می‌کنیم.
+        if client_fingerprint and client_fingerprint != client_ip:
+             if not check_and_update_limit(client_fingerprint):
+                 return JSONResponse(content={"error": "LIMIT_REACHED"}, status_code=429)
+    # ---------------------------
+
     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.