Add examples (stored with Xet)

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/** filter=lfs diff=lfs merge=lfs -text
+examples/ filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+__pycache__/

app.py

@@ -156,7 +156,9 @@ def predictions_to_glb(
     # Rotate scene for better visualize
     align_rotation = np.eye(4)
     align_rotation[:3, :3] = Rotation.from_euler("y", 100, degrees=True).as_matrix()            # plane rotate
-    align_rotation[:3, :3] = align_rotation[:3, :3] @ Rotation.from_euler("x", 155, degrees=True).as_matrix()           # roll
+    align_rotation[:3, :3] = align_rotation[:3, :3] @ Rotation.from_euler("x", 208, degrees=True).as_matrix()           # roll
+    align_rotation[:3, :3] = align_rotation[:3, :3] @ Rotation.from_euler("y", 180, degrees=True).as_matrix()           # roll
+    align_rotation[:3, :3] = align_rotation[:3, :3] @ Rotation.from_euler("z", 350, degrees=True).as_matrix()           # roll
     scene_3d.apply_transform(align_rotation)
 
     print("GLB Scene built")
@@ -548,6 +550,72 @@ def update_visualization(
 # Example images
 # -------------------------------------------------------------------------
 
+EXAMPLES_BASE_DIR = os.path.join(os.path.dirname(__file__), "examples")
+
+def build_example_index(base_dir: str):
+    examples = []
+    if not os.path.isdir(base_dir):
+        return examples
+    for folder in sorted(os.listdir(base_dir)):
+        full_path = os.path.join(base_dir, folder)
+        if not os.path.isdir(full_path):
+            continue
+        images = [
+            os.path.join(full_path, name)
+            for name in sorted(os.listdir(full_path))
+            if name.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
+        ]
+        if not images:
+            continue
+        examples.append(
+            {
+                "name": folder,
+                "thumb": images[0],
+                "images": images,
+            }
+        )
+    return examples[:5]
+
+
+EXAMPLE_DATA = build_example_index(EXAMPLES_BASE_DIR)
+EXAMPLE_THUMBNAILS = [item["thumb"] for item in EXAMPLE_DATA]
+EXAMPLE_BY_THUMB = {os.path.abspath(item["thumb"]): item for item in EXAMPLE_DATA}
+
+
+def _normalize_gallery_value(value):
+    if value is None:
+        return None
+    if isinstance(value, dict):
+        value = value.get("name")
+    if isinstance(value, (list, tuple)) and value:
+        value = value[0]
+    if isinstance(value, str):
+        return os.path.abspath(value)
+    return None
+
+
+def select_example(evt: gr.SelectData):
+    if evt is None:
+        return None, gr.update(), gr.update(), "No example selected.", "False", gr.update(selected_index=None)
+
+    example = None
+    selected_path = _normalize_gallery_value(getattr(evt, "value", None))
+    if selected_path:
+        example = EXAMPLE_BY_THUMB.get(selected_path)
+
+    if example is None and getattr(evt, "index", None) is not None:
+        idx = evt.index[0] if isinstance(evt.index, (list, tuple)) else evt.index
+        if isinstance(idx, int) and 0 <= idx < len(EXAMPLE_DATA):
+            example = EXAMPLE_DATA[idx]
+
+    if example is None:
+        return None, gr.update(), gr.update(), "No example selected.", "False", gr.update(selected_index=None)
+
+    target_dir, image_paths = handle_uploads(None, example["images"])
+    log_msg = f"Example '{example['name']}' loaded. Click 'Reconstruct' to begin 3D processing."
+    return None, target_dir, image_paths, log_msg, "False", gr.update(selected_index=None)
+
+
 great_wall_video = "examples/videos/great_wall.mp4"
 colosseum_video = "examples/videos/Colosseum.mp4"
 room_video = "examples/videos/room.mp4"
@@ -611,9 +679,9 @@ with gr.Blocks(
     # Instead of gr.State, we use a hidden Textbox:
     is_example = gr.Textbox(label="is_example", visible=False, value="None")
     num_images = gr.Textbox(label="num_images", visible=False, value="None")
-
+    
     gr.HTML(
-        """
+    """
     <h1>Flow3r: Factored Flow Prediction for Visual Geometry Learning</h1>
     <p>
     <a href="https://github.com/Kidrauh/flow3r">GitHub Repository</a> |
@@ -730,4 +798,43 @@ with gr.Blocks(
         inputs=[input_video, input_images],
         outputs=[reconstruction_output, target_dir_output, image_gallery, log_output],
     )
+
+    # -------------------------------------------------------------------------
+    # Examples
+    # -------------------------------------------------------------------------
+    gr.Markdown("**Examples**")
+    gr.Markdown("Click any image to load an example.")
+    example_gallery = gr.Gallery(
+        value=[],
+        columns=5,
+        height="140px",
+        object_fit="contain",
+        label="Example Sets",
+        show_label=False,
+        preview=False,
+        selected_index=None,
+    )
+    example_gallery.select(
+        fn=select_example,
+        inputs=[],
+        outputs=[reconstruction_output, target_dir_output, image_gallery, log_output, is_example, example_gallery],
+    ).then(
+        fn=update_log, inputs=[], outputs=[log_output]
+    ).then(
+        fn=gradio_demo,
+        inputs=[
+            target_dir_output,
+            conf_thres,
+            frame_filter,
+            show_cam,
+        ],
+        outputs=[reconstruction_output, log_output, frame_filter],
+    ).then(
+        fn=lambda: "False", inputs=[], outputs=[is_example]
+    )
+    demo.load(
+        fn=lambda: gr.update(value=EXAMPLE_THUMBNAILS, selected_index=None),
+        inputs=[],
+        outputs=[example_gallery],
+    )
     demo.queue(max_size=20).launch(show_error=True, share=True, ssr_mode=False)