Upload app.py with huggingface_hub

app.py

@@ -420,60 +420,38 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
     return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 
 
-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."""
+def prepare_multi_example() -> List[Image.Image]:
+    """Prepare multi-image examples as concatenated images for gr.Examples."""
     example_dir = "assets/example_multi_image"
     if not os.path.exists(example_dir):
         return []
-    files = os.listdir(example_dir)
-    cases = list(set([f.split('_')[0] for f in files if '_' in f and f.endswith('.png')]))
-    examples = []
+    cases = list(set([f.split('_')[0] for f in os.listdir(example_dir) if '_' in f and f.endswith('.png')]))
+    images = []
     for case in sorted(cases):
         case_images = []
         for i in range(1, 10):
             img_path = f'{example_dir}/{case}_{i}.png'
             if os.path.exists(img_path):
-                case_images.append(img_path)
+                img = Image.open(img_path)
+                W, H = img.size
+                img = img.resize((int(W / H * 512), 512))
+                case_images.append(np.array(img))
         if case_images:
-            thumbnail = create_example_thumbnail(case_images)
-            examples.append((case_images, thumbnail))
-    return examples
+            images.append(Image.fromarray(np.concatenate(case_images, axis=1)))
+    return images
 
 
-def load_example(example_paths: List[str]) -> List[Image.Image]:
-    """Load example images from paths."""
+def split_image(image: Image.Image) -> List[Image.Image]:
+    """Split a concatenated multi-view image into separate images based on alpha."""
+    image = np.array(image)
+    alpha = image[..., 3]
+    alpha = np.any(alpha > 0, axis=0)
+    start_pos = np.where(~alpha[:-1] & alpha[1:])[0].tolist()
+    end_pos = np.where(alpha[:-1] & ~alpha[1:])[0].tolist()
     images = []
-    for path in example_paths:
-        img = Image.open(path)
-        images.append(img)
-    return images
+    for s, e in zip(start_pos, end_pos):
+        images.append(Image.fromarray(image[:, s:e+1]))
+    return [preprocess_image(img) for img in images]
 
 
 @spaces.GPU(duration=120)
@@ -708,29 +686,20 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             gr.Markdown("*GLB extraction may take 30+ seconds.*")
 
         with gr.Column(scale=1, min_width=200):
+            # Hidden image for examples input
+            example_image = gr.Image(visible=False, type="pil", image_mode="RGBA")
             gr.Markdown("### Multi-View Examples")
-            example_data = prepare_examples()
-            example_buttons = []
-            for i, (example_paths, thumbnail) in enumerate(example_data[:12]):
-                case_name = os.path.basename(example_paths[0]).split('_')[0]
-                gr.Image(
-                    value=thumbnail,
-                    show_label=False,
-                    height=60,
-                    container=False,
-                )
-                btn = gr.Button(f"{case_name} ({len(example_paths)} views)", size="sm")
-                example_buttons.append((btn, example_paths))
+            examples = gr.Examples(
+                examples=prepare_multi_example(),
+                inputs=[example_image],
+                fn=split_image,
+                outputs=[image_prompt],
+                run_on_click=True,
+                examples_per_page=12,
+            )
 
     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)