Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from PIL import Image
 import tempfile
 
+# Function to visualize texture based on selection criteria
 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')
@@ -41,7 +42,7 @@ def visualize_dynamic_texture(predefined_section, x_min, x_max, y_min, y_max, z_
     temp_file.close()
     return temp_file.name
 
-# Load the train model to get its bounds
+# Load train model to establish bounding box ranges
 train_model = trimesh.load('train.glb', force='mesh')
 train_bounds = train_model.bounds
 
@@ -50,13 +51,14 @@ 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]
 
-# Gradio UI for selection and dynamic visualization with live mode
+# Gradio UI with a single window, dynamic updates, and real-time changes
 with gr.Blocks() as app:
-    gr.Markdown("### 3D Model Texture Application with Live Selection")
+    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="Textured Model")
+    modified_model = gr.Model3D(label="Model with Applied Texture")
 
-    section_dropdown = gr.Dropdown(choices=['right compartments', 'left compartments', 'freight_body', 'custom'], label="Select Section", value='custom')
+    # 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:
@@ -75,11 +77,13 @@ with gr.Blocks() as app:
 
     section_dropdown.change(fn=toggle_custom_controls, inputs=section_dropdown, outputs=custom_controls)
 
-    # Update the model dynamically
+    # Update model dynamically
     def update_model(predefined_section, x_min, x_max, y_min, y_max, z_min, z_max):
         return visualize_dynamic_texture(predefined_section, x_min, x_max, y_min, y_max, z_min, z_max)
 
-    # Set live mode changes for each input
-    section_dropdown.change(fn=update_model, inputs=[section_dropdown, x_min_slider, x_max_slider, y_min_slider, y_max_slider, z_min_slider, z_max_slider], outputs=modified_model)
+    # Add event listeners for real-time updates when sliders or dropdown change
+    inputs = [section_dropdown, x_min_slider, x_max_slider, y_min_slider, y_max_slider, z_min_slider, z_max_slider]
+    for input_component in inputs:
+        input_component.change(fn=update_model, inputs=inputs, outputs=modified_model)
 
 app.launch()