lanbogao commited on
Commit
e6a27f0
1 Parent(s): ff53fa1

Update fetchYoutubeSubtitle.py

Browse files
Files changed (1) hide show
  1. fetchYoutubeSubtitle.py +3 -4
fetchYoutubeSubtitle.py CHANGED
@@ -44,9 +44,8 @@ def fetchYoutubeSubtitleUrls(video_id):
44
  }
45
 
46
  response = session.post(SUBTITLE_DOWNLOADER_URL + '/action/extract', json=data, headers=headers)
47
- if not response.ok:
48
- logger.error(headers, "request header")
49
- logger.error(response.headers.raw())
50
  return {'title': None, 'subtitleList': None, 'error': response.statusText}
51
  else:
52
  try:
@@ -93,7 +92,7 @@ async def fetchYoutubeSubtitle(videoId: str) -> Dict:
93
  "https": proxy,
94
  }
95
  response = session.get(url)
96
-
97
  loader = SRTLoader(await response.blob())
98
  splitter = TokenTextSplitter(
99
  encodingName="cl100k_base",
 
44
  }
45
 
46
  response = session.post(SUBTITLE_DOWNLOADER_URL + '/action/extract', json=data, headers=headers)
47
+ if not response.status_code != 200:
48
+ logger.error(headers)
 
49
  return {'title': None, 'subtitleList': None, 'error': response.statusText}
50
  else:
51
  try:
 
92
  "https": proxy,
93
  }
94
  response = session.get(url)
95
+
96
  loader = SRTLoader(await response.blob())
97
  splitter = TokenTextSplitter(
98
  encodingName="cl100k_base",