Update ui_components.py

ui_components.py

@@ -70,9 +70,7 @@ def create_interface():
     
     # Safe processing function
     def safe_process_video(video_path, bg_method, custom_img, prof_choice, 
-                          grad_type, color1, color2, color3, use_third, 
-                          ai_prompt, ai_style, ai_img, use_two_stage, chroma_preset,
-                          progress: Optional[gr.Progress] = None):
+                          use_two_stage, chroma_preset, progress: Optional[gr.Progress] = None):
         """Safe wrapper for video processing"""
         if not CORE_FUNCTIONS_AVAILABLE:
             return None, "Core processing functions not available", "Error: Core functions not loaded"
@@ -83,6 +81,7 @@ def progress_callback(pct, desc):
                 if progress:
                     progress(pct)
                 print(f"Progress: {pct:.1%} - {desc}")
+                return desc  # Match app.py's expectation
             
             # Call the main processing function
             if bg_method == "professional" and prof_choice:
@@ -90,14 +89,18 @@ def progress_callback(pct, desc):
                     video_path, prof_choice, None, 
                     progress_callback, 
                     use_two_stage=bool(use_two_stage),
-                    chroma_preset=chroma_preset or "standard"
+                    chroma_preset=chroma_preset or "standard",
+                    preview_mask=False,
+                    preview_greenscreen=False
                 )
             elif bg_method == "upload" and custom_img:
                 result = process_video_fixed(
                     video_path, "custom", custom_img, 
                     progress_callback, 
                     use_two_stage=bool(use_two_stage),
-                    chroma_preset=chroma_preset or "standard"
+                    chroma_preset=chroma_preset or "standard",
+                    preview_mask=False,
+                    preview_greenscreen=False
                 )
             else:
                 return None, "Please select a valid background method", "Error: Invalid background selection"
@@ -105,9 +108,8 @@ def progress_callback(pct, desc):
             return result[0], result[1], f"Processing completed successfully"
             
         except Exception as e:
-            error_msg = f"Processing error: {str(e)}"
+            error_msg = f"Processing error: {str(e)}\n{traceback.format_exc()}"
             print(f"Error in safe_process_video: {error_msg}")
-            traceback.print_exc()
             return None, error_msg, f"Error: {error_msg}"
     
     # Safe model loading function
@@ -121,12 +123,13 @@ def progress_callback(pct, desc):
                 if progress:
                     progress(pct)
                 print(f"Model loading: {pct:.1%} - {desc}")
+                return desc  # Match app.py's expectation
             
             result = load_models_with_validation(progress_callback)
             return result, f"Model loading completed"
             
         except Exception as e:
-            error_msg = f"Model loading error: {str(e)}"
+            error_msg = f"Model loading error: {str(e)}\n{traceback.format_exc()}"
             print(f"Error in safe_load_models: {error_msg}")
             return error_msg, error_msg
     
@@ -220,6 +223,14 @@ def update_visibility(method):
                     interactive=False,
                     lines=3
                 )
+                
+                # Add model and cache status buttons
+                with gr.Row():
+                    model_status_btn = gr.Button("Check Model Status", variant="secondary")
+                    cache_status_btn = gr.Button("Check Cache Status", variant="secondary")
+                
+                model_status = gr.JSON(label="Model Status", interactive=False)
+                cache_status = gr.JSON(label="Cache Status", interactive=False)
             
             with gr.Column(scale=1):
                 gr.Markdown("### 🎥 Result")
@@ -253,14 +264,6 @@ def update_visibility(method):
                 background_method,
                 custom_background,
                 professional_choice,
-                gr.State("vertical"),  # grad_type placeholder
-                gr.State("#3498db"),   # color1 placeholder
-                gr.State("#2ecc71"),   # color2 placeholder
-                gr.State("#e74c3c"),   # color3 placeholder
-                gr.State(False),       # use_third placeholder
-                gr.State(""),          # ai_prompt placeholder
-                gr.State("photorealistic"), # ai_style placeholder
-                gr.State(None),        # ai_img placeholder
                 use_two_stage,
                 chroma_preset
             ],
@@ -268,6 +271,18 @@ def update_visibility(method):
             show_progress=True
         )
         
+        model_status_btn.click(
+            fn=get_model_status,
+            outputs=[model_status],
+            show_progress=False
+        )
+        
+        cache_status_btn.click(
+            fn=get_cache_status,
+            outputs=[cache_status],
+            show_progress=False
+        )
+        
         # Info section
         with gr.Accordion("ℹ️ Information", open=False):
             gr.Markdown(f"""
@@ -282,13 +297,4 @@ def update_visibility(method):
             - Professional backgrounds are optimized for best results
             """)
     
-    return demo
-
-if __name__ == "__main__":
-    # Test the interface creation
-    try:
-        demo = create_interface()
-        demo.launch(share=True)
-    except Exception as e:
-        print(f"Failed to create interface: {e}")
-        traceback.print_exc()
+    return demo