Update app.py

app.py

@@ -9,31 +9,6 @@ import yt_dlp as youtube_dl
 # App Configuration
 st.set_page_config(page_title="Video-to-Text Summarization", layout="wide")
 
-# Navbar-like Section
-st.markdown("""
-    <style>
-        .nav-links {
-            display: flex;
-            justify-content: center;
-            gap: 20px;
-        }
-        .nav-links a {
-            text-decoration: none;
-            padding: 10px 20px;
-            background: #4CAF50;
-            color: white;
-            border-radius: 5px;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
-st.markdown("""
-    <div class='nav-links'>
-        <a href='#local-upload'>Upload Local Video</a>
-        <a href='#youtube-upload'>Upload YouTube Video</a>
-    </div>
-""", unsafe_allow_html=True)
-
 # Header
 st.title("🎥 Smart Video-to-Text Summarization App")
 st.markdown("""
@@ -50,10 +25,15 @@ if "video_path" not in st.session_state:
 # Upload Video Section
 st.header("Upload Your Video")
 
+# Buttons for navigation
 col1, col2 = st.columns(2)
-
 with col1:
-    st.subheader("📁 Upload Local Video")
+    local_upload_btn = st.button("📁 Upload Local Video")
+with col2:
+    youtube_upload_btn = st.button("📺 Upload YouTube Video")
+
+# Upload Local File
+if local_upload_btn:
     video_file = st.file_uploader("Upload your video file", type=["mp4", "mkv", "avi"], key="local-upload")
     if video_file:
         with open("uploaded_video.mp4", "wb") as f:
@@ -61,8 +41,8 @@ with col1:
         st.session_state.video_path = "uploaded_video.mp4"
         st.success("Video uploaded successfully!")
 
-with col2:
-    st.subheader("📺 Upload YouTube Video")
+# Download Video from YouTube
+if youtube_upload_btn:
     youtube_url = st.text_input("Enter YouTube URL", key="youtube-upload")
     if youtube_url:
         try:
@@ -77,35 +57,41 @@ if st.session_state.video_path:
     st.header("Process Your Video")
     st.write(f"Processing {st.session_state.video_path}...")
     
-    def extract_audio(video_path):
-        try:
-            audio = AudioSegment.from_file(video_path)
-            audio.export("extracted_audio.mp3", format="mp3")
-            st.success("Audio extracted successfully!")
-            return "extracted_audio.mp3"
-        except Exception as e:
-            st.error(f"Error in extracting audio: {str(e)}")
-            return None
-
-    audio_path = extract_audio(st.session_state.video_path)
+    process_btn = st.button("🔄 Start Processing")
     
-    def transcribe_audio(audio_path):
-        try:
-            model = whisper.load_model("base")
-            result = model.transcribe(audio_path)
-            st.text_area("Transcription", result['text'], height=200)
-            return result['text']
-        except Exception as e:
-            st.error(f"Error in transcription: {str(e)}")
-            return None
-
-    if audio_path:
-        transcription = transcribe_audio(audio_path)
+    if process_btn:
+        def extract_audio(video_path):
+            try:
+                audio = AudioSegment.from_file(video_path)
+                audio.export("extracted_audio.mp3", format="mp3")
+                st.success("Audio extracted successfully!")
+                return "extracted_audio.mp3"
+            except Exception as e:
+                st.error(f"Error in extracting audio: {str(e)}")
+                return None
+
+        audio_path = extract_audio(st.session_state.video_path)
+        
+        def transcribe_audio(audio_path):
+            try:
+                model = whisper.load_model("base")
+                result = model.transcribe(audio_path)
+                st.text_area("Transcription", result['text'], height=200)
+                return result['text']
+            except Exception as e:
+                st.error(f"Error in transcription: {str(e)}")
+                return None
+
+        if audio_path:
+            transcription = transcribe_audio(audio_path)
 
 # Summarize and Translate Section
 if 'transcription' in locals():
     st.header("Results")
 
+    summarize_btn = st.button("📝 Summarize Text")
+    translate_btn = st.button("🌍 Translate Summary")
+
     def summarize_text(text):
         try:
             summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
@@ -116,7 +102,9 @@ if 'transcription' in locals():
             st.error(f"Error in summarization: {str(e)}")
             return None
 
-    summary = summarize_text(transcription)
+    summary = None
+    if summarize_btn:
+        summary = summarize_text(transcription)
 
     def translate_text(text, src_lang="en", tgt_lang="es"):
         try:
@@ -131,9 +119,8 @@ if 'transcription' in locals():
             st.error(f"Error in translation: {str(e)}")
             return None
 
-    target_language = st.selectbox("Select Translation Language", ["es", "fr", "de", "zh"])
-    if target_language:
+    if summary and translate_btn:
+        target_language = st.selectbox("Select Translation Language", ["es", "fr", "de", "zh"], key="lang-select")
         translated_summary = translate_text(summary, tgt_lang=target_language)
-
 else:
     st.info("Please upload a video to start the process.")