Update app.py

app.py

@@ -18,7 +18,7 @@ This app helps you:
 - Process videos with multiple speakers.
 """)
 
-# Temporary video storage
+# Initialize session state
 if "video_path" not in st.session_state:
     st.session_state.video_path = None
 if "transcription" not in st.session_state:
@@ -26,26 +26,18 @@ if "transcription" not in st.session_state:
 
 # Upload Video Section
 st.header("Upload Your Video")
+upload_option = st.radio("Choose upload method:", ["📁 Local File", "📺 YouTube URL"])
 
-# Buttons for navigation
-col1, col2 = st.columns(2)
-with col1:
-    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 upload_option == "📁 Local File":
+    video_file = st.file_uploader("Upload your video file", type=["mp4", "mkv", "avi"])
     if video_file:
         with open("uploaded_video.mp4", "wb") as f:
             f.write(video_file.read())
         st.session_state.video_path = "uploaded_video.mp4"
         st.success("Video uploaded successfully!")
 
-# Download Video from YouTube
-if youtube_upload_btn:
-    youtube_url = st.text_input("Enter YouTube URL", key="youtube-upload")
+elif upload_option == "📺 YouTube URL":
+    youtube_url = st.text_input("Enter YouTube URL")
     if youtube_url:
         try:
             os.system(f"yt-dlp -o video.mp4 {youtube_url}")
@@ -54,14 +46,13 @@ if youtube_upload_btn:
         except Exception as e:
             st.error(f"Error downloading video: {str(e)}")
 
-# Process Video Section
+# Display video
 if st.session_state.video_path:
-    st.header("Process Your Video")
+    st.header("Preview & Process Video")
     st.video(st.session_state.video_path)
-    st.write(f"Processing {st.session_state.video_path}...")
-    
-    process_btn = st.button("🚀 Process Video", key="process-video", help="Click to start processing", use_container_width=True)
-    
+
+    process_btn = st.button("🚀 Process Video", key="process-video", use_container_width=True)
+
     if process_btn:
         def extract_audio(video_path):
             try:
@@ -70,11 +61,9 @@ if st.session_state.video_path:
                 st.success("Audio extracted successfully!")
                 return "extracted_audio.mp3"
             except Exception as e:
-                st.error(f"Error in extracting audio: {str(e)}")
+                st.error(f"Error 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")
@@ -86,10 +75,11 @@ if st.session_state.video_path:
                 st.error(f"Error in transcription: {str(e)}")
                 return None
 
+        audio_path = extract_audio(st.session_state.video_path)
         if audio_path:
             st.session_state.transcription = transcribe_audio(audio_path)
 
-# Summarize and Translate Section
+# Summarization & Translation
 if st.session_state.transcription:
     st.header("Results")
 
@@ -126,5 +116,6 @@ if st.session_state.transcription:
     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.")