Update app.py

app.py

@@ -158,9 +158,18 @@ prompt_template = PromptTemplate(
 # -------------------------------------------------
 # 5️⃣ Streamlit App UI
 # -------------------------------------------------
+import re
+
+def extract_video_id(url: str) -> str:
+    # Handles both youtube.com and youtu.be formats
+    pattern = r"(?:v=|\/)([0-9A-Za-z_-]{11}).*"
+    match = re.search(pattern, url)
+    return match.group(1) if match else None
+    
 st.title("🎬 YouTube Transcript Chatbot (RAG)")
 
-video_id = st.text_input("YouTube Video ID", value="lv1_-RER4_I")
+video_url = st.text_input("Enter YouTube Video URL", value="lv1_-RER4_I")
+video_id = extract_video_id(video_url)
 query = st.text_area("Your Query", value="What is RAG?")
 model_choice = st.radio("Model to Use", ["Llama-3.2-1B", "Gemma-2-3B", "DeepSeek-685B", "OpenAI-20B"])
 temperature = st.slider("Temperature", 0, 100, value=50) / 100.0