Update main.py

main.py

@@ -4,6 +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
 
 app = FastAPI()
 
@@ -36,10 +37,27 @@ async def compress_image(image_url: str = Query(..., description="URL of the ima
         # Handle any error and return a response
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
+# 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")):
+    try:
+        # Step 1: Create a handler object for the Y2Mate API
+        api = Handler(video_url)
+
+        # Step 2: Fetch video metadata and download links
+        video_metadata = [metadata for metadata in api.run()]
+
+        # Step 3: Return the metadata or download links
+        return JSONResponse(content={"video_metadata": video_metadata})
+
+    except Exception as e:
+        # Handle any error and return a response
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+
 # Root endpoint
 @app.get("/")
 def root():
-    return "Welcome to the Image Compression API"
+    return "Welcome to the Image Compression and YouTube Downloader API"
 
 if __name__ == "__main__":
     import uvicorn