lanbogao commited on
Commit
21ede15
1 Parent(s): ce80040

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import whisper
 
3
  from pytube import YouTube
4
  from fastapi import FastAPI, Response, Request
5
  import yt_dlp
@@ -22,7 +23,9 @@ def get_subtitle(url, lang='en'):
22
  subtitle_file = f"{video_id}.{lang}.vtt"
23
  if not os.path.exists(subtitle_file):
24
  return None
25
- return subtitle_file
 
 
26
 
27
  def download_audio(video_url, quality: str = '128', speed: float = None):
28
  ydl_opts = {
 
1
  import gradio as gr
2
  import whisper
3
+ import subprocess
4
  from pytube import YouTube
5
  from fastapi import FastAPI, Response, Request
6
  import yt_dlp
 
23
  subtitle_file = f"{video_id}.{lang}.vtt"
24
  if not os.path.exists(subtitle_file):
25
  return None
26
+
27
+ subtitle_text = subprocess.check_output(['webvtt-to-text', subtitle_file]).decode('utf-8')
28
+ return subtitle_text
29
 
30
  def download_audio(video_url, quality: str = '128', speed: float = None):
31
  ydl_opts = {