Update app.py

app.py

@@ -6,31 +6,42 @@ import os
 
 # Sample simulation function for illustration purposes
 def run_simulation(material_yield, part_thickness, press_force, punch_speed, safety_factor):
-    # 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()
-
-    # 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()
-    plotter_output_path = "/tmp/3d_model_output.png"
-    plotter.screenshot(plotter_output_path)
-
-    return graph_output_path, plotter_output_path
+    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
+
+    except Exception as e:
+        print(f"Error during simulation: {e}")
+        return None, None
 
 # Gradio interface for input parameters and output visualization
 iface = gr.Interface(