Upload folder using huggingface_hub

app.py

@@ -8,6 +8,12 @@ import gradio as gr
 import base64
 import time
 from collections import defaultdict
+import fastapi
+from gradio.context import Context
+import logging
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
 
 
 load_dotenv(override=True)
@@ -163,10 +169,27 @@ in which they provide their email, then give a summary of the conversation so fa
         return system_prompt
     
     def chat(self, message, history):
-        # Apply rate limiting with user's IP as the key
-        # In a production app, you would get the real client IP
-        user_id = "default_user"  # Use request.client.host in a proper web framework
-        
+        # Get the client IP from Gradio's request context
+        try:
+            # Try to get the real client IP from request headers
+            request = Context.get_context().request
+            # Check for X-Forwarded-For header (common in reverse proxies like HF Spaces)
+            forwarded_for = request.headers.get("X-Forwarded-For")
+            # Check for Cf-Connecting-IP header (Cloudflare)
+            cloudflare_ip = request.headers.get("Cf-Connecting-IP")
+            
+            if forwarded_for:
+                # X-Forwarded-For contains a comma-separated list of IPs, the first one is the client
+                user_id = forwarded_for.split(",")[0].strip()
+            elif cloudflare_ip:
+                user_id = cloudflare_ip
+            else:
+                # Fall back to direct client address
+                user_id = request.client.host
+        except (AttributeError, RuntimeError, fastapi.exceptions.FastAPIError):
+            # Fallback if we can't get context or if running outside of FastAPI
+            user_id = "default_user"
+        logger.debug(f"User ID: {user_id}")
         if self.rate_limiter.is_rate_limited(user_id):
             return "You're sending messages too quickly. Please wait a moment before sending another message."