new update fix

app.py

@@ -838,6 +838,7 @@ def minecraftify(
     return _minecraftify_image(image, steps, guidance_scale, seed, extra_details)
 
 
+@spaces.GPU(duration=120)
 def _minecraftify_live_frame(
     frame: Image.Image,
     steps: int,
@@ -845,8 +846,8 @@ def _minecraftify_live_frame(
     seed: int,
     extra_details: str,
 ):
-    """Inference for individual live frames (GPU held at loop level)."""
-    log_event(f"Processing live frame with steps={steps}")
+    """GPU-wrapped inference for individual live frames."""
+    log_event(f"GPU-allocated: Processing live frame with steps={steps}")
     return _minecraftify_image(frame, steps, guidance_scale, seed, extra_details)
 
 
@@ -869,14 +870,13 @@ def capture_live_frame(image: Image.Image):
     return None
 
 
-def _minecraftify_live_loop_gpu(
+def minecraftify_live_loop(
     steps: int,
     guidance_scale: float,
     seed: int,
     extra_details: str,
 ):
-    """GPU-held loop for live processing - holds GPU for 120 seconds."""
-    log_event("Live GPU loop started (GPU allocated for 120s)")
+    log_event("Live GPU loop started")
     last_processed_frame_id = 0
 
     while True:
@@ -903,24 +903,13 @@ def _minecraftify_live_loop_gpu(
         yield output
 
 
-@spaces.GPU(duration=120)
-def minecraftify_live_loop(
-    steps: int,
-    guidance_scale: float,
-    seed: int,
-    extra_details: str,
-):
-    """Wrapper that holds GPU for entire 120-second live stream session."""
-    yield from _minecraftify_live_loop_gpu(steps, guidance_scale, seed, extra_details)
-
-
 def _set_live_active(image: Image.Image):
     """Auto-enable live processing when recording starts."""
     global _live_active
     if image is not None and not _live_active:
         _live_active = True
         log_event("Live recording detected, auto-starting processing")
-    return None
+    return gr.update(active=_live_active)  # Enable timer when active
 
 
 def switch_input_mode(mode: str):
@@ -938,6 +927,7 @@ def switch_input_mode(mode: str):
         gr.update(interactive=True),
         gr.update(visible=not image_mode, interactive=False),
         False,
+        gr.update(active=not image_mode),  # Deactivate timer in image mode
     )
 
 
@@ -1082,6 +1072,7 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
 
     # ── Event binding ──
     live_running = gr.State(False)
+    processing_timer = gr.Timer(interval=0.1, active=False)
 
     input_mode.change(
         fn=switch_input_mode,
@@ -1091,6 +1082,7 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
             run_btn,
             live_input_group,
             live_running,
+            processing_timer,
         ],
         queue=False,
     )
@@ -1102,13 +1094,15 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
     )
 
     live_image.stream(
-        fn=lambda img: _set_live_active(img),
+        fn=_set_live_active,
         inputs=live_image,
-        outputs=None,
+        outputs=processing_timer,
         stream_every=0.05,
         trigger_mode="always_last",
         queue=False,
-    ).then(
+    )
+
+    processing_timer.tick(
         fn=minecraftify_live_loop,
         inputs=[steps, guidance_scale, seed, extra_details],
         outputs=[output_image],