Update app.py

app.py

@@ -124,6 +124,7 @@ async def get_video_url(youtube_url: str):
 global_download_dir = tempfile.mkdtemp()
 
 
+
 class RateLimiter:
     def __init__(self, max_requests: int, time_window: timedelta):
         self.max_requests = max_requests
@@ -142,13 +143,20 @@ class RateLimiter:
         """Check if the user has exceeded their rate limit."""
         self._cleanup_old_requests(user_ip)
         
-        # Add current request
+        # Get current count after cleanup
+        current_count = len(self.requests[user_ip])
+        
+        # Add current request timestamp (incrementing the count)
         current_time = time.time()
         self.requests[user_ip].append(current_time)
         
         # Check if user has exceeded the maximum requests
-        return len(self.requests[user_ip]) > self.max_requests
-
+        return (current_count + 1) > self.max_requests
+    
+    def get_current_count(self, user_ip: str) -> int:
+        """Get the current request count for an IP."""
+        self._cleanup_old_requests(user_ip)
+        return len(self.requests[user_ip])
 
 
 # Initialize rate limiter with 100 requests per day
@@ -164,6 +172,7 @@ def get_user_ip(request: Request) -> str:
         return forwarded.split(",")[0]
     return request.client.host
 
+
 @app.post("/maxs")
 async def download_high_quality_video(request: Request):
     user_ip = get_user_ip(request)