Upload app.py with huggingface_hub

app.py

@@ -103,14 +103,30 @@ DEFAULT_STEP = 3
 
 
 css = """
-/* Minimal color overrides for dark theme */
+/* ColmapView Dark Theme */
 :root {
+    --body-background-fill: #0a0a0a !important;
+    --background-fill-primary: #0f0f0f !important;
+    --background-fill-secondary: #161616 !important;
+    --block-background-fill: #161616 !important;
+    --input-background-fill: #1a1a1a !important;
+    --body-text-color: #e8e8e8 !important;
+    --block-label-text-color: #8a8a8a !important;
+    --block-title-text-color: #e8e8e8 !important;
+    --border-color-primary: #2a2a2a !important;
     --color-accent: #b8b8b8 !important;
     --color-accent-soft: rgba(184, 184, 184, 0.15) !important;
+    --button-primary-background-fill: #b8b8b8 !important;
+    --button-primary-text-color: #0a0a0a !important;
 }
 
+body { background: #0a0a0a !important; }
+.gradio-container { background: #0f0f0f !important; }
+.dark { background: #0f0f0f !important; }
+
 /* Previewer (required for custom HTML viewer) */
 .previewer-container {
+    background: #0a0a0a;
     position: relative;
     font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
     width: 100%;
@@ -404,8 +420,35 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
     return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 
 
-def prepare_examples() -> List[List[str]]:
-    """Prepare multi-image examples as lists of image paths."""
+def create_example_thumbnail(image_paths: List[str], thumb_size: int = 80) -> Image.Image:
+    """Create a composite thumbnail from multiple images."""
+    images = [Image.open(p) for p in image_paths[:3]]  # Use first 3 images
+    n = len(images)
+
+    # Resize each image to thumbnail size
+    thumbs = []
+    for img in images:
+        img = img.convert('RGBA')
+        img.thumbnail((thumb_size, thumb_size), Image.Resampling.LANCZOS)
+        thumbs.append(img)
+
+    # Create composite image
+    width = thumb_size * n + (n - 1) * 4  # 4px gap between images
+    height = thumb_size
+    composite = Image.new('RGBA', (width, height), (0, 0, 0, 0))
+
+    x = 0
+    for thumb in thumbs:
+        # Center vertically
+        y = (height - thumb.height) // 2
+        composite.paste(thumb, (x + (thumb_size - thumb.width) // 2, y), thumb)
+        x += thumb_size + 4
+
+    return composite
+
+
+def prepare_examples() -> List[Tuple[List[str], Image.Image]]:
+    """Prepare multi-image examples with thumbnails."""
     example_dir = "assets/example_multi_image"
     if not os.path.exists(example_dir):
         return []
@@ -419,7 +462,8 @@ def prepare_examples() -> List[List[str]]:
             if os.path.exists(img_path):
                 case_images.append(img_path)
         if case_images:
-            examples.append(case_images)
+            thumbnail = create_example_thumbnail(case_images)
+            examples.append((case_images, thumbnail))
     return examples
 
 
@@ -663,19 +707,32 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                     download_btn = gr.DownloadButton(label="Download GLB")
             gr.Markdown("*GLB extraction may take 30+ seconds.*")
 
-        with gr.Column(scale=1, min_width=172):
+        with gr.Column(scale=1, min_width=200):
             gr.Markdown("### Multi-View Examples")
             example_data = prepare_examples()
-            for i, example_paths in enumerate(example_data[:12]):
+            example_buttons = []
+            for i, (example_paths, thumbnail) in enumerate(example_data[:12]):
                 case_name = os.path.basename(example_paths[0]).split('_')[0]
-                btn = gr.Button(f"{case_name} ({len(example_paths)} views)", size="sm")
-                btn.click(
-                    fn=lambda paths=example_paths: load_example(paths),
-                    outputs=[image_prompt]
+                gr.Image(
+                    value=thumbnail,
+                    show_label=False,
+                    height=60,
+                    container=False,
+                    show_download_button=False,
+                    show_fullscreen_button=False,
                 )
+                btn = gr.Button(f"{case_name} ({len(example_paths)} views)", size="sm")
+                example_buttons.append((btn, example_paths))
 
     output_buf = gr.State()
 
+    # Connect example buttons
+    for btn, paths in example_buttons:
+        btn.click(
+            fn=lambda p=paths: load_example(p),
+            outputs=[image_prompt]
+        )
+
     # Handlers
     demo.load(start_session)
     demo.unload(end_session)