lanbogao commited on
Commit
f7b2b93
1 Parent(s): eb6bfe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
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
- 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
- print(os.list)
62
- print(info_dict)
63
- subtitle_file = f"{video_id}.{lang}.vtt"
64
- with open(subtitle_file, 'r') as f:
65
- subtitle_content = f.read()
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