Update app.py

app.py

@@ -2,21 +2,34 @@ import gradio as gr
 from ai_models import train_models, load_models
 from ansys_simulation import run_ansys_simulation
 from visualization import generate_graph
+import os
 
 # Load AI models
 stress_model, deformation_model, pass_fail_model = load_models()
 
 def ai_and_simulation_workflow(thickness, hole_diameter, force):
     # AI Pre-Screening
-    pre_screen = pass_fail_model.predict([[thickness, hole_diameter, force]])[0]
+    try:
+        pre_screen = pass_fail_model.predict([[thickness, hole_diameter, force]])[0]
+    except Exception as e:
+        return f"Error with AI model prediction: {str(e)}", None
+
     if pre_screen == 0:
         return "AI predicts failure. Adjust parameters.", None
     
     # ANSYS Simulation
-    max_stress, max_deformation = run_ansys_simulation(thickness, hole_diameter, force)
+    try:
+        max_stress, max_deformation = run_ansys_simulation(thickness, hole_diameter, force)
+    except Exception as e:
+        return f"Error with ANSYS simulation: {str(e)}", None
 
     # Generate Visualization
-    graph_path = generate_graph(thickness, hole_diameter, force, max_stress, max_deformation)
+    try:
+        graph_path = generate_graph(thickness, hole_diameter, force, max_stress, max_deformation)
+        if not os.path.exists(graph_path):
+            return f"Error: Graph path '{graph_path}' does not exist.", None
+    except Exception as e:
+        return f"Error with graph generation: {str(e)}", None
 
     return f"Pass: Stress={max_stress} Pa, Deformation={max_deformation} mm", graph_path