Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+-import gradio as gr
 import trimesh
 import numpy as np
 from PIL import Image
@@ -11,7 +11,24 @@ def visualize_dynamic_texture(predefined_section, x_min, x_max, y_min, y_max, z_
     rust_texture = Image.open('rust_steel.png').convert('RGB')
 
     # Predefined sections
-if predefined_section == 'right compartments':
+    if predefined_section == 'right compartments':
+        selected_indices = np.where(mesh.vertices[:, 0] > (train_bounds[0][0] + train_bounds[1][0]) / 2)[0]
+    elif predefined_section == 'left compartments':
+        selected_indices = np.where(mesh.vertices[:, 0] <= (train_bounds[0][0] + train_bounds[1][0]) / 2)[0]
+    elif predefined_section == 'freight_body':
+        selected_indices = np.where((mesh.vertices[:, 0] >= train_bounds[0][0]) & (mesh.vertices[:, 0] <= train_bounds[1][0]) &
+                                    (mesh.vertices[:, 2] <= (train_bounds[0][2] + train_bounds[1][2]) / 2))[0]
+    elif predefined_section == 'custom':
+        # Use custom sliders for custom section
+        selected_indices = np.where((mesh.vertices[:, 0] >= x_min) & (mesh.vertices[:, 0] <= x_max) &
+                                    (mesh.vertices[:, 1] >= y_min) & (mesh.vertices[:, 1] <= y_max) &
+                                    (mesh.vertices[:, 2] >= z_min) & (mesh.vertices[:, 2] <= z_max))[0]
+    else:
+        selected_indices = np.array([])
+
+    # Initialize UV mapping
+    uv = np.random.rand(len(mesh.vertices), 2)
+    new_uv = np.zeros_like(uv)
     new_uv[selected_indices, :] = uv[selected_indices, :]
 
     # Create material and apply the new texture
@@ -20,6 +37,17 @@ if predefined_section == 'right compartments':
     textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process=False)
 
     # Save the mesh to a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.glb')
+    textured_mesh.export(temp_file.name, file_type='glb')
+    temp_file.close()
+    return temp_file.name
+
+# Load train model to establish bounding box ranges
+train_model = trimesh.load('train.glb', force='mesh')
+train_bounds = train_model.bounds
+
+# Get slider ranges based on train model bounds
+x_min_range, x_max_range = train_bounds[0][0], train_bounds[1][0]
 y_min_range, y_max_range = train_bounds[0][1], train_bounds[1][1]
 z_min_range, z_max_range = train_bounds[0][2], train_bounds[1][2]
 
@@ -28,8 +56,10 @@ with gr.Blocks() as app:
     gr.Markdown("### 3D Model Texture Application with Predefined and Custom Sections")
     original_model = gr.Model3D('train.glb', label="Original Model")
     modified_model = gr.Model3D(label="Model with Applied Texture")
+
     # Dropdown for predefined and custom selection
     section_dropdown = gr.Radio(choices=['right compartments', 'left compartments', 'freight_body', 'custom'], label="Select Section", value='custom')
+
     # Custom sliders for the bounding box selection
     with gr.Row(visible=True) as custom_controls:
         x_min_slider = gr.Slider(minimum=x_min_range, maximum=x_max_range, step=0.01, label="X Min", value=x_min_range)
@@ -41,10 +71,10 @@ with gr.Blocks() as app:
         z_min_slider = gr.Slider(minimum=z_min_range, maximum=z_max_range, step=0.01, label="Z Min", value=z_min_range)
         z_max_slider = gr.Slider(minimum=z_min_range, maximum=z_max_range, step=0.01, label="Z Max", value=z_max_range)
 
-
     # Toggle visibility of custom controls
     def toggle_custom_controls(predefined_section):
         return gr.update(visible=(predefined_section == 'custom'))
+
     section_dropdown.change(fn=toggle_custom_controls, inputs=section_dropdown, outputs=custom_controls)
 
     # Update model dynamically
@@ -56,5 +86,4 @@ with gr.Blocks() as app:
     for input_component in inputs:
         input_component.change(fn=update_model, inputs=inputs, outputs=modified_model)
 
-
 app.launch()