Update app.py

app.py

@@ -15,9 +15,6 @@ class SearchRequest(BaseModel):
     query: str
     top_k: int = 3
 
-class VideoRequest(BaseModel):
-    video_id: str
-
 @lru_cache(maxsize=1)
 def load_model():
     return SentenceTransformer('model/BGE-Medico-Model')
@@ -51,65 +48,6 @@ def search_device(request: SearchRequest):
     
     return {"results": results}
 
-@app.post('/get_sub')
-def get_subtitles(request: VideoRequest):
-    api = YouTubeTranscriptApi()
-    logging.info('Yes we are started')
-    try:
-        transcript_list = api.list(request.video_id)
-        logging.info(transcript_list)
-        english_codes = ['en', 'en-US', 'en-GB']
-        
-        def fetch_text(transcript):
-            subtitles = transcript.fetch()
-            text = " ".join([snippet.text for snippet in subtitles])
-            return {'lang':transcript.language, 'text': text}
-                    
-        try:
-            transcript = transcript_list.find_manually_created_transcript(english_codes)
-            return fetch_text(transcript)
-        
-        except CouldNotRetrieveTranscript:
-            pass
-        
-        # Step 2: Check any manually created transcript, translate if possible
-        manual_transcripts = [t for t in transcript_list if not t.is_generated]
-        if manual_transcripts:
-
-            orig_transcript = manual_transcripts[0]
-            if orig_transcript.is_translatable:
-                try:
-                    translated_transcript = orig_transcript.translate('en')
-                    return fetch_text(translated_transcript)
-
-                except (TranslationLanguageNotAvailable, NotTranslatable):
-                    pass
-            return fetch_text(orig_transcript)
-        
-        # Step 3: Check generated transcripts in English
-        try:
-            transcript = transcript_list.find_generated_transcript(english_codes)
-            return fetch_text(transcript)
-        except CouldNotRetrieveTranscript:
-            pass
-        
-        # Step 4: Any generated transcript
-        generated_transcripts = [t for t in transcript_list if t.is_generated]
-        if generated_transcripts:
-            orig_transcript = generated_transcripts[0]
-            if orig_transcript.is_translatable:
-                try:
-                    translated_transcript = orig_transcript.translate('en')
-                    return fetch_text(translated_transcript)
-                except (TranslationLanguageNotAvailable, NotTranslatable):
-                    pass
-
-            return fetch_text(orig_transcript)
-        
-        raise NoTranscriptFound("No subtitles available for this video.")
-    except Exception as e:
-        logging.info(e)
-        return e
 
 @app.get("/")
 def root():