Upload 150 files

app.py

@@ -487,6 +487,18 @@ def build_demo():
     sample_sat_images_with_csv = [p for p in sample_sat_images if p.with_suffix(".csv").exists()]
     sample_sky_pairs, default_sky_path = resolve_demo_sky_pairs(demo_dir)
 
+    # Build thumbnail paths for faster UI loading
+    sat_thumb_dir = demo_dir / "satellite" / "thumbnails"
+    pano_thumb_dir = demo_dir / "panorama" / "thumbnails"
+
+    def get_thumbnail(original_path: Path) -> str:
+        """Return thumbnail path if it exists, otherwise fall back to original."""
+        thumb_dir = sat_thumb_dir if "satellite" in str(original_path) else pano_thumb_dir
+        thumb_path = thumb_dir / (original_path.stem + ".jpg")
+        if thumb_path.exists():
+            return str(thumb_path)
+        return str(original_path)
+
     with gr.Blocks(title="Sat3DGen Demo", theme=gr.themes.Soft()) as demo:
         gr.Markdown(
             """
@@ -524,12 +536,27 @@ def build_demo():
                     download_button = gr.DownloadButton("💾 Download Mesh (.obj)", variant="secondary")
 
             if sample_sat_images:
-                gr.Markdown("### Sample Images")
-                gr.Examples(
-                    examples=[[str(p)] for p in sample_sat_images],
-                    inputs=[sat_input],
+                gr.Markdown("### Sample Images — click to load")
+                mesh_sat_gallery = gr.Gallery(
+                    value=[get_thumbnail(p) for p in sample_sat_images],
                     label="Click to load a sample satellite image",
-                    examples_per_page=len(sample_sat_images),
+                    columns=5,
+                    rows=2,
+                    height=160,
+                    object_fit="cover",
+                    allow_preview=False,
+                )
+
+                def load_sat_for_mesh(evt: gr.SelectData):
+                    """Load the full-resolution image when a thumbnail is clicked."""
+                    if evt.index is None or evt.index >= len(sample_sat_images):
+                        return None
+                    return Image.open(str(sample_sat_images[evt.index]))
+
+                mesh_sat_gallery.select(
+                    fn=load_sat_for_mesh,
+                    inputs=None,
+                    outputs=[sat_input],
                 )
 
             gr.Markdown(
@@ -656,7 +683,7 @@ def build_demo():
             if sample_sat_images_with_csv:
                 gr.Markdown("### 🛰️ Sample Satellite Images — click to load")
                 sat_gallery = gr.Gallery(
-                    value=[str(p) for p in sample_sat_images_with_csv],
+                    value=[get_thumbnail(p) for p in sample_sat_images_with_csv],
                     label="Sample Satellite Images (with trajectory)",
                     columns=5,
                     rows=1,
@@ -681,7 +708,7 @@ def build_demo():
                 sky_gallery = gr.Gallery(
                     value=[
                         (
-                            str(pano_path),
+                            get_thumbnail(pano_path),
                             f"{pano_path.name} (Default)" if pano_path == default_sky_path else pano_path.name,
                         )
                         for pano_path, _ in sample_sky_pairs