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 search and download options endpoint
+# YouTube video download options endpoint
 @app.get("/youtube_video_options")
 async def youtube_video_options(
     query: str = Query(..., description="Search query for the YouTube video")
@@ -39,28 +39,42 @@ 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)
 
-        # Collect metadata for each search result
+        # Get the top video result from search results
+        target_video = search_results.items[0]
+
+        # Initialize an empty list for available download options
         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
-            })
 
-        # Return the search results with metadata
+        # 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,
+                            "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 JSONResponse(content={
-            "query": query,
+            "video_title": target_video.title,
             "available_downloads": available_downloads
         })
 
@@ -70,7 +84,7 @@ async def youtube_video_options(
 # Root endpoint
 @app.get("/")
 def root():
-    return "حسن زلقو"
+    return "Welcome to the Image Compression and YouTube Downloader API"
 
 if __name__ == "__main__":
     import uvicorn