Update app.py

app.py

@@ -5,6 +5,11 @@ import gradio as gr
 import json
 import requests
 from PIL import Image
+from flask import request 
+import sqlite3 
+from datetime import datetime, timedelta 
+# Initialize SQLite database 
+
 
 css = """
 .example-image img{
@@ -44,6 +49,40 @@ css = """
 }
 """
 
+# Initialize SQLite database
+conn = sqlite3.connect("ip_requests.db")
+cursor = conn.cursor()
+cursor.execute("""
+    CREATE TABLE IF NOT EXISTS requests (
+        ip_address TEXT PRIMARY KEY,
+        count INTEGER,
+        last_request TIMESTAMP
+    )
+""")
+conn.commit()
+
+def track_requests(ip_address):
+    now = datetime.now()
+    cursor.execute("SELECT count, last_request FROM requests WHERE ip_address=?", (ip_address,))
+    result = cursor.fetchone()
+
+    if result:
+        count, last_request = result
+        last_request = datetime.strptime(last_request, "%Y-%m-%d %H:%M:%S")
+        if now - last_request > timedelta(days=1):
+            count = 0
+    else:
+        count = 0
+
+    count += 1
+    cursor.execute("""
+        INSERT OR REPLACE INTO requests (ip_address, count, last_request)
+        VALUES (?, ?, ?)
+    """, (ip_address, count, now.strftime("%Y-%m-%d %H:%M:%S")))
+    conn.commit()
+    return count
+
+
 screenReplayThreshold = 0.5
 portraitReplaceThreshold = 0.5
 printedCopyThreshold = 0.5
@@ -134,6 +173,14 @@ def check_liveness(frame):
     return [liveness_result, {"status": "error", "result": f"{result.text}"}]
     
 def idcard_recognition(frame1):
+    ip_address = request.remote_addr
+    request_count = track_requests(ip_address)
+
+    if request_count > 3:
+        print("you have exceeded the daily limit of 5 requests")
+        return "You have exceeded the daily limit of 5 requests."
+        
+        
   url = "https://api.cortex.cerebrium.ai/v4/p-4f1d877e/my-first-project/process-image/"
   # url = "https://edreesi-ocr-api.hf.space/process-image/"