Update main.py

main.py

@@ -4,7 +4,7 @@ from fastapi.responses import JSONResponse
 import cloudinary
 import cloudinary.uploader
 from cloudinary.utils import cloudinary_url
-from y2mate_api import Handler  # Importing the Y2Mate handler
+from ytube_api import Ytube  # Importing the Ytube module
 
 app = FastAPI()
 
@@ -39,16 +39,32 @@ async def compress_image(image_url: str = Query(..., description="URL of the ima
 
 # YouTube video download endpoint
 @app.get("/download_youtube_video")
-async def download_youtube_video(video_url: str = Query(..., description="URL of the YouTube video to download")):
+async def download_youtube_video(
+    query: str = Query(..., description="Search query for the YouTube video"),
+    format: str = Query("mp4", description="Video format to download, e.g., mp4"),
+    quality: str = Query("1080", description="Video quality, e.g., 1080")
+):
     try:
-        # Step 1: Create a handler object for the Y2Mate API
-        api = Handler(video_url)
+        # Initialize the Ytube instance
+        yt = Ytube()
 
-        # Step 2: Fetch video metadata and download links
-        video_metadata = [metadata for metadata in api.run()]
+        # Search for videos using the provided query
+        search_results = yt.search_videos(query)
+        if not search_results.items:
+            return JSONResponse(content={"error": "No videos found for the query"}, status_code=404)
 
-        # Step 3: Return the metadata or download links
-        return JSONResponse(content={"video_metadata": video_metadata})
+        # Select the first result from the search results
+        target_video = search_results.items[0]
+
+        # Get the download link for the specified format and quality
+        download_link = yt.get_download_link(target_video, format=format, quality=quality)
+
+        # Return the download link and metadata
+        return JSONResponse(content={
+            "download_url": download_link.url,
+            "filename": download_link.filename,
+            "status": download_link.status
+        })
 
     except Exception as e:
         # Handle any error and return a response