Fix HF UI: restore gr.Interface layout instead of bare Blocks

The Blocks + bottom Examples gallery broke badly in the HF iframe.
Use Interface with Soft theme and 4 curated examples for consistent
local and Space appearance.

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

app.py

@@ -65,11 +65,27 @@ 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():
         return pairs
+    preferred = [
+        "Short_06_short_11_rot_idx010_input.jpg",
+        "Open_circuit_04_open_circuit_14_idx002_input.jpg",
+        "Mouse_bite_05_mouse_bite_02_idx007_input.jpg",
+        "Missing_hole_07_missing_hole_03_rot_idx004_input.jpg",
+    ]
+    seen: set[str] = set()
+    for name in preferred:
+        inp = ex_dir / name
+        ref = inp.with_name(name.replace("_input.", "_reference."))
+        if inp.is_file() and ref.is_file():
+            pairs.append([str(inp), str(ref)])
+            seen.add(name)
     for inp in sorted(ex_dir.glob("*_input.jpg")):
+        if inp.name in seen or len(pairs) >= 4:
+            break
         ref = inp.with_name(inp.name.replace("_input.", "_reference."))
         if ref.is_file():
             pairs.append([str(inp), str(ref)])
@@ -84,27 +100,26 @@ def _preload_model() -> None:
         print(f"[RefDiffNet] WARN: {e}")
 
 
-def build_demo() -> gr.Blocks:
-    with gr.Blocks(title=TITLE) as demo:
-        gr.Markdown(DESCRIPTION.strip())
-        with gr.Row():
-            defect_in = gr.Image(type="filepath", label="Defect image (input)")
-            ref_in = gr.Image(type="filepath", label="Golden reference")
-        run_btn = gr.Button("Run enrichment", variant="primary")
-        with gr.Row():
-            out_enriched = gr.Image(type="pil", label="Enriched output")
-            out_row = gr.Image(type="pil", label="Defect | Reference | Enriched")
-        run_btn.click(run_demo, inputs=[defect_in, ref_in], outputs=[out_enriched, out_row])
-
-        examples = _example_pairs()
-        if examples:
-            gr.Examples(
-                examples=examples,
-                inputs=[defect_in, ref_in],
-                examples_per_page=3,
-                label="Example pairs (pre-aligned HR-IPCB samples)",
-            )
-    return demo
+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=[
+            gr.Image(type="filepath", label="Defect image (input)", height=320),
+            gr.Image(type="filepath", label="Golden reference", height=320),
+        ],
+        outputs=[
+            gr.Image(type="pil", label="Enriched output", height=360),
+            gr.Image(type="pil", label="Defect | Reference | Enriched", height=360),
+        ],
+        title=TITLE,
+        description=DESCRIPTION.strip(),
+        examples=_example_pairs() or None,
+        examples_per_page=2,
+        cache_examples=False,
+        flagging_mode="never",
+        theme=gr.themes.Soft(),
+    )
 
 
 demo = build_demo()