Update ui.py

ui.py

@@ -4,6 +4,7 @@
 - Clean, responsive layout with Streamlit
 - File uploaders, previews, and processing controls
 - Log viewer and download buttons
+- Passes proper PIL.Image or color string into pipeline
 """
 import streamlit as st
 import os
@@ -13,6 +14,7 @@
 import logging
 from datetime import datetime
 import time
+
 logger = logging.getLogger("Advanced Video Background Replacer")
 
 # --- UI Helpers ---
@@ -44,8 +46,6 @@ def render_ui(process_video_func):
         with st.sidebar:
             st.subheader("System Status")
             st.markdown("**Log file:** `/tmp/app.log`")
-            # (log-level selector has been REMOVED, handled in streamlit_app.py)
-
             if st.session_state.get('gpu_available', False):
                 try:
                     import torch
@@ -55,10 +55,8 @@ def render_ui(process_video_func):
                 st.success(f"GPU: {dev}")
             else:
                 st.error("GPU: Not Available")
-            # Log viewer 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")
-            # Download logs button
             log_bytes = read_file_bytes("/tmp/app.log")
             st.download_button(
                 "Download Logs",
@@ -68,7 +66,6 @@ def render_ui(process_video_func):
                 use_container_width=True,
                 disabled=not bool(log_bytes)
             )
-            # Log tail viewer
             with st.expander("View Log Tail", expanded=True):
                 if st.session_state.get('auto_refresh_logs', False):
                     time.sleep(2)
@@ -107,7 +104,8 @@ def render_ui(process_video_func):
                 horizontal=True,
                 key="bg_type_radio"
             )
-            # Background Image Upload
+            background = None
+            # --- Background Image Upload ---
             if bg_type == "Image":
                 bg_image = st.file_uploader(
                     "Upload Background Image",
@@ -115,17 +113,21 @@ def render_ui(process_video_func):
                     key="bg_image_uploader"
                 )
                 bg_preview_placeholder = st.empty()
+                background_pil = None
                 if bg_image is not None:
                     try:
-                        bg_image_pil = Image.open(bg_image)
+                        bg_image.seek(0)
+                        background_pil = Image.open(bg_image).convert("RGB")
                         with bg_preview_placeholder.container():
-                            st.image(bg_image_pil, caption="Selected Background", use_column_width=True)
+                            st.image(background_pil, caption="Selected Background", use_column_width=True)
                     except Exception as e:
                         logger.error(f"[UI] Background image error: {e}", exc_info=True)
                         bg_preview_placeholder.error(f"Cannot display image: {e}")
+                        background_pil = None
                 else:
                     bg_preview_placeholder.empty()
-            # Background Color Picker
+                background = background_pil
+            # --- Background Color Picker ---
             else:
                 selected_color = st.color_picker(
                     "Choose Background Color",
@@ -145,22 +147,22 @@ def render_ui(process_video_func):
                         st.image(st.session_state.color_display_cache, caption="Selected Color", width=200)
                 else:
                     color_preview_placeholder.empty()
+                background = selected_color  # "#RRGGBB"
             # --- Process Button ---
             st.header("3. Process Video")
-            can_process = (uploaded_video is not None and not st.session_state.get('processing', False))
+            can_process = (uploaded_video is not None and not st.session_state.get('processing', False) and (background is not None))
             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 bg_image is None:
+                    if bg_type == "Image" and background is None:
                         st.error("Please upload a background image")
+                        return
+                    with st.spinner("Processing video..."):
+                        success = process_video_func(uploaded_video, background, bg_type)
+                    if success:
+                        st.success("Video processing complete!")
                     else:
-                        background = bg_image if bg_type == "Image" else st.session_state.get('bg_color')
-                        with st.spinner("Processing video..."):
-                            success = process_video_func(uploaded_video, bg_image, background, bg_type)
-                        if success:
-                            st.success("Video processing complete!")
-                        else:
-                            st.error("Video processing failed. Check logs for details.")
+                        st.error("Video processing failed. Check logs for details.")
                 except Exception as e:
                     logger.error(f"[UI] Process video error: {e}", exc_info=True)
                     st.error(f"Processing error: {str(e)}. Check logs for details.")