Update app.py

app.py

@@ -3,7 +3,8 @@ import json
 import random
 import asyncio
 import re
-import fcntl  # برای قفل کردن فایل و هماهنگی بین Worker ها
+import fcntl  # کتابخانه قفل کردن فایل برای دقت ۱۰۰ درصد
+import time
 from datetime import datetime
 from concurrent.futures import ThreadPoolExecutor
 from fastapi import FastAPI, HTTPException, Request
@@ -69,24 +70,36 @@ def sync_generate_content(prompt):
 
 def check_and_increment_usage(ip: str, fingerprint: str):
     """
-    این تابع با استفاده از قفل فایل، دسترسی همزمان چندین Worker را مدیریت می‌کند
-    تا محدودیت دقیقاً اعمال شود.
+    این تابع با استفاده از مکانیزم قفل (Lock) فایل، دسترسی همزمان را مدیریت می‌کند.
+    این روش دقیق‌ترین راه برای اعمال محدودیت در محیط‌های چند-رشته‌ای است.
     """
     today = datetime.now().strftime("%Y-%m-%d")
     
-    # اطمینان از وجود فایل
+    # تلاش برای ایجاد فایل اگر وجود ندارد (با مدیریت خطا برای جلوگیری از تداخل)
     if not os.path.exists(USAGE_DB_FILE):
-        with open(USAGE_DB_FILE, 'w') as f:
-            json.dump({}, f)
+        try:
+            with open(USAGE_DB_FILE, 'w') as f:
+                json.dump({}, f)
+        except OSError:
+            pass # ممکن است یک پردازش دیگر همزمان فایل را ساخته باشد
 
     # باز کردن فایل با قابلیت خواندن و نوشتن
-    with open(USAGE_DB_FILE, 'r+') as f:
+    try:
+        f = open(USAGE_DB_FILE, 'r+')
+    except FileNotFoundError:
+        # اگر فایل در لحظه باز کردن حذف شده باشد دوباره می‌سازیم
+        f = open(USAGE_DB_FILE, 'w+')
+        json.dump({}, f)
+        f.seek(0)
+
+    with f:
         try:
-            # اعمال قفل انحصاری (فقط یک Worker در لحظه می‌تواند فایل را بخواند/بنویسد)
+            # اعمال قفل انحصاری: بقیه درخواست‌ها منتظر می‌مانند تا این یکی تمام شود
             fcntl.flock(f, fcntl.LOCK_EX)
             
             try:
-                data = json.load(f)
+                content = f.read()
+                data = json.loads(content) if content else {}
             except json.JSONDecodeError:
                 data = {}
 
@@ -94,23 +107,24 @@ def check_and_increment_usage(ip: str, fingerprint: str):
             ip_key = f"ip:{ip}"
             fp_key = f"fp:{fingerprint}"
             
-            # خواندن وضعیت فعلی IP
+            # --- منطق بررسی 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}
             
-            # خواندن وضعیت فعلی Fingerprint
+            # --- منطق بررسی اثر انگشت ---
             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
             
@@ -118,10 +132,11 @@ 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() # اطمینان از نوشته شدن روی دیسک
             
             return True
 
@@ -139,8 +154,14 @@ async def refine_text(request: IdeaRequest, req: Request):
         client_ip = req.client.host
         client_fp = request.fingerprint
         
-        # فراخوانی تابع امن با قفل فایل
-        allowed = check_and_increment_usage(client_ip, client_fp)
+        # فراخوانی تابع ��ا قفل فایل
+        try:
+            allowed = check_and_increment_usage(client_ip, client_fp)
+        except Exception as e:
+            print(f"Error checking limit: {e}")
+            # در صورت بروز خطای سیستمی، برای اطمینان کاربر را بلاک می‌کنیم یا باز می‌گذاریم؟
+            # اینجا برای امنیت بیشتر، اجازه نمی‌دهیم.
+            return JSONResponse(content={"error": "System Error checking limits"}, status_code=500)
         
         if not allowed:
             # کد 429 به معنی Too Many Requests