Update ui.py

ui.py

@@ -3,6 +3,7 @@
 Modern UI for Video Background Replacer (PRO)
 - Clean, responsive Streamlit layout
 - Shows live stage progress and all errors during processing
+- NO automatic reruns (no st.rerun) to avoid killing long-running jobs
 """
 import streamlit as st
 import os
@@ -13,6 +14,7 @@
 import time
 
 logger = logging.getLogger("Advanced Video Background Replacer")
+UI_BUILD = "ui-2025-10-04-10-30Z"  # visible in sidebar so you know this file is active
 
 def tail_file(path: str, lines: int = 400) -> str:
     if not os.path.exists(path):
@@ -34,7 +36,7 @@ def read_file_bytes(path: str) -> bytes:
         return b""
 
 def _render_background_settings():
-    # Available stock images (replace with your real logic or API)
+    # Available stock images (may not exist in Space)
     stock_images = {
         "Sunset Beach": "stock_images/sunset_beach.jpg",
         "Urban Office": "stock_images/urban_office.jpg",
@@ -93,7 +95,6 @@ def _render_background_settings():
         ai_ready = False
         if st.button("Generate Background", key="gen_bg_btn") and prompt:
             # TODO: Plug in real AI image generator
-            # For now, stub with a purple placeholder
             background = Image.new("RGB", (512, 320), (64, 32, 96))
             st.session_state.generated_bg = background
             st.success("AI-generated background (stub). Replace with your generator!")
@@ -111,7 +112,10 @@ def render_ui(process_video_func):
         # --- Sidebar: System Status & Logs ---
         with st.sidebar:
             st.subheader("System Status")
+            st.caption(f"UI build: {UI_BUILD}")
             st.markdown("**Log file:** `/tmp/app.log`")
+
+            # GPU indicator (best-effort)
             if st.session_state.get('gpu_available', False):
                 try:
                     import torch
@@ -122,11 +126,8 @@ def render_ui(process_video_func):
             else:
                 st.error("GPU: Not Available")
 
-            # Controls
-            st.checkbox("Auto-refresh log tail (every 2s)", key="auto_refresh_logs")
+            # Log tail controls (NO auto refresh, only manual button)
             st.number_input("Tail last N lines", min_value=50, max_value=5000, step=50, key="log_tail_lines")
-
-            # Download logs
             log_bytes = read_file_bytes("/tmp/app.log")
             st.download_button(
                 "Download Logs",
@@ -136,18 +137,10 @@ def render_ui(process_video_func):
                 use_container_width=True,
                 disabled=not bool(log_bytes)
             )
-
-            # Log tail
             with st.expander("View Log Tail", expanded=True):
-                # IMPORTANT: do NOT rerun while a job is active
-                if st.session_state.get('auto_refresh_logs', False) and st.session_state.get('processing', False):
-                    st.info("⏸ Auto-refresh paused while processing…")
-                elif st.session_state.get('auto_refresh_logs', False):
-                    time.sleep(2)
-                    st.rerun()
-
+                if st.button("Refresh log", use_container_width=True, key="refresh_log_btn"):
+                    st.session_state['_last_log_refresh'] = time.time()
                 log_text = tail_file("/tmp/app.log", st.session_state.get('log_tail_lines', 400))
-                # Make sure this area scrolls and doesn't expand the page infinitely
                 st.code(log_text, language="text")
 
         # --- Main UI: Two-Column Layout ---
@@ -167,6 +160,7 @@ def render_ui(process_video_func):
                 try:
                     uploaded_video.seek(0)
                     video_bytes = uploaded_video.read()
+                    st.session_state.video_bytes_cache = video_bytes  # persist across reruns
                     with video_preview_placeholder.container():
                         st.video(video_bytes)
                 except Exception as e:
@@ -179,9 +173,10 @@ def render_ui(process_video_func):
         with col2:
             background, bg_type = _render_background_settings()
             st.header("3. Process Video")
-            status_box = st.empty()  # <--- Status/progress area
+            status_box = st.empty()  # progress area
+
             can_process = (
-                uploaded_video is not None
+                st.session_state.get('video_bytes_cache') is not None
                 and not st.session_state.get('processing', False)
                 and (background is not None)
             )
@@ -190,20 +185,29 @@ def progress_callback(msg):
                 status_box.info(msg)
                 logger.info(f"[STAGE] {msg}")
 
-            # Don’t allow starting if auto-refresh is on (optional safety)
-            if st.session_state.get('auto_refresh_logs', False):
-                st.warning("Disable **Auto-refresh log tail** in the sidebar before processing.")
-                can_process = False
-
             if st.button("Process Video", disabled=not can_process, use_container_width=True):
                 try:
                     logger.info("Process Video button clicked")
-                    if bg_type == "Image" and background is None:
-                        st.error("Please upload a background image")
-                        return
+
+                    # Rebuild a file-like object from cached bytes so we don't depend on the upload widget object
+                    import io
+                    class _MemFile:
+                        def __init__(self, name, data):
+                            self.name = name
+                            self._b = io.BytesIO(data)
+                        def read(self):
+                            self._b.seek(0)
+                            return self._b.read()
+                        def seek(self, pos):
+                            self._b.seek(pos)
+
+                    st.session_state.processing = True
+                    mem_video = _MemFile("uploaded.mp4", st.session_state.video_bytes_cache)
+
                     with st.spinner("Processing video..."):
-                        # bg_type always lowercase for pipeline
-                        success = process_video_func(uploaded_video, background, bg_type.lower(), progress_callback)
+                        success = process_video_func(mem_video, background, bg_type.lower(), progress_callback)
+
+                    st.session_state.processing = False
                     if success:
                         status_box.success("✅ Video processing complete!")
                         st.success("Video processing complete!")
@@ -211,6 +215,7 @@ def progress_callback(msg):
                         status_box.error("❌ Video processing failed. Check logs for details.")
                         st.error("Video processing failed. Check logs for details.")
                 except Exception as e:
+                    st.session_state.processing = False
                     logger.error(f"[UI] Process video error: {e}", exc_info=True)
                     status_box.error(f"Processing error: {str(e)}. Check logs for details.")
                     st.error(f"Processing error: {str(e)}. Check logs for details.")
@@ -238,7 +243,7 @@ def progress_callback(msg):
                 st.error(st.session_state.last_error)
                 if st.button("Clear Error"):
                     st.session_state.last_error = None
-                    st.rerun()
+                    # No st.rerun() — the next interaction will refresh naturally
     except Exception as e:
         logger.error(f"[UI] Render UI error: {e}", exc_info=True)
         st.error(f"UI rendering error: {str(e)}. Check logs for details.")