Update app.py

app.py

@@ -1,70 +1,54 @@
 import streamlit as st
-import tempfile
 import os
-import yt_dlp
+import tempfile
 from utils.transcriber import transcribe_audio
 from utils.summarizer import summarize_text
+import yt_dlp
+import ffmpeg
 
-# Title
-st.set_page_config(page_title="🎬 Video Summarizer", layout="centered")
-st.title("🎬 LLM-Powered Video Summarizer")
-
-# API key status
-OPENAI_KEY = os.getenv("OPENAI_API_KEY")
-if OPENAI_KEY:
-    st.success("✅ Using OpenAI API mode")
-else:
-    st.warning("⚠️ No API key found — using local models (slower, may be less accurate)")
+st.title("🎥 Video Summarizer (Local Models, No API Key Needed)")
 
-# Video input
-st.subheader("📥 Upload or Link a Video")
-video_source = st.radio("Choose input method:", ["YouTube Link", "Upload File"])
+# Step 1: Select input type
+input_method = st.radio("📥 Upload or Link a Video", ["YouTube Link", "Upload File"])
 
 video_path = None
 
-if video_source == "YouTube Link":
+if input_method == "YouTube Link":
     yt_url = st.text_input("Enter YouTube URL:")
-    if yt_url:
-        with st.spinner("Downloading video..."):
-            try:
-                tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    if st.button("Download & Process") and yt_url:
+        try:
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_file:
                 ydl_opts = {
-                    "outtmpl": tmp_file.name,
-                    "format": "mp4",
-                    "quiet": True,
+                    'format': 'best',
+                    'outtmpl': tmp_file.name,
                 }
                 with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                     ydl.download([yt_url])
                 video_path = tmp_file.name
-                st.success("YouTube video downloaded!")
-            except Exception as e:
-                st.error(f"Error downloading video: {e}")
+        except Exception as e:
+            st.error(f"Error downloading video: {e}")
 
-elif video_source == "Upload File":
-    uploaded_file = st.file_uploader("Upload a video file", type=["mp4", "mov", "avi", "mkv"])
-    if uploaded_file:
-        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        tmp_file.write(uploaded_file.read())
-        tmp_file.close()
-        video_path = tmp_file.name
-        st.success("File uploaded successfully!")
+elif input_method == "Upload File":
+    uploaded_file = st.file_uploader("Upload Video", type=["mp4", "mov", "avi", "mkv"])
+    if uploaded_file is not None:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_file:
+            tmp_file.write(uploaded_file.read())
+            video_path = tmp_file.name
 
-# Process
-if video_path and st.button("🚀 Process Video"):
-    with st.spinner("Transcribing video..."):
-        try:
-            transcript = transcribe_audio(video_path)
-            st.subheader("📝 Transcript")
-            st.text_area("", transcript, height=200)
+# Step 2: Process video if available
+if video_path:
+    try:
+        st.info("Transcribing video... This may take a while.")
+        transcript = transcribe_audio(video_path)
 
-            with st.spinner("Summarizing..."):
-                summary = summarize_text(transcript)
-                st.subheader("✨ Summary")
-                st.write(summary)
+        st.subheader("📜 Transcript")
+        st.write(transcript)
 
-        except Exception as e:
-            st.error(f"Processing failed: {e}")
+        st.info("Summarizing transcript...")
+        summary = summarize_text(transcript)
+
+        st.subheader("📝 Summary")
+        st.write(summary)
 
-# Footer
-st.markdown("---")
-st.caption("Built with ❤️ using Streamlit, Whisper, and GPT")
+    except Exception as e:
+        st.error(f"Processing failed: {e}")