Update app.py

app.py

@@ -18,7 +18,7 @@ from collections import defaultdict
 import time
 from datetime import timedelta
 import gc
-
+from typing import Dict
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -42,13 +42,13 @@ BASE_URL = "https://chrunos-depot.hf.space"
 # Deezer ARL token (required for deezspot downloads)
 ARL_TOKEN = os.getenv('ARL')
 
+
 class DownloadRequest(BaseModel):
     url: str
     quality: str
     arl: str
 
 
-
 def convert_deezer_short_link_async(short_link: str) -> str:
     try:
         response = requests.get(short_link, allow_redirects=True)
@@ -90,7 +90,7 @@ class RateLimiter:
         self.max_requests = max_requests
         self.time_window = time_window
         self.requests: Dict[str, list] = defaultdict(list)
-    
+
     def _cleanup_old_requests(self, user_ip: str) -> None:
         """Remove requests that are outside the time window."""
         current_time = time.time()
@@ -98,21 +98,21 @@ class RateLimiter:
             timestamp for timestamp in self.requests[user_ip]
             if current_time - timestamp < self.time_window.total_seconds()
         ]
-    
+
     def is_rate_limited(self, user_ip: str) -> bool:
         """Check if the user has exceeded their rate limit."""
         self._cleanup_old_requests(user_ip)
-        
+
         # 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 (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)
@@ -125,6 +125,7 @@ rate_limiter = RateLimiter(
     time_window=timedelta(days=1)
 )
 
+
 def get_user_ip(request: Request) -> str:
     """Helper function to get user's IP address."""
     forwarded = request.headers.get("X-Forwarded-For")
@@ -135,10 +136,10 @@ def get_user_ip(request: Request) -> str:
 
 # Download a track and return a download URL
 @app.post("/download/track")
-def download_track(request: DownloadRequest):
+def download_track(request: Request, download_request: DownloadRequest):
     try:
         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(
@@ -148,15 +149,15 @@ def download_track(request: DownloadRequest):
                     "help": "https://t.me/chrunoss"
                 }
             )
-        if request.arl is None or request.arl.strip() == "":
+        if download_request.arl is None or download_request.arl.strip() == "":
             ARL = ARL_TOKEN
         else:
-            ARL = request.arl
+            ARL = download_request.arl
         logger.info(f'arl: {ARL}')
-        url = request.url
+        url = download_request.url
         if 'deezer.page' in url:
             url = convert_deezer_short_link_async(url)
-        quality = request.quality
+        quality = download_request.quality
         dl = DeeLogin(arl=ARL)
 
         if quality not in ["MP3_320", "MP3_128", "FLAC"]:
@@ -226,6 +227,7 @@ def download_track(request: DownloadRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+
 # Pydantic model for album request
 class AlbumRequest(BaseModel):
     album_id: str