Update app.py

app.py

@@ -9,6 +9,14 @@ import yt_dlp
 langs = ["None"] + sorted(list(whisper.tokenizer.LANGUAGES.values()))
 model_size = list(whisper._MODELS.keys())
 
+
+@app.get("/subtitle")
+async def get_subtitle(url: str):
+    # Download the subtitle with download_subtitle()
+    subtitle_url = download_subtitle(url)
+    # Stream the subtitle as a response
+    return StreamingResponse(requests.get(subtitle_url, stream=True).iter_content(chunk_size=1024))
+
 def get_subtitle(url, lang='en'):
     # Download subtitles if available
     ydl_opts = {
@@ -23,9 +31,10 @@ def get_subtitle(url, lang='en'):
         if video_id is None:
             return None
         subtitle_file = f"{video_id}.{lang}.vtt"
-        if  os.path.exists(subtitle_file):
-            subtitle_text = subprocess.check_output(['webvtt-to-text', subtitle_file]).decode('utf-8')
-            return subtitle_text
+        with open(subtitle_file, 'r') as f:
+            subtitle_content = f.read()
+            subtitle_content = re.sub(r"<[^>]+>", "", subtitle_content)
+            return subtitle_content
     return None
 
 def download_audio(video_url, quality: str = '128', speed: float = None):