Update app.py

app.py

@@ -1,9 +1,9 @@
 from fastapi import FastAPI, HTTPException, Security, Depends
 from fastapi.security import APIKeyHeader
-from pydantic import HttpUrl, BaseModel
+from pydantic import HttpUrl
 import yt_dlp
 import uvicorn
-from typing import List, Optional
+from typing import Dict, Optional
 from datetime import timedelta
 import os
 import logging
@@ -31,22 +31,7 @@ async def get_api_key(api_key_header: Optional[str] = Security(api_key_header)):
 # Path to cookies file (optional)
 COOKIES_FILE = "cookies.txt"
 
-# Pydantic models for structured response
-class VideoLink(BaseModel):
-    quality: str
-    url: str
-    format_id: str
-    type: str  # "direct" or "streaming"
-
-class VideoInfoResponse(BaseModel):
-    title: str
-    duration: str
-    file_size: str
-    download_links: List[VideoLink]
-    thumbnail: str
-    link_type: str
-
-@app.get("/video-info", response_model=VideoInfoResponse)
+@app.get("/video-info", response_model=Dict)
 async def get_video_info(video_url: HttpUrl, api_key: str = Depends(get_api_key)):
     """
     Extract direct download links for video-only MP4 (720p and higher) or streaming links if no direct links are available,
@@ -111,7 +96,7 @@ async def get_video_info(video_url: HttpUrl, api_key: str = Depends(get_api_key)
             # If no direct formats, fall back to streaming formats (720p or higher)
             if not direct_formats:
                 logger.warning(f"No direct MP4 video-only formats (720p+) found for {video_url_str}, falling back to streaming formats")
-                streaming FORMATS = [
+                streaming_formats = [
                     f for f in info.get('formats', [])
                     if f.get('vcodec') != 'none'
                     and f.get('acodec') == 'none'
@@ -134,12 +119,12 @@ async def get_video_info(video_url: HttpUrl, api_key: str = Depends(get_api_key)
 
             # Get download links
             download_links = [
-                VideoLink(
-                    quality=f"{f.get('height')}p",
-                    url=f.get('url'),
-                    format_id=f.get('format_id'),
-                    type=link_type
-                )
+                {
+                    "quality": f"{f.get('height')}p",
+                    "url": f.get('url'),
+                    "format_id": f.get('format_id'),
+                    "type": link_type
+                }
                 for f in formats
             ]
 
@@ -152,14 +137,14 @@ async def get_video_info(video_url: HttpUrl, api_key: str = Depends(get_api_key)
             ).get('url', 'No thumbnail available')
 
             logger.info(f"Successfully extracted info for {video_url_str}: {len(download_links)} {link_type} formats found")
-            return VideoInfoResponse(
-                title=title,
-                duration=duration_str,
-                file_size=file_size_str,
-                download_links=download_links,
-                thumbnail=best_thumbnail,
-                link_type=link_type
-            )
+            return {
+                "title": title,
+                "duration": duration_str,
+                "file_size": file_size_str,
+                "download_links": download_links,
+                "thumbnail": best_thumbnail,
+                "link_type": link_type
+            }
 
     except HTTPException as he:
         raise he