From c6855d27f0ee381481cb42d50dcb70969d26a8ea Mon Sep 17 00:00:00 2001
From: tcsenpai <dev@tcsenpai.com>
Date: Wed, 20 Nov 2024 22:23:53 +0100
Subject: [PATCH] support for youtu.be

---
 src/main.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main.py b/src/main.py
index 2c17f28..62c930d 100644
--- a/src/main.py
+++ b/src/main.py
@@ -313,6 +313,10 @@ def main():
         else:
             st.error("Please enter a valid YouTube video URL.")
             return
+    # Support short urls as well
+    if "https://youtu.be/" in video_url:
+        video_id = video_url.split("youtu.be/")[-1]
+        video_url = f"https://www.youtube.com/watch?v={video_id}"
 
     if st.button("Summarize"):
         if video_url: