more yolo

app.py

@@ -278,16 +278,48 @@ def try_import_loop():
             print("✓ nvdiffrast imported successfully")
         except ImportError as e:
             print(f"✗ nvdiffrast import failed: {e}")
-            loop_import_error = f"Critical dependency missing: nvdiffrast - {str(e)}"
-            return False
+            # Try to install nvdiffrast if missing
+            try:
+                print("🔄 Attempting to install nvdiffrast...")
+                import subprocess
+                result = subprocess.run([sys.executable, "-m", "pip", "install", "nvdiffrast"], 
+                                     capture_output=True, text=True, timeout=300)
+                if result.returncode == 0:
+                    print("✅ nvdiffrast installed successfully")
+                    import nvdiffrast
+                    print("✓ nvdiffrast now imported successfully")
+                else:
+                    print(f"⚠️ nvdiffrast installation failed: {result.stderr}")
+                    loop_import_error = f"Critical dependency missing: nvdiffrast - {str(e)}"
+                    return False
+            except Exception as install_e:
+                print(f"⚠️ Could not install nvdiffrast: {install_e}")
+                loop_import_error = f"Critical dependency missing: nvdiffrast - {str(e)}"
+                return False
             
         try:
             import pytorch3d
             print("✓ pytorch3d imported successfully")
         except ImportError as e:
             print(f"✗ pytorch3d import failed: {e}")
-            loop_import_error = f"Critical dependency missing: pytorch3d - {str(e)}"
-            return False
+            # Try to install pytorch3d if missing
+            try:
+                print("🔄 Attempting to install pytorch3d...")
+                import subprocess
+                result = subprocess.run([sys.executable, "-m", "pip", "install", "pytorch3d", "--no-deps"], 
+                                     capture_output=True, text=True, timeout=300)
+                if result.returncode == 0:
+                    print("✅ pytorch3d installed successfully")
+                    import pytorch3d
+                    print("✓ pytorch3d now imported successfully")
+                else:
+                    print(f"⚠️ pytorch3d installation failed: {result.stderr}")
+                    loop_import_error = f"Critical dependency missing: pytorch3d - {str(e)}"
+                    return False
+            except Exception as install_e:
+                print(f"⚠️ Could not install pytorch3d: {install_e}")
+                loop_import_error = f"Critical dependency missing: pytorch3d - {str(e)}"
+                return False
         
         # Try to import fashion_clip
         try:
@@ -295,8 +327,29 @@ def try_import_loop():
             print("✓ FashionCLIP imported successfully")
         except ImportError as e:
             print(f"✗ FashionCLIP import failed: {e}")
-            loop_import_error = f"Critical dependency missing: FashionCLIP - {str(e)}"
-            return False
+            # Try to install FashionCLIP if missing
+            try:
+                print("🔄 Attempting to install FashionCLIP...")
+                if os.path.exists("packages/fashion_clip"):
+                    import subprocess
+                    result = subprocess.run([sys.executable, "-m", "pip", "install", "-e", "packages/fashion_clip"], 
+                                         capture_output=True, text=True, timeout=300)
+                    if result.returncode == 0:
+                        print("✅ FashionCLIP installed successfully")
+                        from packages.fashion_clip.fashion_clip.fashion_clip import FashionCLIP
+                        print("✓ FashionCLIP now imported successfully")
+                    else:
+                        print(f"⚠️ FashionCLIP installation failed: {result.stderr}")
+                        loop_import_error = f"Critical dependency missing: FashionCLIP - {str(e)}"
+                        return False
+                else:
+                    print("⚠️ FashionCLIP directory not found")
+                    loop_import_error = f"Critical dependency missing: FashionCLIP - {str(e)}"
+                    return False
+            except Exception as install_e:
+                print(f"⚠️ Could not install FashionCLIP: {install_e}")
+                loop_import_error = f"Critical dependency missing: FashionCLIP - {str(e)}"
+                return False
                     
         # Now try to import the loop module - this is the core processing engine
         try:
@@ -489,8 +542,6 @@ if is_hf_spaces:
 else:
     print("🏠 Local development environment detected")
 
-
-
 # Try to import the loop module after dependency installation attempts
 print("🔄 Attempting to import processing engine after dependency checks...")
 import_success = try_import_loop()
@@ -546,8 +597,8 @@ DEFAULT_CONFIG = {
     'consistency_clip_model': 'ViT-B/32',
     'consistency_vit_stride': 8,
     'consistency_vit_layer': 11,
-    'mesh': './meshes/longsleeve.obj',
-    'target_mesh': './meshes_target/jacket_sdf_new.obj',
+    'mesh': os.path.join(os.path.dirname(os.path.abspath(__file__)), 'meshes', 'longsleeve.obj'),
+    'target_mesh': os.path.join(os.path.dirname(os.path.abspath(__file__)), 'meshes_target', 'jacket_sdf_new.obj'),
     'retriangulate': 0,
     'bsdf': 'diffuse',
     'lr': 0.0025,
@@ -571,6 +622,8 @@ DEFAULT_CONFIG = {
     'elev_alpha': 1.0,
     'elev_beta': 5.0,
     'elev_max': 60.0,
+    'azim_alpha': 1.0,
+    'azim_beta': 5.0,
     'azim_min': 0.0,
     'azim_max': 360.0,
     'aug_loc': 1,
@@ -650,12 +703,17 @@ def process_garment(input_type, text_prompt, base_text_prompt, mesh_target_image
                         "dress_shortsleeve": "dress_shortsleeve"
                     }
                     mesh_file = mesh_mapping.get(source_mesh_type, "tshirt")
-                    source_mesh_file = f"./meshes/{mesh_file}.obj"
+                    
+                    # Use absolute paths for mesh files
+                    current_dir = os.path.dirname(os.path.abspath(__file__))
+                    source_mesh_file = os.path.join(current_dir, "meshes", f"{mesh_file}.obj")
                     
                     # Check if the mesh file exists
                     if not os.path.exists(source_mesh_file):
                         return f"Error: Mesh file {source_mesh_file} not found. Please check if the mesh files are available."
                     
+                    print(f"Using source mesh: {source_mesh_file}")
+                    
                     # Configure for image-to-mesh processing
                     config.update({
                         'mesh': source_mesh_file,
@@ -733,11 +791,67 @@ def process_garment(input_type, text_prompt, base_text_prompt, mesh_target_image
                 # Run the loop with error handling
                 try:
                     print("🚀 Starting garment generation with real processing engine...")
+                    print(f"Configuration: {config}")
+                    
+                    # Validate mesh files before processing
+                    if 'mesh' in config and config['mesh']:
+                        mesh_path = config['mesh']
+                        if not os.path.exists(mesh_path):
+                            error_message = f"Error: Source mesh file not found: {mesh_path}"
+                            print(error_message)
+                            return error_message
+                        
+                        # Check if mesh file is valid
+                        try:
+                            import pymeshlab
+                            ms = pymeshlab.MeshSet()
+                            ms.load_new_mesh(mesh_path)
+                            if ms.current_mesh().vertex_number() == 0:
+                                error_message = f"Error: Source mesh file has no vertices: {mesh_path}"
+                                print(error_message)
+                                return error_message
+                            print(f"✓ Source mesh validated: {ms.current_mesh().vertex_number()} vertices, {ms.current_mesh().face_number()} faces")
+                        except Exception as mesh_e:
+                            print(f"Warning: Could not validate mesh file: {mesh_e}")
+                    
                     loop(config)
                     print("✅ Garment generation completed successfully!")
+                except ValueError as ve:
+                    print(f"❌ Validation error during garment generation: {ve}")
+                    if "no vertices" in str(ve).lower() or "no faces" in str(ve).lower():
+                        error_message = f"Error: Invalid mesh data detected. The source mesh appears to be corrupted or empty. Please try a different mesh file."
+                    elif "jacobian" in str(ve).lower():
+                        error_message = f"Error: Jacobian computation failed. This may indicate an issue with the mesh structure or processing pipeline."
+                    elif "index" in str(ve).lower() and "bounds" in str(ve).lower():
+                        error_message = f"Error: Mesh processing failed due to invalid data structure. This may indicate corrupted mesh files or processing errors."
+                    else:
+                        error_message = f"Error during processing: {str(ve)}"
+                    return error_message
+                except FileNotFoundError as fe:
+                    print(f"❌ File not found error during garment generation: {fe}")
+                    if "mesh" in str(fe).lower():
+                        error_message = f"Error: Required mesh file not found during processing. This may indicate an issue with the mesh loading pipeline."
+                    elif "mtl" in str(fe).lower():
+                        error_message = f"Error: Material file not found. This may indicate an issue with the mesh file structure."
+                    else:
+                        error_message = f"Error: Required file not found during processing: {str(fe)}"
+                    return error_message
                 except Exception as e:
                     print(f"❌ Error during garment generation: {e}")
-                    error_message = f"Error during processing: {str(e)}"
+                    import traceback
+                    traceback.print_exc()
+                    
+                    # Provide more specific error messages based on error type
+                    if "nvdiffrast" in str(e).lower():
+                        error_message = "Error: Rendering engine (nvdiffrast) failed. This may be due to OpenGL/EGL compatibility issues."
+                    elif "clip" in str(e).lower():
+                        error_message = "Error: CLIP model failed to load or process. This may be due to model availability or compatibility issues."
+                    elif "cuda" in str(e).lower() or "gpu" in str(e).lower():
+                        error_message = "Error: GPU/CUDA processing failed. This may be due to hardware compatibility or driver issues."
+                    elif "memory" in str(e).lower():
+                        error_message = "Error: Insufficient memory during processing. Try reducing the number of epochs or using a smaller mesh."
+                    else:
+                        error_message = f"Error during processing: {str(e)}"
                     return error_message
             except RuntimeError as e:
                 print(f"Runtime error during processing: {e}")
@@ -752,6 +866,7 @@ def process_garment(input_type, text_prompt, base_text_prompt, mesh_target_image
             except Exception as e:
                 print(f"Error during processing: {e}")
                 error_message = f"Error during processing: {str(e)}"
+                print(error_message)
                 return error_message
                 
             progress(0.9, desc="Processing complete, preparing output...")
@@ -992,10 +1107,11 @@ def create_interface():
                 status_msg += "Upload a garment image and select mesh type, then click Generate."
             
             print(f"Text visibility: {text_visibility}, Image to Mesh visibility: {image_to_mesh_visibility}")
+            print(f"Returning updates: text_group={text_visibility}, image_to_mesh_group={image_to_mesh_visibility}")
                 
             return (
-                gr.update(visible=text_visibility),
-                gr.update(visible=image_to_mesh_visibility),
+                gr.Group.update(visible=text_visibility),
+                gr.Group.update(visible=image_to_mesh_visibility),
                 status_msg
             )