Update app.py

app.py

@@ -4,22 +4,33 @@ import numpy as np
 from PIL import Image
 import tempfile
 
-def visualize_dynamic_texture(x_min, x_max, y_min, y_max, z_min, z_max):
+def visualize_dynamic_texture(predefined_section, x_min, x_max, y_min, y_max, z_min, z_max):
     # Load the original mesh
     mesh = trimesh.load('train.glb', force='mesh')
-    rust_texture = Image.open('rust_steel.png').convert('RGB')
+    rust_texture = Image.open('/mnt/data/A_texture_image_that_combines_rust_and_yellow_patt.png').convert('RGB')
 
-    # Identify vertices within the given bounding box
-    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]
+    # Predefined sections
+    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 coordinates
+    # 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 texture
+    # Create material and apply the new texture
     material = trimesh.visual.texture.SimpleMaterial(image=rust_texture)
     color_visuals = trimesh.visual.TextureVisuals(uv=new_uv, image=rust_texture, material=material)
     textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process=False)
@@ -30,32 +41,34 @@ def visualize_dynamic_texture(x_min, x_max, y_min, y_max, z_min, z_max):
     temp_file.close()
     return temp_file.name
 
-# Get bounding box values from the original mesh to define slider ranges
-mesh = trimesh.load('train.glb', force='mesh')
-bounds = mesh.bounds
-x_min_range, x_max_range = bounds[0][0], bounds[1][0]
-y_min_range, y_max_range = bounds[0][1], bounds[1][1]
-z_min_range, z_max_range = bounds[0][2], bounds[1][2]
+# Load the train model to get its bounds
+train_model = trimesh.load('train.glb', force='mesh')
+train_bounds = train_model.bounds
 
+# Get bounds for slider ranges
+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]
+
+# Create Gradio UI with predefined and custom selection options
 with gr.Blocks() as app:
-    gr.Markdown("### Select Model Portion with Sliders")
+    gr.Markdown("### 3D Model Texture Application with Real-Time Visualization")
     original_model = gr.Model3D('train.glb', label="Original Model")
     modified_model = gr.Model3D(label="Textured Model")
 
-    # Add sliders for bounding box selection
-    with gr.Row():
+    section_dropdown = gr.Dropdown(choices=['right compartments', 'left compartments', 'freight_body', 'custom'], label="Select Section", value='custom')
+
+    # Add sliders for bounding box selection, visible only when 'custom' is selected
+    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)
         x_max_slider = gr.Slider(minimum=x_min_range, maximum=x_max_range, step=0.01, label="X Max", value=x_max_range)
 
-    with gr.Row():
         y_min_slider = gr.Slider(minimum=y_min_range, maximum=y_max_range, step=0.01, label="Y Min", value=y_min_range)
         y_max_slider = gr.Slider(minimum=y_min_range, maximum=y_max_range, step=0.01, label="Y Max", value=y_max_range)
 
-    with gr.Row():
-        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)
-
-    button = gr.Button("Apply Texture")
-    button.click(visualize_dynamic_texture, inputs=[x_min_slider, x_max_slider, y_min_slider, y_max_slider, z_min_slider, z_max_slider], outputs=modified_model)
+        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)
 
-app.launch()
+    # Toggle visibility of custom controls
+    def toggle_custom_controls(predefined_section):
+        return gr.update(visible=(predefined_section == 'custom'))