Fixed image upload errors

gradio_app.py

@@ -491,21 +491,38 @@ def _gen_shape(
     num_chunks=200000,
     randomize_seed: bool = False,
 ):
-    if not MV_MODE and image is None and caption is None:
+    # Check if we're using multi-view mode based on inputs
+    using_multiview = (mv_image_front is not None or mv_image_back is not None or 
+                      mv_image_left is not None or mv_image_right is not None)
+    
+    # Single image mode validation
+    if not using_multiview and image is None and caption is None:
         raise gr.Error("Please provide either a caption or an image.")
-    if MV_MODE:
-        if mv_image_front is None and mv_image_back is None \
-            and mv_image_left is None and mv_image_right is None:
-            raise gr.Error("Please provide at least one view image.")
+    
+    # Multi-view mode validation and processing
+    if using_multiview:
+        # Check if any valid images were provided
+        has_valid_image = False
         image = {}
-        if mv_image_front:
+        
+        if mv_image_front is not None:
             image['front'] = mv_image_front
-        if mv_image_back:
+            has_valid_image = True
+            
+        if mv_image_back is not None:
             image['back'] = mv_image_back
-        if mv_image_left:
+            has_valid_image = True
+            
+        if mv_image_left is not None:
             image['left'] = mv_image_left
-        if mv_image_right:
+            has_valid_image = True
+            
+        if mv_image_right is not None:
             image['right'] = mv_image_right
+            has_valid_image = True
+            
+        if not has_valid_image:
+            raise gr.Error("Please provide at least one view image.")
 
     seed = int(randomize_seed_fn(seed, randomize_seed))
 
@@ -540,18 +557,29 @@ def _gen_shape(
 
     # remove disk io to make responding faster, uncomment at your will.
     # image.save(os.path.join(save_folder, 'input.png'))
-    if MV_MODE:
-        start_time = time.time()
+    # Process images based on whether we're using multi-view mode
+    start_time = time.time()
+    
+    if isinstance(image, dict):  # Multi-view mode
         for k, v in image.items():
-            if check_box_rembg or v.mode == "RGB":
-                img = rmbg_worker(v.convert('RGB'))
-                image[k] = img
-        time_meta['remove background'] = time.time() - start_time
-    else:
-        if check_box_rembg or image.mode == "RGB":
-            start_time = time.time()
-            image = rmbg_worker(image.convert('RGB'))
-            time_meta['remove background'] = time.time() - start_time
+            if v is not None and (check_box_rembg or v.mode == "RGB"):
+                try:
+                    img = rmbg_worker(v.convert('RGB'))
+                    image[k] = img
+                except Exception as e:
+                    print(f"Error processing {k} view: {e}")
+                    # Keep the original image if background removal fails
+                    pass
+    else:  # Single image mode
+        if image is not None and (check_box_rembg or image.mode == "RGB"):
+            try:
+                image = rmbg_worker(image.convert('RGB'))
+            except Exception as e:
+                print(f"Error removing background: {e}")
+                # Keep the original image if background removal fails
+                pass
+                
+    time_meta['remove background'] = time.time() - start_time
 
     # remove disk io to make responding faster, uncomment at your will.
     # image.save(os.path.join(save_folder, 'rembg.png'))
@@ -581,7 +609,13 @@ def _gen_shape(
     stats['number_of_vertices'] = mesh.vertices.shape[0]
 
     stats['time'] = time_meta
-    main_image = image if not MV_MODE else image['front']
+    
+    # Select the main image for display based on what's available
+    if isinstance(image, dict) and 'front' in image:
+        main_image = image['front']  # Use front view as main image in multi-view mode
+    else:
+        main_image = image  # Use the single image in single-image mode
+        
     return mesh, main_image, save_folder, stats, seed
 
 @spaces.GPU(duration=180)