Update app.py

app.py

@@ -8,8 +8,58 @@ headers = {
     "x-api-key": os.getenv("JIGSAWSTACK_API_KEY")
 }
 
+
+# Rate limiting configuration
+request_times = defaultdict(list)
+MAX_REQUESTS = 1  # 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 = get_real_ip(request)
+    now = time.time()
+    
+    print(f"Ip: {ip}")
+    print(f"Now: {now}")
+    print(f"Request times: {request_times[ip]}")
+    # Clean up old timestamps outside the time window
+    request_times[ip] = [t for t in request_times[ip] if now - t < TIME_WINDOW]
+    
+    print(f"Request times: {request_times[ip]}")
+
+    # Check if rate limit exceeded
+    if len(request_times[ip]) >= MAX_REQUESTS:
+        time_remaining = int(TIME_WINDOW - (now - request_times[ip][0]))
+        return False, f"Rate limit exceeded. You can make {MAX_REQUESTS} requests per {TIME_WINDOW} seconds. Try again in {time_remaining} seconds."
+    
+    # Add current request timestamp
+    request_times[ip].append(now)
+    return True, ""
+
+
 def transcribe_audio(input_type, audio_url, file_store_key, language):
     """Transcribe audio using JigsawStack Speech-to-Text API"""
+
+    # Check rate limit first
+    rate_limit_ok, rate_limit_msg = check_rate_limit(request)
+    if not rate_limit_ok:
+        return rate_limit_msg, ""
+    
     if input_type == "Audio URL" and not audio_url:
         return "Error: Please provide an audio URL.", ""
     if input_type == "File Store Key" and not file_store_key: