Update streamlit_app.py

streamlit_app.py

@@ -235,57 +235,54 @@ def main():
     with col2:
         st.header("2. Background Settings")
         bg_type = handle_background_selection()
-        
+
         st.header("3. Process & Download")
-        if st.button(
-            "🚀 Process Video", 
-            type="primary", 
-            disabled=not st.session_state.uploaded_video or st.session_state.processing,
-            use_container_width=True
-        ):
-            with st.spinner("Processing video (this may take a few minutes)..."):
+        # ------- FORM WRAP START --------
+        with st.form("process_form"):
+            submitted = st.form_submit_button(
+                "🚀 Process Video",
+                disabled=not st.session_state.uploaded_video or st.session_state.processing
+            )
+
+            process_status = st.empty()
+            if submitted and not st.session_state.processing:
                 st.session_state.processing = True
-                
-                try:
-                    # Prepare background based on type
-                    background = None
-                    if bg_type == "image" and 'bg_image' in st.session_state and st.session_state.bg_image is not None:
-                        background = st.session_state.bg_image
-                    elif bg_type == "color" and 'bg_color' in st.session_state:
-                        background = st.session_state.bg_color
-                    
-                    # Process the video
-                    output_path = process_video(
-                        st.session_state.uploaded_video,
-                        background,
-                        bg_type=bg_type
-                    )
-                    
-                    if output_path and os.path.exists(output_path):
-                        # Store the path to the processed video
-                        st.session_state.processed_video_path = output_path
-                        st.success("✅ Video processing complete!")
-                    else:
-                        st.error("❌ Failed to process video. Please check the logs for details.")
-                
-                except Exception as e:
-                    st.error(f"❌ An error occurred: {str(e)}")
-                    logger.exception("Video processing failed")
-                
-                finally:
-                    st.session_state.processing = False
-        
-        # Show processed video if available
+                with st.spinner("Processing video (this may take a few minutes)..."):
+                    try:
+                        # Prepare background based on type
+                        background = None
+                        if bg_type == "image" and 'bg_image' in st.session_state and st.session_state.bg_image is not None:
+                            background = st.session_state.bg_image
+                        elif bg_type == "color" and 'bg_color' in st.session_state:
+                            background = st.session_state.bg_color
+
+                        # Process the video
+                        output_path = process_video(
+                            st.session_state.uploaded_video,
+                            background,
+                            bg_type=bg_type
+                        )
+
+                        if output_path and os.path.exists(output_path):
+                            st.session_state.processed_video_path = output_path
+                            process_status.success("✅ Video processing complete!")
+                        else:
+                            process_status.error("❌ Failed to process video. Please check the logs for details.")
+
+                    except Exception as e:
+                        process_status.error(f"❌ An error occurred: {str(e)}")
+                        logger.exception("Video processing failed")
+                    finally:
+                        st.session_state.processing = False
+        # ------- FORM WRAP END --------
+
+        # Show processed video if available (outside form, stable)
         if 'processed_video_path' in st.session_state and st.session_state.processed_video_path:
             st.markdown("### Processed Video")
-            
             try:
-                # Display the video directly from the file
                 with open(st.session_state.processed_video_path, 'rb') as f:
                     video_bytes = f.read()
                     st.video(video_bytes)
-                    
-                    # Download button
                     st.download_button(
                         label="💾 Download Processed Video",
                         data=video_bytes,
@@ -298,4 +295,4 @@ def main():
                 logger.error(f"Error displaying video: {str(e)}", exc_info=True)
 
 if __name__ == "__main__":
-    main()
+    main()