Update ui.py

ui.py

@@ -12,14 +12,12 @@
 from PIL import Image
 import numpy as np
 import logging
-from datetime import datetime
 import time
 
 logger = logging.getLogger("Advanced Video Background Replacer")
 
 # --- UI Helpers ---
 def tail_file(path: str, lines: int = 400) -> str:
-    """Read last N lines from a text file efficiently."""
     if not os.path.exists(path):
         return "(log file not found)"
     try:
@@ -30,7 +28,6 @@ def tail_file(path: str, lines: int = 400) -> str:
         return f"(failed to read log: {e})"
 
 def read_file_bytes(path: str) -> bytes:
-    """Read a file as bytes; return b'' if missing/error."""
     try:
         if not os.path.exists(path):
             return b""
@@ -39,7 +36,6 @@ def read_file_bytes(path: str) -> bytes:
     except Exception:
         return b""
 
-# --- Render UI ---
 def render_ui(process_video_func):
     try:
         # --- Sidebar: System Status & Logs ---
@@ -127,7 +123,6 @@ def render_ui(process_video_func):
                 else:
                     bg_preview_placeholder.empty()
                 background = background_pil
-            # --- Background Color Picker ---
             else:
                 selected_color = st.color_picker(
                     "Choose Background Color",
@@ -158,7 +153,8 @@ def render_ui(process_video_func):
                         st.error("Please upload a background image")
                         return
                     with st.spinner("Processing video..."):
-                        success = process_video_func(uploaded_video, background, bg_type)
+                        # NOTE: Always pass lowercased bg_type!
+                        success = process_video_func(uploaded_video, background, bg_type.lower())
                     if success:
                         st.success("Video processing complete!")
                     else: