Update app.py

app.py

@@ -1,57 +1,65 @@
 import gradio as gr
 import cadquery as cq
 from cadquery import exporters
-import numpy as np
+import tempfile
 import pyvista as pv
-from pyvistaqt import BackgroundPlotter
 import trimesh
 
-# Load the STL file
-mesh = trimesh.load('/mnt/data/Tile_3_3.stl')
-
-# Display basic information
-print(mesh)
-print("Volume:", mesh.volume)
-print("Surface Area:", mesh.area)
-
-
-# Function to load a STEP file and display it
 def load_and_display_step(file):
     try:
-        # Load STEP file
-        model = cq.importers.importStep(file.name)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".step") as temp_file:
+            temp_file.write(file.read())
+            temp_path = temp_file.name
 
-        # Export as STL for visualization
+        model = cq.importers.importStep(temp_path)
         stl_file = "temp_model.stl"
         exporters.export(model, stl_file)
 
         # Render with PyVista
         mesh = pv.read(stl_file)
-        plotter = BackgroundPlotter()
+        plotter = pv.Plotter(off_screen=True)
+        image_file = "rendered_output.png"
         plotter.add_mesh(mesh, color="lightblue")
-        plotter.show()
+        plotter.screenshot(image_file)
+        plotter.close()
+
+        return f"STEP file processed successfully! Render saved as {image_file}"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-        return "File loaded and displayed successfully!"
+def analyze_stl(file):
+    try:
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".stl") as temp_file:
+            temp_file.write(file.read())
+            temp_path = temp_file.name
+
+        mesh = trimesh.load(temp_path)
+        return f"Volume: {mesh.volume:.3f} m³\nSurface Area: {mesh.area:.3f} m²"
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Function to run a simple stress analysis (placeholder logic)
 def stress_analysis(press_force, area):
     try:
-        stress = press_force / area  # Stress = Force / Area
-        return f"Calculated Stress: {stress} N/m²"
-    except ZeroDivisionError:
-        return "Error: Area cannot be zero."
+        if area <= 0:
+            return "Error: Area must be greater than zero."
+        stress = press_force / area
+        return f"Calculated Stress: {stress:.2f} N/m²"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-# Create Gradio Interface
 def main():
     with gr.Blocks() as app:
         gr.Markdown("# Press Tool Design Simulation")
 
         with gr.Row():
-            file_input = gr.File(label="Upload STEP File")
-            file_output = gr.Textbox(label="File Status")
-            file_submit = gr.Button("Load and Display STEP File")
+            step_input = gr.File(label="Upload STEP File")
+            step_output = gr.Textbox(label="STEP File Status")
+            step_submit = gr.Button("Load and Display STEP File")
+
+        with gr.Row():
+            stl_input = gr.File(label="Upload STL File")
+            stl_output = gr.Textbox(label="STL Analysis Result")
+            stl_submit = gr.Button("Analyze STL File")
 
         with gr.Row():
             force_input = gr.Number(label="Press Force (N)", value=1000)
@@ -59,12 +67,11 @@ def main():
             stress_output = gr.Textbox(label="Stress Analysis Result")
             stress_submit = gr.Button("Calculate Stress")
 
-        # Events
-        file_submit.click(fn=load_and_display_step, inputs=file_input, outputs=file_output)
+        step_submit.click(fn=load_and_display_step, inputs=step_input, outputs=step_output)
+        stl_submit.click(fn=analyze_stl, inputs=stl_input, outputs=stl_output)
         stress_submit.click(fn=stress_analysis, inputs=[force_input, area_input], outputs=stress_output)
 
     app.launch()
 
-# Run the app
 if __name__ == "__main__":
     main()