Update ui/ui_components.py

ui/ui_components.py

@@ -10,6 +10,7 @@
 from __future__ import annotations
 import os
 import gradio as gr
+import logging
 
 from ui.callbacks import (
     cb_load_models,
@@ -22,6 +23,8 @@
     cb_preset_bg_preview,
 )
 
+logger = logging.getLogger(__name__)
+
 # Typography & UI polish: sharper text + cleaner cards
 CSS = """
 :root {
@@ -84,6 +87,30 @@
 GRADIENT_COLOR_CHOICES = ["warm_gradient", "tech_dark"]
 
 
+def get_background_status(bg_method, bg_style, custom_bg_path=None, pro_image=None, gradient=None):
+    """Get current background status text for display"""
+    if bg_method == "Upload file" and custom_bg_path:
+        filename = os.path.basename(custom_bg_path) if custom_bg_path else "none"
+        return f"Background: custom({filename})"
+    elif bg_method == "Pre-loaded professional images" and pro_image:
+        return f"Background: {pro_image}"
+    elif bg_method == "Pre-loaded Gradients / Colors" and gradient:
+        return f"Background: {gradient}"
+    elif bg_method == "AI generated background":
+        return f"Background: AI-generated"
+    elif bg_style:
+        return f"Background: {bg_style}"
+    else:
+        return "Background: none"
+
+
+def on_quality_change(quality_value):
+    """Called when quality dropdown changes - sets environment variable"""
+    os.environ["BFX_QUALITY"] = quality_value.lower()
+    logger.info(f"UI: Quality preference set to {quality_value}")
+    return f"Quality mode: {quality_value} (refine_every={'1' if quality_value=='max' else '3' if quality_value=='balanced' else '10'} frames)"
+
+
 def create_interface() -> gr.Blocks:
     with gr.Blocks(
         title="🎬 BackgroundFX Pro",
@@ -235,6 +262,7 @@ def create_interface() -> gr.Blocks:
                 with gr.Column(scale=1):
                     out_video = gr.Video(label="Processed Output", interactive=False)
                     statusbox = gr.Textbox(label="Status", lines=8, elem_id="statusbox")
+                    quality_status = gr.Textbox(label="Quality Settings", lines=1, value="Quality mode: balanced")
                     with gr.Row():
                         btn_refresh = gr.Button("🔍 Refresh Status", variant="secondary")
                         btn_clear   = gr.Button("🧹 Clear",           variant="secondary")
@@ -276,15 +304,36 @@ def _toggle_bg_sections(choice: str):
         # Load models
         btn_load.click(cb_load_models, outputs=statusbox)
 
+        # Quality dropdown handler - updates environment and status
+        quality.change(
+            on_quality_change,
+            inputs=[quality],
+            outputs=[quality_status]
+        )
+
         # Tiny wrapper to set env for quality before calling the existing callback
         def _run_with_quality(video_pth, bg_style_val, custom_bg_pth,
                               use_two_stage_state, chroma_p, key_mode,
-                              prev_mask, prev_gs, quality_val):
+                              prev_mask, prev_gs, quality_val, bg_method_val, 
+                              pro_image_val, gradient_val):
             os.environ["BFX_QUALITY"] = (quality_val or "balanced")
-            return cb_process_video(
+            
+            # Update status with background info
+            bg_status = get_background_status(
+                bg_method_val, bg_style_val, custom_bg_pth, 
+                pro_image_val, gradient_val
+            )
+            
+            result = cb_process_video(
                 video_pth, bg_style_val, custom_bg_pth,
                 use_two_stage_state, chroma_p, key_mode, prev_mask, prev_gs
             )
+            
+            # Append background status to the result message
+            if isinstance(result, tuple) and len(result) >= 2:
+                video_out, msg = result[0], result[1]
+                return video_out, f"{msg}\n{bg_status}"
+            return result
 
         # Always two-stage: pass use_two_stage=True to callback via State
         btn_run.click(
@@ -299,6 +348,9 @@ def _run_with_quality(video_pth, bg_style_val, custom_bg_pth,
                 preview_mask,
                 preview_greenscreen,
                 quality,               # <-- the new control
+                bg_method,            # Added for status
+                pro_image_dd,         # Added for status
+                gradient_dd,          # Added for status
             ],
             outputs=[out_video, statusbox],
         )
@@ -343,4 +395,4 @@ def _run_with_quality(video_pth, bg_style_val, custom_bg_pth,
             outputs=[custom_bg]
         )
 
-    return demo
+    return demo