Significant-Gravitas
/
AutoGPT
mirror of https://github.com/Significant-Gravitas/AutoGPT.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Add provision for other languages in Youtube Video Block (#8630) 

Co-authored-by: Toran Bruce Richards <toran.richards@gmail.com>
pull/8642/head^2
vishesh10 2024-11-14 14:50:46 +05:30 committed by GitHub
parent ce667f6287
commit 639242ac68
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
autogpt_platform/backend/backend/blocks/youtube.py
View File

@ -62,7 +62,22 @@ class TranscribeYoutubeVideoBlock(Block):
@staticmethod @staticmethod
def get_transcript(video_id: str): def get_transcript(video_id: str):
return YouTubeTranscriptApi.get_transcript(video_id) try:
transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
if not transcript_list:
raise ValueError(f"No transcripts found for the video: {video_id}")
for transcript in transcript_list:
first_transcript = transcript_list.find_transcript(
[transcript.language_code]
)
return YouTubeTranscriptApi.get_transcript(
video_id, languages=[first_transcript.language_code]
)
except Exception:
raise ValueError(f"No transcripts found for the video: {video_id}")
def run(self, input_data: Input, **kwargs) -> BlockOutput: def run(self, input_data: Input, **kwargs) -> BlockOutput:
video_id = self.extract_video_id(input_data.youtube_url) video_id = self.extract_video_id(input_data.youtube_url)