Update app.py

app.py

@@ -9,6 +9,7 @@ import asyncio
 import cloudscraper
 from pydantic import BaseModel
 from urllib.parse import urlparse, parse_qs
+import redis
 
 app = Flask(__name__)
 ytmusic = YTMusic()
@@ -206,23 +207,64 @@ async def get_track_download_url(track_id: str, quality: str) -> str:
 
 
     
+
+# Initialize Redis client
+redis_client = redis.Redis(host='localhost', port=6379, db=0)
+
+def get_daily_requests(ip):
+    """Get the number of requests made today by this IP"""
+    today = datetime.now().strftime('%Y-%m-%d')
+    key = f'daily_requests:{ip}:{today}'
+    return int(redis_client.get(key) or 0)
+
+def increment_daily_requests(ip):
+    """Increment the request counter for this IP"""
+    today = datetime.now().strftime('%Y-%m-%d')
+    key = f'daily_requests:{ip}:{today}'
+    # Set expiration to end of day (seconds until midnight)
+    seconds_until_midnight = 86400 - (datetime.now().hour * 3600 + datetime.now().minute * 60 + datetime.now().second)
+    redis_client.incr(key)
+    redis_client.expire(key, seconds_until_midnight)
+
 @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:
+        return jsonify({
+            "error": "Daily limit of 20 downloads exceeded. Please try again tomorrow."
+        }), 429
+    
     data = request.get_json()
     track_id = data.get('track_id')
     quality = data.get('quality', '128')
+    
     try:
         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."}), 400
+            return jsonify({
+                "error": "Quality above 128 or FLAC is for Premium users Only."
+            }), 400
+            
         dl_url = await get_track_download_url(track_id, quality)
+        
         if dl_url and "http" in dl_url:
+            # Increment counter only on successful download
+            increment_daily_requests(client_ip)
             result = {"url": dl_url}
             return jsonify(result)
         else:
-            return jsonify({"error": "Failed to Fetch the Track."}), 400
+            return jsonify({
+                "error": "Failed to Fetch the Track."
+            }), 400
+            
     except ValueError:
-        return jsonify({"error": "Invalid quality value provided. It should be a valid integer or FLAC."}), 400
+        return jsonify({
+            "error": "Invalid quality value provided. It should be a valid integer or FLAC."
+        }), 400