Update app.py

app.py

@@ -15,45 +15,58 @@ def generate_3d_model(prompt, output_path="assistant_3d.obj"):
             low_cpu_mem_usage=True
         ).to("cpu")
         
-        # Generate with minimal parameters
+        # Generate with fixed dimensions
         outputs = pipe(
             prompt,
             num_inference_steps=16,
-            frame_size=24,
-            guidance_scale=7.5
+            guidance_scale=7.5,
+            # Using standard dimensions that work with the model
+            frame_size=32,  # Changed from 24 to standard size
         )
         
-        # Get the mesh data
-        mesh = outputs.meshes[0]  # Get the first mesh
+        # Extract mesh - using the correct accessor
+        vertices = outputs.vertices[0].detach().cpu().numpy()
+        faces = outputs.faces[0].detach().cpu().numpy()
         
-        # Convert to vertices and faces format
-        verts = mesh.verts.detach().cpu().numpy()
-        faces = mesh.faces.detach().cpu().numpy()
-        
-        # Create trimesh object
-        mesh_obj = trimesh.Trimesh(vertices=verts, faces=faces)
+        # Create trimesh object with explicit vertex and face arrays
+        mesh_obj = trimesh.Trimesh(
+            vertices=vertices,
+            faces=faces,
+            process=True  # Enable post-processing to fix any mesh issues
+        )
         
-        # Export in desired format
-        if output_path.endswith('.obj'):
-            mesh_obj.export(output_path)
-        elif output_path.endswith('.glb'):
-            mesh_obj.export(output_path)
-        elif output_path.endswith('.stl'):
-            mesh_obj.export(output_path)
+        # Export with error checking
+        try:
+            if output_path.endswith('.obj'):
+                mesh_obj.export(output_path, include_normals=True)
+            elif output_path.endswith('.glb'):
+                mesh_obj.export(output_path)
+            elif output_path.endswith('.stl'):
+                mesh_obj.export(output_path)
+            print(f"Successfully exported 3D model to: {output_path}")
+        except Exception as export_error:
+            print(f"Error during export: {export_error}")
+            # Try alternate export format if primary fails
+            backup_path = output_path.rsplit('.', 1)[0] + '.ply'
+            mesh_obj.export(backup_path)
+            print(f"Exported backup model to: {backup_path}")
         
-        print(f"Successfully exported 3D model to: {output_path}")
         return output_path
         
     except Exception as e:
         print(f"Error during generation: {e}")
         print(f"Error type: {type(e)}")
         print(f"Full error details: {str(e)}")
+        
+        # Additional debug information
+        if hasattr(outputs, 'shape'):
+            print(f"Output shape: {outputs.shape}")
         raise
 
 if __name__ == "__main__":
-    prompt = "a simple ring"
+    # Using a very specific prompt for better results
+    prompt = "a simple 3D ring, perfect circle, clean geometry"
     try:
-        # You can change the extension to .glb or .stl as needed
         generate_3d_model(prompt, "assistant_3d.obj")
     except Exception as e:
         print(f"Generation failed: {e}")