update multiimages api

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio.events import SelectData
 import spaces
 from gradio_litmodel3d import LitModel3D
 import json
@@ -20,6 +21,54 @@ MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
+def to_pil_list(
+    multiimages: List[
+        Union[
+            Image.Image,
+            Tuple[Image.Image, str],
+            gr.File,
+            Tuple[gr.File, str],
+            str,               # fallback: plain path
+            Path
+        ]
+    ]
+) -> List[Image.Image]:
+    """
+    Convert a heterogeneous `multiimages` list into a homogeneous
+    `List[Image.Image]`.
+
+    Accepts elements in any of the following forms:
+      • PIL.Image
+      • (PIL.Image, caption)
+      • gr.File        (gr.File.name is the temp‑file path)
+      • (gr.File, caption)
+      • str / pathlib.Path  (direct file path)
+
+    Returns:
+        List[Image.Image]  -- guaranteed PIL images
+    """
+    pil_imgs: List[Image.Image] = []
+
+    for item in multiimages:
+        # Unpack tuple/list, keep first element
+        if isinstance(item, (tuple, list)):
+            item = item[0]
+
+        if isinstance(item, Image.Image):          # already PIL
+            pil_imgs.append(item)
+
+        elif hasattr(item, "name"):                # gr.File
+            pil_imgs.append(Image.open(item.name))
+
+        elif isinstance(item, (str, Path)):        # file path
+            pil_imgs.append(Image.open(item))
+
+        else:
+            raise TypeError(
+                f"Unsupported element in multiimages: {type(item)}"
+            )
+
+    return pil_imgs
 
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
@@ -160,8 +209,9 @@ def image_to_3d(
             },
         )
     else:
+        pil_images = to_pil_list(multiimages)
         outputs = pipeline.run_multi_image(
-            [img[0] for img in multiimages],
+            pil_images,
             seed=seed,
             formats=["gaussian", "mesh"],
             preprocess_image=False,
@@ -335,9 +385,8 @@ def test_for_api_gen(image: Image.Image) -> Image.Image:
     """
     return image
 
-def update_is_multiimage(selected_tab: str):
-    return "true" if selected_tab == "Multiple Images" else "false"
-
+def update_is_multiimage(event: SelectData):
+    return "true" if event.index == 1 else "false"
 
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
@@ -378,8 +427,17 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                         *NOTE: this is an experimental algorithm without training a specialized model. It may not produce the best results for all images, especially those having different poses or inconsistent details.*
                     """)
             
-            is_multiimage = gr.Radio(choices=["true", "false"],value="false",label="Use multi-image mode",visible=True)
-            input_tabs.select(fn=update_is_multiimage,inputs=None,outputs=is_multiimage)
+            is_multiimage = gr.Radio(
+                choices=["true", "false"],
+                value="false",
+                label="Use multi-image mode",
+                visible=True         
+            )
+
+            input_tabs.select(
+                fn=update_is_multiimage,
+                outputs=is_multiimage
+            )
             
             with gr.Accordion(label="Generation Settings", open=False):
                 seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)