Update app.py

app.py

@@ -3,7 +3,6 @@ import trimesh
 import numpy as np
 from PIL import Image
 import tempfile
-import os
 
 def visualize_texture(section):
     # Load the original mesh
@@ -31,23 +30,16 @@ def visualize_texture(section):
     # Get indices for the selected section
     selected_indices = sections[section]
 
-    # Generate new UV coordinates and maintain existing colors
-    uv = np.random.rand(len(mesh.vertices), 2)
-    original_colors = mesh.visual.vertex_colors
+    # Initialize the UV map with the original values
+    original_uv = mesh.visual.uv if mesh.visual.uv is not None else np.random.rand(len(mesh.vertices), 2)
 
-    # Apply the new UV coordinates only to the selected section, keeping the original color elsewhere
-    new_uv = np.zeros_like(uv)
-    new_uv[selected_indices, :] = uv[selected_indices, :]
-    updated_vertex_colors = np.copy(original_colors)
+    # Initialize a new UV map and keep the original values for non-selected vertices
+    new_uv = np.copy(original_uv)
+    new_uv[selected_indices, :] = np.random.rand(len(selected_indices), 2)  # Randomized UV for the selected section
 
-    # Blend or mix the new texture into existing colors (example mix formula)
-    blend_factor = 0.5  # Adjust to change blending strength
-    new_texture = trimesh.visual.texture.SimpleMaterial(image=im)
-    for index in selected_indices:
-        updated_vertex_colors[index] = blend_factor * new_texture.image[index % new_texture.image.shape[0], index % new_texture.image.shape[1]] + (1 - blend_factor) * original_colors[index]
-
-    # Create a texture visual with blended colors and updated UVs
-    color_visuals = trimesh.visual.TextureVisuals(uv=new_uv, vertex_colors=updated_vertex_colors, material=new_texture)
+    # Create material and apply texture only to the selected section
+    material = trimesh.visual.texture.SimpleMaterial(image=im)
+    color_visuals = trimesh.visual.TextureVisuals(uv=new_uv, material=material)
 
     # Create a new mesh with the visual
     textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process=False)