Upload folder using huggingface_hub

app.py

@@ -6,10 +6,35 @@ import requests
 from pypdf import PdfReader
 import gradio as gr
 import base64
+import time
+from collections import defaultdict
 
 
 load_dotenv(override=True)
 
+class RateLimiter:
+    def __init__(self, max_requests=5, time_window=5):
+        # max_requests per time_window seconds
+        self.max_requests = max_requests
+        self.time_window = time_window  # in seconds
+        self.request_history = defaultdict(list)
+        
+    def is_rate_limited(self, user_id):
+        current_time = time.time()
+        # Remove old requests
+        self.request_history[user_id] = [
+            timestamp for timestamp in self.request_history[user_id]
+            if current_time - timestamp < self.time_window
+        ]
+        
+        # Check if user has exceeded the limit
+        if len(self.request_history[user_id]) >= self.max_requests:
+            return True
+        
+        # Add current request
+        self.request_history[user_id].append(current_time)
+        return False
+
 def push(text):
     requests.post(
         "https://api.pushover.net/1/messages.json",
@@ -100,6 +125,7 @@ class Me:
     def __init__(self):
         self.openai = OpenAI()
         self.name = "Sagarnil Das"
+        self.rate_limiter = RateLimiter(max_requests=5, time_window=60)  # 5 messages per minute
         reader = PdfReader("me/linkedin.pdf")
         self.linkedin = ""
         for page in reader.pages:
@@ -137,6 +163,13 @@ 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
+        
+        if self.rate_limiter.is_rate_limited(user_id):
+            return "You're sending messages too quickly. Please wait a moment before sending another message."
+        
         messages = [{"role": "system", "content": self.system_prompt()}]
 
         # Check if history is a list of dicts (Gradio "messages" format)