From a1e28966fb4d4788d1afd301a0e585a59fe58410 Mon Sep 17 00:00:00 2001 From: Mert Cobanov Date: Wed, 1 May 2024 20:04:16 +0300 Subject: [PATCH] Fix YouTube link validation in yt_summarizer.py --- yt_summarizer.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/yt_summarizer.py b/yt_summarizer.py index 65cf432..356069c 100644 --- a/yt_summarizer.py +++ b/yt_summarizer.py @@ -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):