Create stress_analysis.py

stress_analysis.py

@@ -0,0 +1,49 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+def stress_analysis(force, die_width, die_height, material_strength, youngs_modulus, poisson_ratio, graph_type):
+    try:
+        if die_width <= 0 or die_height <= 0:
+            return "Width and height must be greater than zero.", None
+
+        stress = force / (die_width * die_height)
+        safety_factor = material_strength / stress
+        strain = stress / youngs_modulus
+        lateral_strain = poisson_ratio * strain
+
+        fig = plt.figure(figsize=(8, 6))
+
+        if graph_type == "3D Stress Visualization":
+            ax = fig.add_subplot(111, projection='3d')
+            x = np.linspace(0, die_width, 50)
+            y = np.linspace(0, die_height, 50)
+            X, Y = np.meshgrid(x, y)
+            Z = stress * (1 - (X / die_width) ** 2 - (Y / die_height) ** 2)
+
+            surf = ax.plot_surface(X, Y, Z, cmap='viridis')
+            ax.set_xlabel('Width (m)')
+            ax.set_ylabel('Height (m)')
+            ax.set_zlabel('Stress (MPa)')
+            fig.colorbar(surf, shrink=0.5, aspect=5)
+            ax.set_title("3D Stress Visualization")
+
+        else:
+            ax = fig.add_subplot(111)
+            categories = ["Stress (MPa)", "Material Strength (MPa)", "Strain", "Lateral Strain"]
+            values = [stress, material_strength, strain, lateral_strain]
+
+            if graph_type == "Bar Plot":
+                ax.bar(categories, values, color='blue')
+                ax.set_title("Stress Analysis - Bar Plot")
+
+            elif graph_type == "Line Plot":
+                ax.plot(categories, values, marker='o', linestyle='-', color='green')
+                ax.set_title("Stress Analysis - Line Plot")
+
+        plt.tight_layout()
+        plt.close(fig)
+
+        return f"Safety Factor: {round(safety_factor, 2)}", fig
+
+    except Exception as e:
+        return f"Error in stress analysis: {str(e)}", None