Update app.py

app.py

@@ -211,10 +211,11 @@ async def get_track_download_url(track_id: str, quality: str) -> str:
 
 
 
+
 # In-memory storage for request counting
-# Structure: {ip: [(timestamp, count), ...]}
 request_counts = defaultdict(list)
 request_lock = threading.Lock()
+DAILY_LIMIT = 5
 
 def cleanup_old_records():
     """Remove records older than 24 hours"""
@@ -247,14 +248,16 @@ def increment_daily_requests(ip):
 
 @app.route('/track_dl', methods=['POST'])
 async def track_dl():
-    # Get client IP
     client_ip = request.remote_addr
     
-    # Check if daily limit exceeded
-    daily_requests = get_daily_requests(client_ip)
-    if daily_requests >= 20:
+    # Check current usage
+    current_requests = get_daily_requests(client_ip)
+    remaining_requests = DAILY_LIMIT - current_requests
+    
+    if remaining_requests <= 0:
         return jsonify({
-            "error": "Daily limit of 20 downloads exceeded. Please try again tomorrow."
+            "error": "Daily limit of 20 downloads exceeded. Please try again tomorrow.",
+            "remaining_requests": 0
         }), 429
     
     data = request.get_json()
@@ -265,7 +268,8 @@ async def track_dl():
         quality_num = int(quality)
         if quality_num > 128 or quality.upper() == 'FLAC':
             return jsonify({
-                "error": "Quality above 128 or FLAC is for Premium users Only."
+                "error": "Quality above 128 or FLAC is for Premium users Only.",
+                "remaining_requests": remaining_requests
             }), 400
             
         dl_url = await get_track_download_url(track_id, quality)
@@ -273,16 +277,23 @@ async def track_dl():
         if dl_url and "http" in dl_url:
             # Increment counter only on successful download
             increment_daily_requests(client_ip)
-            result = {"url": dl_url}
+            remaining_requests -= 1  # Decrease remaining count after successful increment
+            
+            result = {
+                "url": dl_url,
+                "remaining_requests": remaining_requests
+            }
             return jsonify(result)
         else:
             return jsonify({
-                "error": "Failed to Fetch the Track."
+                "error": "Failed to Fetch the Track.",
+                "remaining_requests": remaining_requests
             }), 400
             
     except ValueError:
         return jsonify({
-            "error": "Invalid quality value provided. It should be a valid integer or FLAC."
+            "error": "Invalid quality value provided. It should be a valid integer or FLAC.",
+            "remaining_requests": remaining_requests
         }), 400