Update app.py

app.py

@@ -1,25 +1,22 @@
+import os
 import trimesh
 import gradio as gr
 import matplotlib.pyplot as plt
 import pyvista as pv
 from pyvista import Plotter
 
-# Enable headless rendering
+# Install and start Xvfb
+os.system("apt-get update && apt-get install -y xvfb libgl1-mesa-glx")
 pv.start_xvfb()
 
-
 def generate_cad_and_apdl(length, width, height):
     """
     Generate a 3D CAD model (box only) using Trimesh and output APDL script.
     """
-    # 1. Create a CAD model as a box
     cad_model = trimesh.creation.box(extents=(length, width, height))
-    
-    # Save the CAD model as an STL file
     cad_path = "/tmp/demo_model.stl"
     cad_model.export(cad_path)
-    
-    # 2. Generate APDL Script (basic block)
+
     apdl_script = f"""
 /prep7
 block, 0, {length}, 0, {width}, 0, {height}
@@ -34,19 +31,12 @@ solve
     apdl_path = "/tmp/demo_model_apdl.txt"
     with open(apdl_path, "w") as f:
         f.write(apdl_script)
-    
-    return cad_model, cad_path, apdl_path
 
+    return cad_model, cad_path, apdl_path
 
 def visualize_2d(cad_model):
-    """
-    Generate a 2D visualization of the CAD model using Matplotlib.
-    """
-    # Extract vertices and edges
     vertices = cad_model.vertices
     edges = cad_model.edges
-
-    # Plot the edges in 2D
     fig, ax = plt.subplots(figsize=(6, 6))
     for edge in edges:
         start, end = edge
@@ -61,78 +51,45 @@ def visualize_2d(cad_model):
     ax.set_ylabel("Y-axis")
     ax.grid(True)
     ax.set_aspect('equal')
-
-    # Save the figure to a file
     image_path = "/tmp/cad_2d_visualization.png"
     plt.savefig(image_path)
     plt.close(fig)
-    
     return image_path
 
-
 def visualize_3d(cad_model):
-    """
-    Generate a 3D visualization of the CAD model using PyVista.
-    """
-    # Convert Trimesh model to PyVista
     vertices = cad_model.vertices
     faces = cad_model.faces
     faces_pv = []
     for face in faces:
         faces_pv.append(len(face))
         faces_pv.extend(face)
-    
     mesh = pv.PolyData(vertices, faces_pv)
-    
-    # Create a PyVista plotter
     plotter = Plotter(off_screen=True)
     plotter.add_mesh(mesh, color="lightblue", show_edges=True)
     plotter.set_background("white")
     plotter.view_isometric()
-
-    # Save the visualization as an image
     image_path = "/tmp/cad_3d_visualization.png"
     plotter.screenshot(image_path)
     plotter.close()
-    
     return image_path
 
-
 def create_files_and_visualizations(length, width, height):
-    """
-    Generate CAD and APDL files, and visualize the model in 2D and 3D.
-    """
-    # Generate CAD and APDL files
     cad_model, cad_file, apdl_file = generate_cad_and_apdl(length, width, height)
-    
-    # Generate 2D visualization
     cad_2d_path = visualize_2d(cad_model)
-    
-    # Generate 3D visualization
     cad_3d_path = visualize_3d(cad_model)
-    
     return cad_file, apdl_file, cad_2d_path, cad_3d_path
 
-
-# Gradio interface
 with gr.Blocks() as app:
-    # Input fields
     with gr.Row():
         length = gr.Number(label="Length (mm)", value=100, precision=2)
         width = gr.Number(label="Width (mm)", value=50, precision=2)
         height = gr.Number(label="Height (mm)", value=20, precision=2)
-    
-    # Submit button
     submit_button = gr.Button("Submit")
-
-    # Output fields
     with gr.Row():
         cad_file = gr.File(label="Download CAD Model (STL)")
         apdl_file = gr.File(label="Download APDL Script")
         cad_2d_image = gr.Image(label="2D Visualization (Top View)")
         cad_3d_image = gr.Image(label="3D Visualization")
-
-    # Button click event
     submit_button.click(
         fn=create_files_and_visualizations,
         inputs=[length, width, height],