Fix HF Space unstyled UI: stop overriding Gradio proxy settings

Forcing server_name=0.0.0.0 made Gradio serve CSS from http://0.0.0.0,
which breaks behind the HF HTTPS proxy (plain black-and-white HTML).
Use default demo.launch() on Spaces; keep custom bind only for local runs.

Co-authored-by: Cursor <cursoragent@cursor.com>

app.py

@@ -14,14 +14,14 @@ def _is_hf_space() -> bool:
 
 
 def _configure_runtime() -> None:
-    os.environ.setdefault("GRADIO_SERVER_PORT", os.environ.get("PORT", "7860"))
     os.environ.setdefault(
         "PREBACKBONE_ONLY_WEIGHTS",
         str(_ROOT / "weights" / "prebackbone_a11_ca.pt"),
     )
     if _is_hf_space():
         os.environ.setdefault("PRELOAD_MODEL", "1")
-        os.environ.setdefault("GRADIO_SERVER_NAME", "0.0.0.0")
+        # Do NOT set GRADIO_SERVER_NAME here — HF injects GRADIO_ROOT_PATH; forcing
+        # 0.0.0.0 breaks CSS/JS (unstyled black-and-white page behind the proxy).
 
 
 _configure_runtime()
@@ -65,7 +65,6 @@ def run_demo(defect_path: str | None, reference_path: str | None):
 
 
 def _example_pairs() -> list[list[str]]:
-    """Curated examples (Interface layout; keep short for HF iframe width)."""
     pairs: list[list[str]] = []
     ex_dir = _ROOT / "examples"
     if not ex_dir.is_dir():
@@ -101,7 +100,6 @@ def _preload_model() -> None:
 
 
 def build_demo() -> gr.Interface:
-    """gr.Interface gives the same polished card layout locally and on HF Spaces."""
     return gr.Interface(
         fn=run_demo,
         inputs=[
@@ -125,13 +123,7 @@ def build_demo() -> gr.Interface:
 demo = build_demo()
 
 
-def _launch_kwargs() -> dict:
-    if _is_hf_space():
-        return {
-            "server_name": "0.0.0.0",
-            "server_port": int(os.environ.get("PORT", "7860")),
-            "share": False,
-        }
+def _local_launch_kwargs() -> dict:
     server_name = os.environ.get("GRADIO_SERVER_NAME", "127.0.0.1")
     preferred_port = int(os.environ.get("PORT", os.environ.get("GRADIO_SERVER_PORT", "7860")))
     bind_host = "127.0.0.1" if server_name in ("127.0.0.1", "localhost") else server_name
@@ -159,9 +151,13 @@ if __name__ == "__main__":
     if os.environ.get("PRELOAD_MODEL", "0").lower() in ("1", "true", "yes"):
         _preload_model()
 
-    demo.launch(
-        **(_launch_kwargs()),
-        show_error=True,
-        inbrowser=False,
-        prevent_thread_lock=False,
-    )
+    if _is_hf_space():
+        # HF Spaces proxy sets PORT + GRADIO_ROOT_PATH — plain launch keeps themed UI.
+        demo.launch(show_error=True)
+    else:
+        demo.launch(
+            **_local_launch_kwargs(),
+            show_error=True,
+            inbrowser=False,
+            prevent_thread_lock=False,
+        )

setup.sh

@@ -1,12 +1,10 @@
 #!/usr/bin/env bash
-# Hugging Face Spaces: install deps before app starts (no ultralytics).
+# Hugging Face Spaces: install ML deps only (Gradio version comes from README sdk_version).
 set -euo pipefail
 cd "$(dirname "$0")"
 pip install -q -r requirements.txt
-pip install -q "numpy>=1.23.0,<2" "gradio-client==1.7.0"
-python -c "import gradio_patch; gradio_patch.apply(); print('gradio_patch OK')"
-export GRADIO_SERVER_NAME="${GRADIO_SERVER_NAME:-0.0.0.0}"
-export GRADIO_SERVER_PORT="${GRADIO_SERVER_PORT:-${PORT:-7860}}"
+pip install -q "numpy>=1.23.0,<2"
+# Do not reinstall gradio-client here — mismatched client breaks HF Space CSS/JS loading.
 export PRELOAD_MODEL="${PRELOAD_MODEL:-1}"
 export PREBACKBONE_ONLY_WEIGHTS="${PREBACKBONE_ONLY_WEIGHTS:-$(pwd)/weights/prebackbone_a11_ca.pt}"
-echo "RefDiffNet: standalone A11_CA (GRADIO_SERVER_NAME=${GRADIO_SERVER_NAME})"
+echo "RefDiffNet: deps ready (gradio from sdk_version in README)"