new update fix

app.py

@@ -646,11 +646,11 @@ BASE_MODEL_ID = "black-forest-labs/FLUX.2-klein-4B"
 LORA_ID = "AnimeOverlord/flux2-klein-4b-mc"
 
 MODEL_CACHE_ROOT = os.environ.get(
-    "MODEL_CACHE_ROOT",
-    "/data/minecraftify" if os.path.isdir("/data") else ".",
+    "HF_HOME",
+    os.path.join(os.path.expanduser("~"), ".cache", "huggingface"),
 )
-MODEL_DIR = os.path.join(MODEL_CACHE_ROOT, "black-forest-labs", "FLUX.2-klein-4B")
-LORA_DIR = os.path.join(MODEL_CACHE_ROOT, "AnimeOverlord", "flux2-klein-4b-mc-v2")
+MODEL_DIR = os.path.join(MODEL_CACHE_ROOT, "hub", "models--black-forest-labs--FLUX.2-klein-4B")
+LORA_DIR = os.path.join(MODEL_CACHE_ROOT, "hub", "models--AnimeOverlord--flux2-klein-4b-mc")
 
 _pipe = None
 _pipe_ready = False
@@ -689,7 +689,7 @@ def _ensure_model_files():
     os.makedirs(MODEL_DIR, exist_ok=True)
     os.makedirs(LORA_DIR, exist_ok=True)
 
-    # Download base model once into the volume
+    # Download base model on first run
     if not os.path.exists(os.path.join(MODEL_DIR, "model_index.json")):
         log_event(f"Downloading base model: {BASE_MODEL_ID}")
         snapshot_download(
@@ -697,10 +697,11 @@ def _ensure_model_files():
             local_dir=MODEL_DIR,
             local_dir_use_symlinks=False,
         )
+        log_event("Base model download complete")
     else:
-        log_event("Base model already present")
+        log_event("Base model already cached")
 
-    # Download LoRA once into the volume
+    # Download LoRA on first run
     if not os.path.exists(os.path.join(LORA_DIR, "pytorch_lora_weights.safetensors")):
         log_event(f"Downloading LoRA: {LORA_ID}")
         snapshot_download(
@@ -709,8 +710,9 @@ def _ensure_model_files():
             local_dir_use_symlinks=False,
             allow_patterns=["pytorch_lora_weights.safetensors"],
         )
+        log_event("LoRA download complete")
     else:
-        log_event("LoRA already present")
+        log_event("LoRA already cached")
 
 
 def _load_pipe():
@@ -790,7 +792,10 @@ def _minecraftify_image(
         )
 
     log_event("Inference complete")
-    return out.images[0]
+    output_img = out.images[0]
+    log_event(f"Output image size: {output_img.size}")
+    torch.cuda.empty_cache()
+    return output_img
 
 
 @spaces.GPU(duration=120)
@@ -874,6 +879,15 @@ def minecraftify_live_loop(
         yield output
 
 
+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
+
+
 def switch_input_mode(mode: str):
     global _live_active
 
@@ -892,37 +906,6 @@ def switch_input_mode(mode: str):
     )
 
 
-def start_live(image: Image.Image):
-    global _live_active
-
-    with _live_lock:
-        has_captured_frame = _live_latest_frame is not None
-
-    if not has_captured_frame:
-        log_event("Start live blocked: webcam is not on or no streamed frame was captured")
-        raise gr.Error("📷 Webcam is not streaming yet. Click Record and wait for the preview before starting live.")
-
-    _live_active = True
-    log_event(f"Live mode started with webcam frame size={getattr(image, 'size', None)}")
-    return (
-        True,
-        gr.update(interactive=False),
-        gr.update(interactive=True),
-    )
-
-
-def stop_live():
-    global _live_active
-
-    _live_active = False
-    log_event("Live mode stopped")
-    return (
-        False,
-        gr.update(interactive=True),
-        gr.update(interactive=False),
-    )
-
-
 # ----------------------------------------------------------------------
 # UI
 # ----------------------------------------------------------------------
@@ -976,18 +959,7 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
                         elem_classes="image-upload mc-upload-area",
                         height=400,
                     )
-
-                    with gr.Row():
-                        start_live_btn = gr.Button(
-                            "▶ START LIVE",
-                            variant="primary",
-                            elem_classes="mc-btn-primary",
-                        )
-                        stop_live_btn = gr.Button(
-                            "■ STOP LIVE",
-                            variant="secondary",
-                            interactive=False,
-                        )
+                    gr.HTML('<div style="color: var(--mc-gold); font-family: VT323; text-align: center; margin-top: 10px;">▶ Click Record to start capturing and auto-process live frames</div>')
 
                 extra_details = gr.Textbox(
                     label="🗒️ Extra scene details (optional)",
@@ -1069,8 +1041,6 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
             image_input_group,
             run_btn,
             live_input_group,
-            start_live_btn,
-            stop_live_btn,
             live_running,
         ],
         queue=False,
@@ -1082,32 +1052,25 @@ with gr.Blocks(**BLOCKS_KWARGS) as demo:
         outputs=output_image,
     )
 
-    start_live_event = start_live_btn.click(
-        fn=start_live,
+    live_image.stream(
+        fn=lambda img: _set_live_active(img),
         inputs=live_image,
-        outputs=[live_running, start_live_btn, stop_live_btn],
+        outputs=None,
+        stream_every=0.05,
+        trigger_mode="always_last",
         queue=False,
-    )
-
-    start_live_event.then(
+    ).then(
         fn=minecraftify_live_loop,
         inputs=[steps, guidance_scale, seed, extra_details],
         outputs=[output_image],
         concurrency_limit=1,
     )
 
-    stop_live_btn.click(
-        fn=stop_live,
-        inputs=None,
-        outputs=[live_running, start_live_btn, stop_live_btn],
-        queue=False,
-    )
-
     live_image.stream(
         fn=capture_live_frame,
         inputs=live_image,
         outputs=None,
-        stream_every=0.1,
+        stream_every=0.05,
         trigger_mode="always_last",
         queue=False,
     )