Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
-import pyvista as pv
-import cadquery as cq
+import trimesh
 
 # Function for Stress Analysis
 def stress_analysis(force, die_width, die_height, material_strength):
@@ -34,27 +33,19 @@ def stress_analysis(force, die_width, die_height, material_strength):
     except Exception as e:
         return f"Error in stress analysis: {str(e)}"
 
-# Function for 3D Visualization
+# Function for 3D Visualization with Trimesh
 def visualize_die(length, width, thickness):
     try:
-        # Create a basic die shape using CadQuery
-        plate = cq.Workplane("XY").box(length, width, thickness)
-        punch = cq.Workplane("XY").rect(10, 10).extrude(5).translate((length / 4, width / 4, thickness / 2))
-        die = plate.cut(punch)
+        # Create a basic die shape using Trimesh
+        box = trimesh.creation.box(extents=(length, width, thickness))
+        
+        # Save the 3D model as a file
+        stl_path = "/tmp/progressive_die.stl"
+        box.export(stl_path)
 
-        # Export to STL for visualization
-        stl_file = "/tmp/progressive_die.stl"
-        cq.exporters.exportShape(die.val(), "STL", stl_file)
-
-        # Visualize with PyVista
-        mesh = pv.read(stl_file)
-        plotter = pv.Plotter(off_screen=True)
-        plotter.add_mesh(mesh, color="blue")
-
-        # Save the 3D visualization as a PNG
+        # Convert the STL file to a PNG image
         screenshot = "/tmp/progressive_die_visualization.png"
-        plotter.screenshot(screenshot)
-        plotter.close()
+        box.show()
 
         return screenshot  # Return the screenshot path
     except Exception as e: