Update app.py

app.py

@@ -23,13 +23,53 @@ JS_FUNC = os.getenv("JS_FUNC")
 def base64_to_image(base64_str):
     return base64.b64decode(base64_str + '=' * (-len(base64_str) % 4))
 
-def search_face(file, token=None):
+user_attempts = {}
+def clear_old_entries():
+    today = datetime.now().date()
+    # Create a list of keys to remove
+    keys_to_remove = [key for key, value in user_attempts.items() if value != today]
+    # Remove old entries
+    for key in keys_to_remove:
+        del user_attempts[key]
+
+def if_limited(request):
+    clear_old_entries()
+    user_ip = None
+    if request.headers.get("x-forwarded-for"):
+        user_ip = request.headers["x-forwarded-for"].split(",")[0]  # First IP in the list
+
+    cookie_value = request.headers.get("cookie", "")
+    if "user_id=" in cookie_value:
+        user_id = cookie_value.split("user_id=")[1].split(";")[0]
+    else:
+        user_id = None
+    print("##### Coming", user_id, user_ip)
+    # Get today's date
+    today = datetime.now().date()
+
+    # Check if the user has already tried today (by IP or cookie)
+    for key, value in user_attempts.items():
+        if (key == user_ip or key == user_id) and value == today:
+            return True
+
+    # Record the attempt (store both hashed IP and hashed cookie)
+    if user_ip:
+        user_attempts[user_ip] = today
+    if user_id:
+        user_attempts[user_id] = today
+    return False
+
+def search_face(file, token, request: gr.Request):
     try:
         file_1 = handle_file(file)
     except Exception as e:
         gr.Info("Please upload an image file.")
         return []
 
+    if token == "" and if_limited(request):
+        gr.Info("Your free credit for today has been used. Buy Premium Token.", duration=12)
+        return []
+
     result_text = backend.predict(
             file=file_1,
             token=token, 
@@ -46,7 +86,7 @@ def search_face(file, token=None):
         outarray.append((image, item['url']))
     
     if result['status'] == 201:
-        gr.Info(STATUS_MESSAGES[result['status']], duration=11)
+        gr.Info(STATUS_MESSAGES[result['status']], duration=12)
     return outarray
 
 def export_images(items):