ÆÆ

app.py

@@ -223,17 +223,15 @@ def _post_launch_diag():
         logger.warning("Post-launch CUDA diag failed: %s", e)
 
 # ---------------------------------------------------------------------
-# UI factory (your existing UI builder)
+# UI factory (uses your existing builder)
 # ---------------------------------------------------------------------
 def build_ui() -> gr.Blocks:
-    from ui import create_interface  # your module
+    # FIX: import from ui_core_interface (not from ui)
+    from ui_core_interface import create_interface
     return create_interface()
 
 # ---------------------------------------------------------------------
 # Optional: custom FastAPI mount mode
-#   Why: if Gradio’s internal FastAPI route schema generation glitches
-#   under certain dependency combos, we can run our own FastAPI app,
-#   mount Gradio, and provide a known-good /config shim.
 # ---------------------------------------------------------------------
 def build_fastapi_with_gradio(demo: gr.Blocks):
     """
@@ -245,26 +243,22 @@ def build_fastapi_with_gradio(demo: gr.Blocks):
 
     app = FastAPI(title="VideoBackgroundReplacer2")
 
-    # Simple health
     @app.get("/healthz")
     def _healthz():
         return {"ok": True, "ts": time.time()}
 
-    # Config shim — this bypasses Gradio's internal /config route
-    # and returns the same structure the frontend expects.
     @app.get("/config")
     def _config():
         try:
-            cfg = demo.get_config_file()  # Gradio builds the JSON-able config dict
+            cfg = demo.get_config_file()
             return JSONResponse(content=cfg)
         except Exception as e:
-            # If something fails, return explicit JSON (so the frontend won't choke on HTML)
             return JSONResponse(
                 status_code=500,
                 content={"error": "config_generation_failed", "detail": str(e)},
             )
 
-    # Mount Gradio UI at root; static assets & index served by Gradio
+    # Mount Gradio UI at root; our /config route remains at parent level
     app = gr.mount_gradio_app(app, demo, path="/")
     return app
 
@@ -280,13 +274,11 @@ def _config():
     _log_web_stack_versions_and_paths()
 
     demo = build_ui()
-    # Good defaults for Spaces
     demo.queue(max_size=16, api_open=False)
 
     threading.Thread(target=_post_launch_diag, daemon=True).start()
 
     if mount_mode:
-        # Our own FastAPI + /config shim
         try:
             from uvicorn import run as uvicorn_run
         except Exception:
@@ -294,15 +286,13 @@ def _config():
             raise
 
         app = build_fastapi_with_gradio(demo)
-        # NOTE: In Docker Spaces, this process is PID1; we call uvicorn.run programmatically.
         uvicorn_run(app=app, host=host, port=port, log_level="info")
     else:
-        # Standard Gradio server (uses internal FastAPI app & routes)
         demo.launch(
             server_name=host,
             server_port=port,
             share=False,
-            show_api=False,   # keep off in Spaces
+            show_api=False,
             show_error=True,
             quiet=False,
             debug=True,

ui.py

@@ -4,8 +4,16 @@
 Clean, focused main file that coordinates the application
 """
 
+# ============================================================
+# Mount-mode handoff: delegate to app.py when enabled
+# (So we can serve a safe /config JSON via our FastAPI shim)
+# ============================================================
+import os, runpy
+if os.getenv("GRADIO_MOUNT_MODE") == "1":
+    runpy.run_module("app", run_name="__main__")
+    raise SystemExit
+
 # ==== Runtime hygiene & paths (very high in file) ====
-import os
 import sys
 import logging
 from pathlib import Path