Update app.py

app.py

@@ -8,10 +8,11 @@ from fastapi import FastAPI, HTTPException, status, Request
 from yt_dlp import YoutubeDL
 from yt_dlp.version import __version__ as yt_dlp_version
 from typing import Union, Dict
+import uvicorn
 
 app = FastAPI(docs_url=None, redoc_url=None)
 
-# Set cache directory to /tmp which is writable in Vercel
+# Set cache directory to /tmp which is writable in Hugging Face containers
 os.environ["XDG_CACHE_HOME"] = "/tmp"
 
 # Rate limiting configuration
@@ -110,7 +111,7 @@ async def version_info():
 
 @app.get('/')
 def main():
-    return "Chrunos Downloader API Is Running well."
+    return "Chrunos Downloader API Is Running well on Hugging Face."
 
 @app.get("/api/info")
 async def get_info(
@@ -169,8 +170,6 @@ async def get_info(
                 headers={"Cache-Control": "no-store, max-age=0"},
             )
 
-
-
 @app.get("/api/playlist")
 async def get_playlist_info(
     request: Request,
@@ -182,20 +181,16 @@ async def get_playlist_info(
     Fetches paginated items from a playlist or user profile.
     Strictly enforces a maximum of 50 items per request and provides a next_page URL.
     """
-    # 1. Validate inputs
     if start < 1:
         raise HTTPException(status_code=400, detail="'start' must be 1 or greater.")
     if end < start:
         raise HTTPException(status_code=400, detail="'end' must be greater than or equal to 'start'.")
 
-    # 2. Enforce a hard limit of 50 items per request
-    # If a client requests start=51 & end=1000, we force end=100.
     requested_count = end - start + 1
     if requested_count > 50:
         end = start + 49
         requested_count = 50
 
-    # Rate Limiting
     client_ip = get_client_ip(request)
     is_allowed, remaining = check_rate_limit(client_ip)
     
@@ -219,7 +214,6 @@ async def get_playlist_info(
         "ignoreerrors": True,
         "cachedir": "/tmp/yt-dlp-cache",
         "js-runtimes": "node",
-        # Pass the strictly clamped pagination settings to yt-dlp
         "playliststart": start,
         "playlistend": end
     }
@@ -230,26 +224,15 @@ async def get_playlist_info(
             if not response:
                 raise HTTPException(status_code=404, detail="Playlist or profile not found.")
             
-            # Extract entries. yt-dlp might omit 'entries' if out of bounds, so default to empty list.
             raw_entries = response.get("entries") or []
-            
-            # Filter out 'None' entries (yt-dlp sometimes returns None for deleted/private videos in a playlist)
             valid_entries = [e for e in raw_entries if e is not None]
 
-            # 3. Determine if there is a next page
             next_page_url = None
-            
-            # If yt-dlp found enough items to fill our requested quota, there is likely a next page.
-            # (Note: we check the length of raw_entries because deleted videos still consume an index slot)
             if len(raw_entries) >= requested_count:
                 next_start = end + 1
                 next_end = next_start + 49
-                
-                # Safely encode the target URL
                 encoded_url = urllib.parse.quote(url)
                 base_url = str(request.base_url).rstrip('/')
-                
-                # Construct the ready-to-use next_page URL for the client
                 next_page_url = f"{base_url}/api/playlist?url={encoded_url}&start={next_start}&end={next_end}"
 
             clean_response = {
@@ -278,10 +261,8 @@ async def get_playlist_info(
                 headers={"Cache-Control": "no-store, max-age=0"},
             )
 
-
 @app.get("/api/rate-limit-status")
 async def get_rate_limit_status(request: Request):
-    """Endpoint to check current rate limit status for the requesting IP"""
     client_ip = get_client_ip(request)
     rate_limits = load_rate_limits()
     rate_limits = cleanup_old_entries(rate_limits)
@@ -300,4 +281,8 @@ async def get_rate_limit_status(request: Request):
         "used": used,
         "remaining": remaining,
         "reset_time": f"{current_date}T00:00:00Z"
-    })
+    })
+
+# Add the Hugging Face compatible execution block
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)