Update app.py

app.py

@@ -45,30 +45,41 @@ def visualize_dynamic_texture(predefined_section, x_min, x_max, y_min, y_max, z_
 train_model = trimesh.load('train.glb', force='mesh')
 train_bounds = train_model.bounds
 
-# Get bounds for slider ranges
+# 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]
 
-# Create Gradio UI with predefined and custom selection options
+# Gradio UI for selection and dynamic visualization with live mode
 with gr.Blocks() as app:
-    gr.Markdown("### 3D Model Texture Application with Real-Time Visualization")
+    gr.Markdown("### 3D Model Texture Application with Live Selection")
     original_model = gr.Model3D('train.glb', label="Original Model")
     modified_model = gr.Model3D(label="Textured Model")
 
     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
+    # 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)
-        x_max_slider = gr.Slider(minimum=x_min_range, maximum=x_max_range, step=0.01, label="X Max", value=x_max_range)
+        x_min_slider = gr.Slider(minimum=x_min_range, maximum=x_max_range, step=0.01, label="X Min", value=x_min_range, live=True)
+        x_max_slider = gr.Slider(minimum=x_min_range, maximum=x_max_range, step=0.01, label="X Max", value=x_max_range, live=True)
 
-        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)
+        y_min_slider = gr.Slider(minimum=y_min_range, maximum=y_max_range, step=0.01, label="Y Min", value=y_min_range, live=True)
+        y_max_slider = gr.Slider(minimum=y_min_range, maximum=y_max_range, step=0.01, label="Y Max", value=y_max_range, live=True)
 
-        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)
+        z_min_slider = gr.Slider(minimum=z_min_range, maximum=z_max_range, step=0.01, label="Z Min", value=z_min_range, live=True)
+        z_max_slider = gr.Slider(minimum=z_min_range, maximum=z_max_range, step=0.01, label="Z Max", value=z_max_range, live=True)
 
     # 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 the 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)
+
+app.launch()