Update app.py

app.py

@@ -1,15 +1,36 @@
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 import yt_dlp
 import base64
 
-app = FastAPI()
+app = FastAPI(
+    title="YouTube Download API",
+    description="API for retrieving YouTube audio/video download links using yt-dlp.",
+    version="1.0.0",
+    contact={
+        "name": "Your Name",
+        "url": "https://github.com/yourusername",
+        "email": "[email protected]",
+    }
+)
 
 class VideoRequest(BaseModel):
-    video_id: str
+    video_id: str = Field(..., example="dQw4w9WgXcQ", description="YouTube video ID (e.g., dQw4w9WgXcQ)")
+
+class AudioResponse(BaseModel):
+    status: str = Field(..., example="success")
+    audio_url: str = Field(..., description="Base64 encoded direct audio URL")
 
-@app.post("/audio/")
+class VideoResponse(BaseModel):
+    status: str = Field(..., example="success")
+    video_url: str = Field(..., description="Base64 encoded direct video URL")
+
+@app.post("/audio/", response_model=AudioResponse, summary="Get YouTube Audio URL")
 def get_audio_link(req: VideoRequest):
+    """
+    Returns a downloadable audio link for a given YouTube video ID.
+    The URL is base64-encoded.
+    """
     url = f"https://www.youtube.com/watch?v={req.video_id}"
     ydl_opts = {
         'quiet': True,
@@ -17,7 +38,6 @@ def get_audio_link(req: VideoRequest):
         'skip_download': True,
         'forceurl': True,
     }
-
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)
@@ -30,8 +50,12 @@ def get_audio_link(req: VideoRequest):
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
-@app.post("/video/")
+@app.post("/video/", response_model=VideoResponse, summary="Get YouTube Video URL")
 def get_video_link(req: VideoRequest):
+    """
+    Returns a downloadable video link (max 720p) for a given YouTube video ID.
+    The URL is base64-encoded.
+    """
     url = f"https://www.youtube.com/watch?v={req.video_id}"
     ydl_opts = {
         'quiet': True,