Update ui_components.py

ui_components.py

@@ -14,7 +14,7 @@
 from typing import Optional
 
 # Import core processing functions
-from app import load_models_with_validation, process_video_fixed, get_model_status, get_cache_status
+from app import load_models_with_validation, process_video_fixed, get_model_status, get_cache_status, PROCESS_CANCELLED
 from utilities import PROFESSIONAL_BACKGROUNDS, create_procedural_background
 
 # Check if two-stage is available
@@ -34,19 +34,20 @@
 # ============================================================================ #
 # GRADIO MONKEY PATCH (BUG FIX for gradio>=4.44.0)
 # ============================================================================ #
-try:
-    import gradio_client.utils as gc_utils
-    original_get_type = gc_utils.get_type
-    def patched_get_type(schema):
-        if not isinstance(schema, dict):
-            if isinstance(schema, bool):
-                return "boolean"
-            return "string"
-        return original_get_type(schema)
-    gc_utils.get_type = patched_get_type
-    logger.info("Applied Gradio schema validation monkey patch.")
-except (ImportError, AttributeError) as e:
-    logger.warning(f"Could not apply Gradio monkey patch: {e}")
+# Comment out if not needed in current Gradio version
+# try:
+#     import gradio_client.utils as gc_utils
+#     original_get_type = gc_utils.get_type
+#     def patched_get_type(schema):
+#         if not isinstance(schema, dict):
+#             if isinstance(schema, bool):
+#                 return "boolean"
+#             return "string"
+#         return original_get_type(schema)
+#     gc_utils.get_type = patched_get_type
+#     logger.info("Applied Gradio schema validation monkey patch.")
+# except (ImportError, AttributeError) as e:
+#     logger.warning(f"Could not apply Gradio monkey patch: {e}")
 
 # ============================================================================ #
 # DEBUG LOGGING FUNCTIONS
@@ -96,8 +97,7 @@ def update_cache_status():
         return f"Status check error: {str(e)}"
 
 def on_two_stage_change(value):
-    """Debug handler for two-stage checkbox changes"""
-    add_debug_log(f"Two-stage checkbox changed to: {value} (type: {type(value)})")
+    """Handler for two-stage checkbox changes"""
     return gr.update(visible=value)
 
 # ============================================================================ #
@@ -125,14 +125,8 @@ def process_video_enhanced_fixed(
         add_debug_log("ERROR: No video file provided")
         return None, "No video file provided.", get_debug_log()
     
-    # Log checkbox value for debugging
-    add_debug_log(f"CHECKBOX RAW VALUE: use_two_stage = {use_two_stage}")
-    add_debug_log(f"CHECKBOX TYPE: {type(use_two_stage)}")
-    add_debug_log(f"CHECKBOX BOOL CONVERSION: {bool(use_two_stage)}")
-    
     # Force boolean conversion to be safe
     use_two_stage = bool(use_two_stage)
-    add_debug_log(f"FINAL use_two_stage value: {use_two_stage}")
     
     def progress_callback(pct, desc):
         # Update Gradio progress bar
@@ -394,7 +388,7 @@ def create_interface():
                         visible=False
                     )
                     
-                    # Debug checkbox changes
+                    # Handler for checkbox
                     use_two_stage.change(
                         fn=on_two_stage_change,
                         inputs=use_two_stage,
@@ -405,6 +399,7 @@ def create_interface():
                 with gr.Row():
                     load_models_btn = gr.Button("Load Models", variant="secondary")
                     process_btn = gr.Button("Process Video", variant="primary", scale=2)
+                    cancel_btn = gr.Button("Cancel Processing", variant="stop")
 
                 status_text = gr.Textbox(
                     label="Status", 
@@ -473,6 +468,12 @@ def create_interface():
             show_progress=True
         )
         
+        # Cancel processing
+        cancel_btn.click(
+            fn=lambda: (PROCESS_CANCELLED.set(), "Processing cancelled"),
+            outputs=[status_text]
+        )
+        
         # Refresh processing info manually
         refresh_info_btn.click(
             fn=refresh_processing_info,