Base on 8f8ab35: trim omnirooms, add perspective gallery, panorama max edge 1536

app.py

@@ -81,62 +81,24 @@ LOCAL_CHECKPOINT_PATH = Path(
 )
 OUTPUT_ROOT = Path(os.environ.get("UNISHARP_OUTPUT_DIR", "/tmp/unisharp_outputs"))
 STABLE_INPUT_PATH = Path(os.environ.get("UNISHARP_STABLE_INPUT", "/tmp/unisharp_current_input.png"))
-EXAMPLES_DIR = ROOT / "examples"
-EXAMPLE_REPLICA_DIR = EXAMPLES_DIR / "replica"
-EXAMPLE_OMNIROOMS_DIR = EXAMPLES_DIR / "omnirooms"
-EXAMPLE_PERSPECTIVE_DIR = EXAMPLES_DIR / "perspective"
-
-_OMNIROOMS_BASENAMES = {
-    "AI_vol4_02_last_source.jpg",
-    "AI_vol3_03_first_source.jpg",
-    "AI_vol4_03_random_source.jpg",
-    "AI_vol4_02_random_source.jpg",
-    "AI_vol4_03_last_source.jpg",
-    "AI_vol4_04_random_source.jpg",
-    "AI_vol4_05_last_source.jpg",
-    "AI_vol4_05_middle_source.jpg",
-    "AI_vol4_02_middle_source.jpg",
-    "AIUE_V01_004_random_source.jpg",
-}
+EXAMPLE_REPLICA_DIR = ROOT / "examples" / "replica"
+EXAMPLE_OMNIROOMS_DIR = ROOT / "examples" / "omnirooms"
+EXAMPLE_PERSPECTIVE_DIR = ROOT / "examples" / "perspective"
 
-PANORAMA_EXAMPLE_NAMES = [
+REPLICA_EXAMPLE_NAMES = [
     "replica_apartment_0_0004_g01189",
-    "replica_apartment_1_0002_g00410",
     "replica_apartment_0_0004_g00631",
-    "replica_apartment_0_0004_g01786",
-    "replica_apartment_0_0004_g01541",
     "replica_apartment_1_0000_g01727",
-    "replica_apartment_0_0002_g01481",
-    "replica_apartment_0_0002_g01122",
-    "replica_apartment_0_0004_g01035",
-    "replica_apartment_1_0000_g01969",
-    "replica_apartment_1_0000_g00628",
-    "replica_apartment_1_0001_g00895",
-    "replica_apartment_0_0002_g00379",
     "replica_apartment_1_0000_g01186",
-    "replica_apartment_1_0001_g01485",
-    "replica_apartment_1_0001_g01202",
-    "replica_apartment_1_0001_g01975",
-    "replica_apartment_0_0002_g01412",
-    "replica_apartment_1_0000_g01259",
-    "replica_apartment_0_0002_g00968",
     "replica_apartment_1_0000_g01402",
-    "replica_apartment_1_0000_g01539",
-    "replica_apartment_0_0000_g01523",
     "replica_apartment_0_0000_g01456",
-    "replica_apartment_0_0000_g00249",
     "replica_office_3_0000_g01465",
-    "replica_apartment_1_0001_g01551",
-    "replica_apartment_1_0002_g00588",
-    "replica_office_3_0000_g01532",
-    "replica_apartment_1_0000_g01604",
-    "AI_vol4_02_last_source.jpg",
-    "AI_vol3_03_first_source.jpg",
-    "AI_vol4_03_random_source.jpg",
+]
+
+OMNIROOMS_EXAMPLE_NAMES = [
     "AI_vol4_02_random_source.jpg",
     "AI_vol4_03_last_source.jpg",
     "AI_vol4_04_random_source.jpg",
-    "AI_vol4_05_last_source.jpg",
     "AI_vol4_05_middle_source.jpg",
     "AI_vol4_02_middle_source.jpg",
     "AIUE_V01_004_random_source.jpg",
@@ -165,7 +127,7 @@ PERSPECTIVE_EXAMPLE_NAMES = [
 ]
 
 DEFAULT_PERSPECTIVE_MAX_LONG_EDGE = 768
-DEFAULT_PANORAMA_MAX_LONG_EDGE = 0
+DEFAULT_PANORAMA_MAX_LONG_EDGE = 1536
 DEFAULT_ORBIT_VIEWS = 10
 DEFAULT_FORWARD_VIEWS = 10
 DEFAULT_ORBIT_RADIUS_M = 0.10
@@ -457,14 +419,13 @@ def run_unisharp(
     return _run_unisharp_once(image_path, progress=progress)
 
 
-def _panorama_example_path(name: str) -> Path:
-    if name in _OMNIROOMS_BASENAMES or name.lower().endswith((".jpg", ".jpeg", ".webp")):
-        return EXAMPLE_OMNIROOMS_DIR / name
-    return EXAMPLE_REPLICA_DIR / f"{name}.png"
-
-
 def _panorama_example_paths() -> list[Path]:
-    return [path for name in PANORAMA_EXAMPLE_NAMES if (path := _panorama_example_path(name)).is_file()]
+    paths: list[Path] = []
+    for name in REPLICA_EXAMPLE_NAMES:
+        paths.append(EXAMPLE_REPLICA_DIR / f"{name}.png")
+    for name in OMNIROOMS_EXAMPLE_NAMES:
+        paths.append(EXAMPLE_OMNIROOMS_DIR / name)
+    return [path for path in paths if path.is_file()]
 
 
 def _perspective_example_paths() -> list[Path]:
@@ -690,7 +651,7 @@ with gr.Blocks(title="UniSHARP", css=CSS, theme=THEME) as demo:
             )
             run_button = gr.Button("Run", variant="primary", elem_id="run_button")
             with gr.Column(elem_id="demo_gallery_scroll"):
-                gr.Markdown("Panorama", elem_classes=["example-section-label"])
+                gr.Markdown("**Panorama**", elem_classes=["example-section-label"])
                 demo_gallery_panorama = gr.Gallery(
                     value=_demo_gallery_items(panorama_paths),
                     label="Panorama Examples",
@@ -706,7 +667,7 @@ with gr.Blocks(title="UniSHARP", css=CSS, theme=THEME) as demo:
                     show_share_button=False,
                     elem_id="demo_gallery_panorama",
                 )
-                gr.Markdown("Perspective", elem_classes=["example-section-label"])
+                gr.Markdown("**Perspective**", elem_classes=["example-section-label"])
                 demo_gallery_perspective = gr.Gallery(
                     value=_demo_gallery_items(perspective_paths),
                     label="Perspective Examples",