IP limit testing (set 1 per day)

app.py

@@ -7,7 +7,6 @@ import traceback
 import datetime
 import threading
 from collections import defaultdict
-from flask import request
 
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -43,11 +42,13 @@ THINKING_PATTERNS = [
 ]
 
 def get_client_ip():
-    """Get the client's IP address from the request"""
+    """Get the client's IP address from the request context"""
     try:
-        # Try to get IP from request context in a Flask environment
-        if request:
-            return request.remote_addr
+        # Try to get IP from Gradio's request context
+        import contextvars
+        request_context = contextvars.ContextVar("request").get()
+        if hasattr(request_context, "client") and request_context.client:
+            return request_context.client.host
     except:
         pass