Update ui.py

ui.py

@@ -3,7 +3,6 @@
 Modern UI for Video Background Replacer (PRO)
 - Clean, responsive Streamlit layout
 - Shows live stage progress and all errors during processing
-- Robust to missing stock images (won't crash the app)
 """
 import streamlit as st
 import os
@@ -15,8 +14,6 @@
 
 logger = logging.getLogger("Advanced Video Background Replacer")
 
-LOG_PATH = "/tmp/app.log"
-
 def tail_file(path: str, lines: int = 400) -> str:
     if not os.path.exists(path):
         return "(log file not found)"
@@ -37,22 +34,20 @@ def read_file_bytes(path: str) -> bytes:
         return b""
 
 def _render_background_settings():
-    """
-    Returns:
-        background: PIL.Image (RGB) for 'Image'/'Stock', or "#RRGGBB" for 'Color', or generated PIL for 'AI Prompt'
-        bg_type: one of {'Image','Color','Stock','AI Prompt'}
-    """
+    # Available stock images (replace with your real logic or API)
+    stock_images = {
+        "Sunset Beach": "stock_images/sunset_beach.jpg",
+        "Urban Office": "stock_images/urban_office.jpg",
+        "Studio Lighting": "stock_images/studio_light.jpg",
+    }
     st.header("2. Background Settings")
-
     bg_type = st.radio(
         "Select Background Type:",
         ["Image", "Color", "Stock", "AI Prompt"],
         horizontal=True,
         key="bg_type_radio"
     )
-
     background = None
-    app_dir = Path(__file__).resolve().parent
 
     if bg_type == "Image":
         bg_image = st.file_uploader(
@@ -61,69 +56,49 @@ def _render_background_settings():
             key="bg_image_uploader"
         )
         if bg_image is not None:
-            try:
-                bg_image.seek(0)
-                background = Image.open(bg_image).convert("RGB")
-                st.image(background, caption="Selected Background", use_column_width=True)
-            except Exception as e:
-                logger.error(f"[UI] Failed to open uploaded background: {e}", exc_info=True)
-                st.error(f"Could not open uploaded image: {e}")
+            bg_image.seek(0)
+            background = Image.open(bg_image).convert("RGB")
+            st.image(background, caption="Selected Background", use_column_width=True)
 
     elif bg_type == "Color":
-        default_color = st.session_state.get('bg_color', "#00FF00")
-        selected_color = st.color_picker("Choose Background Color", default_color, key="color_picker")
+        selected_color = st.color_picker(
+            "Choose Background Color",
+            st.session_state.get('bg_color', "#00FF00"),
+            key="color_picker"
+        )
         background = selected_color
-        try:
-            rgb = tuple(int(selected_color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4))
-            color_preview = np.full((100, 100, 3), rgb, dtype=np.uint8)
-            st.image(color_preview, caption="Selected Color", width=200)
-        except Exception as e:
-            logger.error(f"[UI] Color preview error: {e}", exc_info=True)
-            st.warning("Invalid color value.")
+        color_preview = np.full(
+            (100, 100, 3),
+            tuple(int(selected_color.lstrip('#')[i:i+2], 16) for i in (0, 2, 4)),
+            dtype=np.uint8
+        )
+        st.image(color_preview, caption="Selected Color", width=200)
 
     elif bg_type == "Stock":
-        # Resolve stock image paths relative to this file; filter to ones that actually exist.
-        stock_dir = app_dir / "stock_images"
-        stock_candidates = {
-            "Sunset Beach": stock_dir / "sunset_beach.jpg",
-            "Urban Office": stock_dir / "urban_office.jpg",
-            "Studio Lighting": stock_dir / "studio_light.jpg",
-        }
-        available = {k: v for k, v in stock_candidates.items() if v.exists()}
-
-        if not stock_dir.exists():
-            st.warning(f"Stock images folder not found: {stock_dir}")
-        if not available:
-            st.info("No stock images found on disk. Upload an image, choose a Color, or use AI Prompt.")
-        else:
-            stock_choice = st.selectbox(
-                "Choose a professional stock background:",
-                list(available.keys()),
-                key="stock_image_select"
-            )
-            selected_path = available.get(stock_choice)
-            if selected_path is not None:
-                try:
-                    background = Image.open(selected_path).convert("RGB")
-                    st.image(background, caption=stock_choice, use_column_width=True)
-                except Exception as e:
-                    logger.error(f"[UI] Failed to open stock image '{selected_path}': {e}", exc_info=True)
-                    st.error(f"Could not open stock image: {selected_path}")
+        stock_choice = st.selectbox(
+            "Choose a professional stock background:",
+            list(stock_images.keys()),
+            key="stock_image_select"
+        )
+        stock_img_path = stock_images[stock_choice]
+        try:
+            background = Image.open(stock_img_path).convert("RGB")
+            st.image(background, caption=stock_choice, use_column_width=True)
+        except FileNotFoundError:
+            st.warning(f"Stock image not found: {stock_img_path}. Upload your own image or choose another.")
+            background = None
 
     elif bg_type == "AI Prompt":
         prompt = st.text_input("Describe the background to generate (AI):", key="ai_bg_prompt")
         ai_ready = False
         if st.button("Generate Background", key="gen_bg_btn") and prompt:
             # TODO: Plug in real AI image generator
-            try:
-                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!")
-                ai_ready = True
-            except Exception as e:
-                logger.error(f"[UI] AI generator stub failed: {e}", exc_info=True)
-                st.error(f"AI generation failed: {e}")
-        elif "generated_bg" in st.session_state and isinstance(st.session_state.generated_bg, Image.Image):
+            # 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!")
+            ai_ready = True
+        elif "generated_bg" in st.session_state:
             background = st.session_state.generated_bg
             ai_ready = True
         if ai_ready and background is not None:
@@ -136,9 +111,7 @@ def render_ui(process_video_func):
         # --- Sidebar: System Status & Logs ---
         with st.sidebar:
             st.subheader("System Status")
-            st.markdown(f"**Log file:** `{LOG_PATH}`")
-
-            # GPU status (best-effort)
+            st.markdown("**Log file:** `/tmp/app.log`")
             if st.session_state.get('gpu_available', False):
                 try:
                     import torch
@@ -149,10 +122,12 @@ 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")
             st.number_input("Tail last N lines", min_value=50, max_value=5000, step=50, key="log_tail_lines")
 
-            log_bytes = read_file_bytes(LOG_PATH)
+            # Download logs
+            log_bytes = read_file_bytes("/tmp/app.log")
             st.download_button(
                 "Download Logs",
                 data=log_bytes if log_bytes else b"Log file not available yet.",
@@ -162,13 +137,18 @@ def render_ui(process_video_func):
                 disabled=not bool(log_bytes)
             )
 
+            # Log tail
             with st.expander("View Log Tail", expanded=True):
-                if st.session_state.get('auto_refresh_logs', False):
+                # 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()
-                log_text = tail_file(LOG_PATH, st.session_state.get('log_tail_lines', 400))
-                # Use a text_area to ensure scrolling works consistently in Spaces
-                st.text_area("Log tail", value=log_text, height=400, label_visibility="collapsed")
+
+                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 ---
         col1, col2 = st.columns([1, 1], gap="large")
@@ -198,10 +178,8 @@ def render_ui(process_video_func):
         # --- Column 2: Background & Processing ---
         with col2:
             background, bg_type = _render_background_settings()
-
             st.header("3. Process Video")
             status_box = st.empty()  # <--- Status/progress area
-
             can_process = (
                 uploaded_video is not None
                 and not st.session_state.get('processing', False)
@@ -212,6 +190,11 @@ 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")
@@ -256,7 +239,6 @@ def progress_callback(msg):
                 if st.button("Clear Error"):
                     st.session_state.last_error = None
                     st.rerun()
-
     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.")