Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -53,18 +53,16 @@ def get_subtitle(url, lang='en'):
|
|
53 |
}
|
54 |
try:
|
55 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
subtitle_content = re.sub(r"<[^>]+>", "", subtitle_content)
|
67 |
-
return subtitle_content
|
68 |
except error:
|
69 |
print(error)
|
70 |
return None
|
|
|
53 |
}
|
54 |
try:
|
55 |
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
56 |
+
info_dict = ydl.extract_info(url, download=True)
|
57 |
+
video_id = info_dict.get("id", None)
|
58 |
+
if video_id is None:
|
59 |
+
return None
|
60 |
+
|
61 |
+
subtitle_file = f"{video_id}.{lang}.vtt"
|
62 |
+
with open(subtitle_file, 'r') as f:
|
63 |
+
subtitle_content = f.read()
|
64 |
+
subtitle_content = re.sub(r"<[^>]+>", "", subtitle_content)
|
65 |
+
return subtitle_content
|
|
|
|
|
66 |
except error:
|
67 |
print(error)
|
68 |
return None
|