Update app.py

app.py

@@ -9,7 +9,8 @@ import asyncio
 import cloudscraper
 from pydantic import BaseModel
 from urllib.parse import urlparse, parse_qs
-import redis
+from collections import defaultdict
+import threading
 
 app = Flask(__name__)
 ytmusic = YTMusic()
@@ -208,23 +209,41 @@ 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)
+
+
+# In-memory storage for request counting
+# Structure: {ip: [(timestamp, count), ...]}
+request_counts = defaultdict(list)
+request_lock = threading.Lock()
+
+def cleanup_old_records():
+    """Remove records older than 24 hours"""
+    today = datetime.now()
+    with request_lock:
+        for ip in list(request_counts.keys()):
+            request_counts[ip] = [
+                (timestamp, count) 
+                for timestamp, count in request_counts[ip] 
+                if (today - timestamp).days < 1
+            ]
+            if not request_counts[ip]:
+                del request_counts[ip]
 
 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)
+    cleanup_old_records()
+    today = datetime.now()
+    with request_lock:
+        return sum(
+            count for timestamp, count in request_counts[ip]
+            if (today - timestamp).days < 1
+        )
 
 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)
+    now = datetime.now()
+    with request_lock:
+        request_counts[ip].append((now, 1))
 
 @app.route('/track_dl', methods=['POST'])
 async def track_dl():
@@ -266,7 +285,6 @@ async def track_dl():
             "error": "Invalid quality value provided. It should be a valid integer or FLAC."
         }), 400
     
-