Update app.py

app.py

@@ -1,8 +1,41 @@
 import gradio as gr
 import os
+import time
+from collections import defaultdict
 
-
+# Environment variables
 token = os.environ["TOKEN"]
-model=os.environ["MODEL"]
+model = os.environ["MODEL"]
+
+# Dictionary to store request timestamps for rate limiting
+request_log = defaultdict(list)
+
+def rate_limiter(fn):
+    def wrapper(*args, **kwargs):
+        user_ip = "user_ip_placeholder"  # Replace with actual IP retrieval logic
+        current_time = time.time()
+
+        # Keep only recent requests within the last minute
+        request_log[user_ip] = [t for t in request_log[user_ip] if current_time - t < 60]
+
+        if len(request_log[user_ip]) >= 5:  # Limit to 5 requests per minute
+            return "Rate limit exceeded. Please try again later."
+
+        request_log[user_ip].append(current_time)
+        return fn(*args, **kwargs)
+    return wrapper
+
+# Load your model from Gradio Spaces and apply the rate limiter
+@rate_limiter
+def limited_model_function(*args, **kwargs):
+    return demo(*args, **kwargs)
+
 demo = gr.load(model, src="spaces", token=token)
-demo.launch(show_api=False, show_error=False, quiet=True, debug=False)
+
+demo.launch(
+    show_api=False,
+    show_error=False,
+    quiet=True,
+    debug=False,
+    fn=limited_model_function  # Wrap the function with rate limiting
+)