Upload app.py

app.py

@@ -11,6 +11,7 @@ app = FastAPI()
 class SearchRequest(BaseModel):
     query: str
     top_k: int = 3
+    video_id : str
 
 @lru_cache(maxsize=1)
 def load_model():
@@ -46,10 +47,10 @@ def search_device(request: SearchRequest):
     return {"results": results}
 
 @app.post('/get_sub')
-def get_subtitles(video_id):
+def get_subtitles(request: SearchRequest):
     api = YouTubeTranscriptApi()
     try:
-        transcript_list = api.list(video_id)
+        transcript_list = api.list(request.video_id)
 
         english_codes = ['en', 'en-US', 'en-GB']