Update app.py

app.py

@@ -2,62 +2,89 @@ import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 import pyvista as pv
-import os
+import cadquery as cq
 
-# Sample simulation function for illustration purposes
-def run_simulation(material_yield, part_thickness, press_force, punch_speed, safety_factor):
+# Function for Stress Analysis
+def stress_analysis(force, die_width, die_height, material_strength):
     try:
-        # Generate dummy data for simulation results
-        x = np.linspace(0, 10, 100)
-        y = np.sin(x) * material_yield * part_thickness / safety_factor  # Simulated stress distribution
-
-        # Plotting the stress analysis graph
-        plt.figure(figsize=(6, 4))
-        plt.plot(x, y, label="Stress Distribution")
-        plt.title(f"Stress Analysis: Yield={material_yield}, Thickness={part_thickness}, Force={press_force}, Speed={punch_speed}")
-        plt.xlabel("X-axis (Position)")
-        plt.ylabel("Stress (MPa)")
-        plt.legend()
-        plt.grid(True)
-        graph_output_path = "/tmp/simulation_output.png"
-        plt.savefig(graph_output_path)
-        plt.close()
-
-        print(f"Graph saved at {graph_output_path}")  # Debug message
-
-        # 3D Model Visualization using PyVista
-        mesh = pv.Sphere(radius=part_thickness)  # Dummy sphere, replace with actual part geometry
-        plotter = pv.Plotter()
-        plotter.add_mesh(mesh)
-        plotter.view_xy()
-
-        # Save the screenshot of the 3D model
-        plotter_output_path = "/tmp/3d_model_output.png"
-        plotter.screenshot(plotter_output_path)
-
-        print(f"3D Model screenshot saved at {plotter_output_path}")  # Debug message
-
-        return graph_output_path, plotter_output_path
+        # Calculate parameters
+        stress = force / (die_width * die_height)  # Stress = Force / Area
+        safety_factor = material_strength / stress  # Safety Factor = Material Strength / Stress
 
+        # Data for plotting
+        x = ["Stress", "Material Strength", "Safety Factor"]
+        y_stress = [stress, material_strength, safety_factor]
+
+        # Create a combined graph
+        fig, ax = plt.subplots()
+        ax.plot(x, y_stress, marker='o', label="Simulation Parameters")
+        ax.axhline(y=1, color='red', linestyle='--', label="Critical Safety Factor (1)")
+        ax.set_ylabel("Value")
+        ax.set_title("Stress Analysis Parameters")
+        ax.legend()
+
+        # Save the figure to a file
+        fig_path = "/tmp/stress_analysis_graph.png"
+        fig.savefig(fig_path)
+
+        # Close the figure after saving it
+        plt.close(fig)
+
+        return fig_path  # Return the image path
     except Exception as e:
-        print(f"Error during simulation: {e}")
-        return None, None
-
-# Gradio interface for input parameters and output visualization
-iface = gr.Interface(
-    fn=run_simulation,
-    inputs=[
-        gr.Slider(minimum=100, maximum=500, value=200, label="Material Yield Strength (MPa)"),
-        gr.Slider(minimum=0.5, maximum=10.0, value=3.0, label="Part Thickness (mm)"),
-        gr.Slider(minimum=5000, maximum=20000, value=12000, label="Press Force (N)"),
-        gr.Slider(minimum=1, maximum=5, value=2, label="Punch Speed (m/s)"),
-        gr.Slider(minimum=1.0, maximum=3.0, value=1.5, label="Safety Factor"),
-    ],
-    outputs=[
-        gr.Image(type="filepath", label="Stress Analysis Graph"),
-        gr.Image(type="filepath", label="3D Model Visualization")
-    ]
-)
-
-# Launch the interface
-iface.launch()
+        return f"Error in stress analysis: {str(e)}"
+
+# Function for 3D Visualization
+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)
+
+        # 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
+        screenshot = "/tmp/progressive_die_visualization.png"
+        plotter.screenshot(screenshot)
+        plotter.close()
+
+        return screenshot  # Return the screenshot path
+    except Exception as e:
+        return f"Error visualizing die: {str(e)}"
+
+# Combined Interface Function
+def interface(force, die_width, die_height, material_strength, length, width, thickness):
+    # Stress Analysis
+    stress_graph_path = stress_analysis(force, die_width, die_height, material_strength)
+    
+    # 3D Visualization
+    die_visualization_path = visualize_die(length, width, thickness)
+    
+    return stress_graph_path, die_visualization_path
+
+# Gradio Interface
+inputs = [
+    gr.Slider(minimum=100, maximum=500, value=200, label="Material Yield Strength (MPa)"),
+    gr.Slider(minimum=100, maximum=500, value=200, label="Die Width (mm)"),
+    gr.Slider(minimum=100, maximum=500, value=200, label="Die Height (mm)"),
+    gr.Slider(minimum=5000, maximum=20000, value=12000, label="Force (N)"),
+    gr.Slider(minimum=100, maximum=500, value=200, label="Length (mm)"),
+    gr.Slider(minimum=100, maximum=500, value=200, label="Width (mm)"),
+    gr.Slider(minimum=10, maximum=50, value=10, label="Thickness (mm)"),
+]
+
+outputs = [
+    gr.Image(type="filepath", label="Stress Analysis Graph"),
+    gr.Image(type="filepath", label="3D Visualization"),
+]
+
+app = gr.Interface(fn=interface, inputs=inputs, outputs=outputs)
+app.launch()