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):
# Get video transcript
if check_link(video_link):
if not check_link(video_link):
return "Invalid YouTube URL."
try:
loader = YoutubeLoader.from_youtube_url(video_link, language=["en", "en-US"])
transcript = loader.load()
return transcript
return "Invalid YouTube URL."
except Exception as e:
return f"Failed to retrieve transcript: {e}"
def split_chunks(transcript):