Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
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 = {
|