Update ui_manager.py

ui_manager.py

@@ -223,10 +223,8 @@ class UIManager:
             return None, None, None, "Please describe the background scene you'd like!", gr.update(visible=False)
 
         try:
-            def progress_callback(msg: str, pct: int):
-                progress(pct / 100, desc=msg)
-
             # Delegate to GPUHandlers
+            # Note: progress_callback removed for ZeroGPU compatibility (can't pickle local functions)
             result = self.gpu_handlers.background_generate(
                 image=uploaded_image,
                 prompt=prompt,
@@ -235,7 +233,7 @@ class UIManager:
                 focus_mode=focus_mode,
                 num_steps=int(steps),
                 guidance_scale=float(guidance),
-                progress_callback=progress_callback
+                progress_callback=None  # ZeroGPU doesn't support local function callbacks
             )
 
             if result["success"]:
@@ -784,9 +782,6 @@ class UIManager:
         if coverage > 0.95:
             return None, None, "⚠️ Mask too large - consider using background generation instead", -1
 
-        def progress_callback(msg: str, pct: int):
-            progress(pct / 100, desc=msg)
-
         try:
             # Get template key if selected
             template_key = None
@@ -796,6 +791,7 @@ class UIManager:
                 )
 
             # Delegate to GPUHandlers
+            # Note: progress_callback removed for ZeroGPU compatibility (can't pickle local functions)
             result_image, control_image, status, used_seed = self.gpu_handlers.inpainting_generate(
                 image=image,
                 mask=mask,
@@ -808,7 +804,7 @@ class UIManager:
                 guidance_scale=guidance_scale,
                 num_steps=num_steps,
                 seed=int(seed) if seed is not None else -1,
-                progress_callback=progress_callback
+                progress_callback=None  # ZeroGPU doesn't support local function callbacks
             )
 
             # Store in history if successful