Update app.py

app.py

@@ -2,50 +2,58 @@ import gradio as gr
 import trimesh
 import numpy as np
 from PIL import Image
-import io
 
-def visualize_texture(x_min, x_max, y_min, y_max, z_min, z_max):
-    # Load the GLB file
+def visualize_texture(section):
     mesh = trimesh.load('train.glb', force='mesh')
-
-    # Load the texture image
-    im = Image.open('defect.jpg')
-    im = im.convert('RGB')  # Ensure the image is in RGB format
-
-    # Create UV coordinates for the mesh
+    im = Image.open('defect.jpg').convert('RGB')
+
+    # Calculate bounds
+    bounds = mesh.bounds
+    min_bounds, max_bounds = bounds[0], bounds[1]
+    mid_x = (max_bounds[0] + min_bounds[0]) / 2
+    mid_y = (max_bounds[1] + min_bounds[1]) / 2
+    mid_z = (max_bounds[2] + min_bounds[2]) / 2
+
+    # Define sections
+    sections = {
+        'upper': np.where(mesh.vertices[:, 2] > mid_z)[0],
+        'lower': np.where(mesh.vertices[:, 2] <= mid_z)[0],
+        'middle': np.where((mesh.vertices[:, 0] > min_bounds[0]) & (mesh.vertices[:, 0] < max_bounds[0]) & 
+                           (mesh.vertices[:, 1] > min_bounds[1]) & (mesh.vertices[:, 1] < max_bounds[1]) & 
+                           (mesh.vertices[:, 2] > min_bounds[2]) & (mesh.vertices[:, 2] < max_bounds[2]))[0],
+        'top': np.where(mesh.vertices[:, 1] > mid_y)[0],
+        'right': np.where(mesh.vertices[:, 0] > mid_x)[0]
+    }
+
+    # Get indices for selected section
+    selected_indices = sections[section]
+
+    # Create UV coordinates
     uv = np.random.rand(len(mesh.vertices), 2)
+    new_uv = np.zeros_like(uv)
+    new_uv[selected_indices, :] = uv[selected_indices, :]
 
-    # Determine vertices within the specified bounds
-    yellow_indices = [i for i, vertex in enumerate(mesh.vertices) if x_min <= vertex[0] <= x_max and y_min <= vertex[1] <= y_max and z_min <= vertex[2] <= z_max]
-
-    # Update UV coordinates for identified vertices
+    # Create material and apply texture
     material = trimesh.visual.texture.SimpleMaterial(image=im)
-    new_uv = np.zeros_like(uv)
-    new_uv[yellow_indices, :] = uv[yellow_indices, :]
-    
     color_visuals = trimesh.visual.TextureVisuals(uv=new_uv, image=im, material=material)
-
-    # Apply the texture to the original mesh
     textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process=False)
-    modified_model = textured_mesh.export('trainm.glb')
-    return modified_model
+
+    return textured_mesh.export(file_type='glb')
 
 with gr.Blocks() as app:
     gr.Markdown("### 3D Model Texture Application")
     original_model = gr.Model3D('train.glb', label="Original Model")
     modified_model = gr.Model3D(label="Textured Model")
-    
-    with gr.Row():
-        x_min = gr.Slider(minimum=-100, maximum=100, label="X Min")
-        x_max = gr.Slider(minimum=-100, maximum=100, label="X Max")
     with gr.Row():
-        y_min = gr.Slider(minimum=-100, maximum=100, label="Y Min")
-        y_max = gr.Slider(minimum=-100, maximum=100, label="Y Max")
-    with gr.Row():
-        z_min = gr.Slider(minimum=-100, maximum=100, label="Z Min")
-        z_max = gr.Slider(minimum=-100, maximum=100, label="Z Max")
-
-    button = gr.Button("Visualize Texture")
-    button.click(visualize_texture, inputs=[x_min, x_max, y_min, y_max, z_min, z_max], outputs=[modified_model])
+        button_upper = gr.Button("Texture Upper Part")
+        button_lower = gr.Button("Texture Lower Part")
+        button_middle = gr.Button("Texture Middle")
+        button_top = gr.Button("Texture Top")
+        button_right = gr.Button("Texture Right")
+    button_upper.click(visualize_texture, inputs='upper', outputs=modified_model)
+    button_lower.click(visualize_texture, inputs='lower', outputs=modified_model)
+    button_middle.click(visualize_texture, inputs='middle', outputs=modified_model)
+    button_top.click(visualize_texture, inputs='top', outputs=modified_model)
+    button_right.click(visualize_texture, inputs='right', outputs=modified_model)
 
 app.launch()