Upload 2 files

app.py

@@ -156,21 +156,41 @@ def process_garment(input_type, text_prompt, base_text_prompt, target_image, bas
                 
                 # Save target image to temp directory
                 target_image_path = os.path.join(temp_dir, "target_image.jpg")
+                
+                # Handle different possible image formats from Gradio
                 if isinstance(target_image, str):
+                    # Image is a file path
                     shutil.copy(target_image, target_image_path)
-                else:
+                elif isinstance(target_image, np.ndarray):
+                    # Image is a numpy array from Gradio
+                    img = Image.fromarray(target_image)
+                    img.save(target_image_path)
+                elif hasattr(target_image, 'save'):
+                    # Image is a PIL image
                     target_image.save(target_image_path)
+                else:
+                    return "Error: Unsupported target image format. Please try again."
+                
+                print(f"Target image saved to {target_image_path}")
                 
                 # Save base image if provided, or use default
                 if base_image is not None:
                     base_image_path = os.path.join(temp_dir, "base_image.jpg")
                     if isinstance(base_image, str):
                         shutil.copy(base_image, base_image_path)
-                    else:
+                    elif isinstance(base_image, np.ndarray):
+                        img = Image.fromarray(base_image)
+                        img.save(base_image_path)
+                    elif hasattr(base_image, 'save'):
                         base_image.save(base_image_path)
+                    else:
+                        # If format is unsupported, use target image
+                        base_image_path = target_image_path
+                    print(f"Base image saved to {base_image_path}")
                 else:
                     # If no base image, use a placeholder or default
                     base_image_path = target_image_path
+                    print("No base image provided, using target image as base")
                 
                 # Configure for image-based processing
                 config.update({
@@ -220,7 +240,18 @@ def process_garment(input_type, text_prompt, base_text_prompt, target_image, bas
             
             progress(0.1, desc="Initializing...")
             
+            # Print configuration for debugging
+            print("Starting processing with configuration:")
+            print(f"Mode: {'Image' if config.get('fashion_image', False) else 'Text'}")
+            if config.get('fashion_image', False):
+                print(f"Target image: {config['image_prompt']}")
+                print(f"Base image: {config['base_image_prompt']}")
+            else:
+                print(f"Target text: {config['text_prompt']}")
+                print(f"Base text: {config['base_text_prompt']}")
+            
             # Run the main processing loop
+            progress(0.2, desc="Running garment generation...")
             loop(config)
             
             progress(0.9, desc="Processing complete, preparing output...")
@@ -230,43 +261,63 @@ def process_garment(input_type, text_prompt, base_text_prompt, target_image, bas
             glb_files = []
             image_files = []
             
+            print("Searching for output files...")
+            
             # First check for mesh files in mesh_final directory (priority)
             mesh_final_dir = Path(temp_dir) / "mesh_final"
             if mesh_final_dir.exists():
+                print(f"Found mesh_final directory at {mesh_final_dir}")
                 for file_path in mesh_final_dir.rglob("*"):
                     if file_path.is_file():
                         if file_path.suffix.lower() == '.obj':
                             obj_files.append(str(file_path))
+                            print(f"Found OBJ file: {file_path}")
                         elif file_path.suffix.lower() == '.glb':
                             glb_files.append(str(file_path))
+                            print(f"Found GLB file: {file_path}")
+            else:
+                print("mesh_final directory not found")
             
             # Check other mesh directories
             for mesh_dir in Path(temp_dir).glob("mesh_*"):
                 if mesh_dir.is_dir() and mesh_dir.name != 'mesh_final':
+                    print(f"Checking directory: {mesh_dir}")
                     for file_path in mesh_dir.rglob("*"):
                         if file_path.is_file():
                             if file_path.suffix.lower() == '.obj':
                                 obj_files.append(str(file_path))
+                                print(f"Found OBJ file: {file_path}")
                             elif file_path.suffix.lower() == '.glb':
                                 glb_files.append(str(file_path))
+                                print(f"Found GLB file: {file_path}")
             
             # Collect image files for visualization
             for file_path in Path(temp_dir).rglob("*"):
                 if file_path.is_file() and file_path.suffix.lower() in ['.png', '.jpg', '.jpeg', '.gif', '.mp4']:
                     image_files.append(str(file_path))
+                    
+            print(f"Found {len(glb_files)} GLB files, {len(obj_files)} OBJ files, and {len(image_files)} image files")
             
             # Prioritize output: GLB, OBJ, then images
             if glb_files:
+                print(f"Returning GLB file: {glb_files[0]}")
                 return glb_files[0]  # Return first GLB file (best for web viewing)
             elif obj_files:
+                print(f"Returning OBJ file: {obj_files[0]}")
                 return obj_files[0]  # Return first OBJ file
             elif image_files:
+                print(f"Returning image file: {image_files[0]}")
                 return image_files[0]  # Return an image if no mesh was found
             else:
-                return "Processing completed but no output files found."
+                print("No output files found")
+                return "Processing completed but no output files found. Please check the logs for more information."
                 
     except Exception as e:
-        return f"Error during processing: {str(e)}"
+        import traceback
+        error_details = traceback.format_exc()
+        print(f"Error during processing: {str(e)}")
+        print(f"Error details: {error_details}")
+        return f"Error during processing: {str(e)}. Please check the logs for more details."
 
 def create_interface():
     """
@@ -278,6 +329,12 @@ def create_interface():
         
         This tool allows you to stylize 3D garments using text prompts or images. Generate a new 3D garment mesh (.obj/.glb) 
         that can be used for virtual try-on applications.
+        
+        ## How to use:
+        1. Choose either **Text** or **Image** input mode using the radio button below
+        2. For **Text** mode: Enter descriptions of your target and base garment styles
+        3. For **Image** mode: Upload an image of your desired garment style
+        4. Click "Generate 3D Garment" to create your 3D mesh file
         """)
         
         with gr.Row():
@@ -306,12 +363,18 @@ def create_interface():
                 # Image inputs (hidden by default)
                 with gr.Group(visible=False) as image_group:
                     target_image = gr.Image(
-                        label="Target Garment Image"
+                        label="Target Garment Image",
+                        type="pil",
+                        image_mode="RGB"
                     )
+                    gr.Markdown("*Upload an image of the desired garment style*")
                     
                     base_image = gr.Image(
-                        label="Base Garment Image (Optional)"
+                        label="Base Garment Image (Optional)",
+                        type="pil",
+                        image_mode="RGB"
                     )
+                    gr.Markdown("*Upload a base garment image (optional)*")
                 
                 # Custom mesh
                 custom_mesh = gr.File(