Fix YouTube link validation in yt_summarizer.py

This commit is contained in:
Mert Cobanov 2024-05-01 20:04:16 +03:00
parent 8914692221
commit a1e28966fb

View File

@ -12,12 +12,14 @@ def check_link(link):
def get_transcript(video_link): def get_transcript(video_link):
# Get video transcript if not check_link(video_link):
if check_link(video_link): return "Invalid YouTube URL."
try:
loader = YoutubeLoader.from_youtube_url(video_link, language=["en", "en-US"]) loader = YoutubeLoader.from_youtube_url(video_link, language=["en", "en-US"])
transcript = loader.load() transcript = loader.load()
return transcript return transcript
return "Invalid YouTube URL." except Exception as e:
return f"Failed to retrieve transcript: {e}"
def split_chunks(transcript): def split_chunks(transcript):