Update app.py

app.py

@@ -1,11 +1,10 @@
-
 import gradio as gr
 import trimesh
 import numpy as np
 from PIL import Image
 import tempfile
 
-def visualize_texture(section):
+def apply_section_texture(section):
     # Load the original mesh
     mesh = trimesh.load('train.glb', force='mesh')
     im = Image.open('rust_steel.png').convert('RGB')
@@ -31,14 +30,16 @@ def visualize_texture(section):
     # Get indices for the selected section
     selected_indices = sections[section]
 
-    # 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)
+    # Prepare new UV coordinates
+    uv = np.random.rand(len(mesh.vertices), 2)
+    new_uv = np.copy(uv)
 
-    # 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
+    # Apply UV changes only for selected section
+    mask = np.zeros(len(mesh.vertices), dtype=bool)
+    mask[selected_indices] = True
+    new_uv[~mask, :] = uv[~mask]  # Keep original UVs for non-selected regions
 
-    # Create material and apply texture only to the selected section
+    # Apply new material texture to the selected indices
     material = trimesh.visual.texture.SimpleMaterial(image=im)
     color_visuals = trimesh.visual.TextureVisuals(uv=new_uv, material=material)
 
@@ -57,7 +58,7 @@ with gr.Blocks() as app:
     modified_model = gr.Model3D(label="Textured Model")
 
     section_dropdown = gr.Dropdown(choices=['upper', 'lower', 'middle', 'top', 'right'], label="Select Section")
-    button = gr.Button("Visualize Texture")
-    button.click(visualize_texture, inputs=section_dropdown, outputs=modified_model)
+    button = gr.Button("Apply Texture to Section")
+    button.click(apply_section_texture, inputs=section_dropdown, outputs=modified_model)
 
-app.launch()
+app.launch()