Update app.py

app.py

@@ -15,12 +15,24 @@ request_times = defaultdict(list)
 MAX_REQUESTS = 10  # Maximum requests per time window
 TIME_WINDOW = 60   # Time window in seconds
 
+def get_real_ip(request: gr.Request):
+    """Extract real IP address using x-forwarded-for header or fallback"""
+    if not request:
+        return "unknown"
+    
+    forwarded = request.headers.get("x-forwarded-for")
+    if forwarded:
+        ip = forwarded.split(",")[0].strip()  # First IP in the list is the client's
+    else:
+        ip = request.client.host  # fallback
+    return ip
+
 def check_rate_limit(request: gr.Request):
     """Check if the current request exceeds rate limits"""
     if not request:
         return True, "Rate limit check failed - no request info"
     
-    ip = request.client.host
+    ip = get_real_ip(request)
     now = time.time()
     
     print(f"Ip: {ip}")