fix

app.py

@@ -16,6 +16,7 @@ from PIL import Image
 from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
+from collections.abc import Sequence
 
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -117,11 +118,29 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
     """
     return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 
+def normalize_multiimages(multiimages: Sequence) -> List[Tuple[Image.Image, str]]:
+    if not multiimages:
+        return []
+
+    if isinstance(multiimages[0], Image.Image):
+        return [
+            (pipeline.preprocess_image(img), f"gallery_{i}.png")
+            for i, img in enumerate(multiimages)
+        ]
+
+    if isinstance(multiimages[0], tuple):
+        return [
+            (pipeline.preprocess_image(img), name)
+            for img, name in multiimages
+        ]
+
+    return preprocess_upload_images(multiimages)
+
 
 @spaces.GPU
 def image_to_3d(
     image: Image.Image,
-    multiimages: Union[List[Tuple[Image.Image, str]], List[Any]],
+    multiimages: List[Any],
     is_multiimage: str,
     seed: int,
     ss_guidance_strength: float,
@@ -154,8 +173,7 @@ def image_to_3d(
     os.makedirs(user_dir, exist_ok=True)
     is_multiimage = is_multiimage.lower() == "true"
 
-    if multiimages and not isinstance(multiimages[0], tuple):
-        multiimages = preprocess_upload_images(multiimages)
+    multiimages = normalize_multiimages(multiimages)
 
     # Run pipeline depending on mode
     if not is_multiimage:
@@ -174,7 +192,7 @@ def image_to_3d(
             },
         )
     else:
-        pil_images = [d[0] for d in multiimages]
+        pil_images = [img for img, _ in multiimages]
         outputs = pipeline.run_multi_image(
             pil_images,
             seed=seed,
@@ -281,6 +299,15 @@ def split_image(image: Image.Image) -> List[Image.Image]:
         images.append(Image.fromarray(image[:, s:e+1]))
     return [preprocess_image(image) for image in images]
 
+def _example_to_multi(img: Image.Image):
+    imgs = split_image(img)
+    return imgs, imgs
+
+def _files_to_gallery_and_state(file_list):
+    tuples = preprocess_upload_images(file_list)
+    gallery_imgs = [img for img, _ in tuples]
+    return gallery_imgs, tuples
+
 @spaces.GPU(api_name="quick_generate_glb")
 def quick_generate_glb(
     image: Image.Image,
@@ -473,8 +500,9 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         examples_multi = gr.Examples(
             examples=prepare_multi_example(),
             inputs=[image_prompt],
-            fn=split_image,
-            outputs=[multiimage_combined],
+            fn=_example_to_multi,
+            outputs=[multiimage_prompt,
+                    multiimage_combined],
             run_on_click=True,
             examples_per_page=8,
         )
@@ -508,9 +536,10 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         outputs=[multiimage_combined],
     )
     uploaded_api_images.upload(
-        fn=preprocess_upload_images,
+        fn=_files_to_gallery_and_state,
         inputs=[uploaded_api_images],
-        outputs=[multiimage_combined],
+        outputs=[multiimage_prompt,
+                multiimage_combined],
         preprocess=False,
     )