Update main.py

main.py

@@ -31,7 +31,7 @@ async def compress_image(image_url: str = Query(..., description="URL of the ima
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
-# YouTube video download options endpoint
+# YouTube video search and download options endpoint
 @app.get("/youtube_video_options")
 async def youtube_video_options(
     query: str = Query(..., description="Search query for the YouTube video")
@@ -39,43 +39,28 @@ async def youtube_video_options(
     try:
         yt = Ytube()
         search_results = yt.search_videos(query)
+        
         if not search_results.items:
             return JSONResponse(content={"error": "No videos found for the query"}, status_code=404)
 
-        # Get the top video result from search results
-        target_video = search_results.items[0]
-
-        # Initialize an empty list for available download options
+        # Collect metadata for each search result
         available_downloads = []
+        
+        for video in search_results.items:
+            available_downloads.append({
+                "title": video.title,
+                "video_id": video.id,
+                "size": video.size,
+                "duration": video.duration,
+                "channel": video.channelTitle,
+                "source": video.source,
+                "quality": "unknown",  # Assuming search does not provide quality; update if available.
+                "format": "mp4"        # Assuming default format mp4
+            })
 
-        # Check common qualities and formats (replace with dynamically available ones if possible)
-        qualities = ["144", "240", "360", "480", "720", "1080"]
-        formats = ["mp4"]
-
-        for quality in qualities:
-            for fmt in formats:
-                try:
-                    # Try to get the download link for each quality and format
-                    download_link = yt.get_download_link(target_video, format=fmt, quality=quality)
-                    
-                    # Only add if a valid link is returned
-                    if download_link.url:
-                        available_downloads.append({
-                            "quality": quality,
-                            "format": fmt,
-                            "size": download_link.size,
-                            "download_url": download_link.url
-                        })
-                except Exception as e:
-                    print(f"Quality {quality} and format {fmt} not available: {e}")
-                    continue
-
-        if not available_downloads:
-            return JSONResponse(content={"error": "No available download options found."}, status_code=404)
-
-        # Return the video metadata and available download options
+        # Return the search results with metadata
         return JSONResponse(content={
-            "video_title": target_video.title,
+            "query": query,
             "available_downloads": available_downloads
         })