Update app.py

app.py

@@ -591,6 +591,17 @@ async def download_audio(
     request: Request
     #api_key: str = Security(verify_api_key)
 ):
+    user_ip = get_user_ip(request)
+    
+    if rate_limiter.is_rate_limited(user_ip):
+        current_count = rate_limiter.get_current_count(user_ip)
+        raise HTTPException(
+            status_code=429,
+            detail={
+                "error": "You have exceeded the maximum number of requests per day. Please try again tomorrow.",
+                "url": "https://t.me/chrunoss"
+            }
+        )
     data = await request.json()
     video_url = data.get('url')
     #cookiefile = "firefox-cookies.txt"
@@ -633,7 +644,7 @@ async def download_audio(
         encoded_filename = urllib.parse.quote(downloaded_file.name)
         download_url = f"{BASE_URL}/file/{encoded_filename}"
         gc.collect()
-        return {"url": download_url}
+        return {"url": download_url, "requests_remaining": rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)}
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)