Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from fastapi import FastAPI, HTTPException, UploadFile, Depends
+from fastapi import FastAPI, HTTPException, UploadFile, Depends, Query
 from fastapi.security import APIKeyHeader
 import aiohttp
 import asyncio
@@ -48,7 +48,7 @@ def get_random_user_agent() -> str:
     ]
     return random.choice(user_agents)
 
-async def search_and_get_videos(query: str) -> List[Dict]:
+async def search_and_get_videos(query: str, num_videos: int = 2) -> List[Dict]:
     for instance in INVIDIOUS_INSTANCES:
         url = f"{instance}/api/v1/search?q={query}&type=video"
         try:
@@ -65,7 +65,7 @@ async def search_and_get_videos(query: str) -> List[Dict]:
                             else "",
                         }
                         for video in search_results
-                    ][:2]
+                    ][:num_videos]
                     return videos
         except aiohttp.ClientError as e:
             logger.error(f"Error performing video search on {instance}: {e}")
@@ -162,8 +162,12 @@ async def verify_api_key(api_key: str = Depends(api_key_header)):
     return api_key
 
 @app.get("/search-videos/")
-async def search_videos(query: str, api_key: str = Depends(verify_api_key)):
-    videos = await search_and_get_videos(query)
+async def search_videos(
+    query: str,
+    num_videos: int = Query(default=2, ge=1, le=10),
+    api_key: str = Depends(verify_api_key)
+):
+    videos = await search_and_get_videos(query, num_videos)
     if not videos:
         raise HTTPException(status_code=404, detail="No videos found or an error occurred during the search.")
     return {"videos": videos}