Update app.py

app.py

@@ -5,37 +5,46 @@ import gradio as gr
 
 # Function for Progressive Die Design
 def generate_die(length, width, thickness):
-    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)
-    filename = "progressive_die.step"
-    cq.exporters.export(die, filename)
-    return f"Progressive die design saved as {filename}"
+    try:
+        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)
+        filename = "progressive_die.step"
+        cq.exporters.export(die, filename)
+        return f"Progressive die design saved as {filename}"
+    except Exception as e:
+        return f"Error generating die: {str(e)}"
 
 # Function for Stress Analysis
 def stress_analysis(force, die_width, die_height, material_strength):
-    stress = force / (die_width * die_height)
-    safety_factor = material_strength / stress
-    
-    fig, ax = plt.subplots()
-    ax.bar(["Stress", "Material Strength"], [stress, material_strength])
-    ax.set_ylabel("Stress (MPa)")
-    plt.title("Stress Analysis")
-    plt.close(fig)
-    
-    return f"Safety Factor: {round(safety_factor, 2)}", fig
+    try:
+        stress = force / (die_width * die_height)
+        safety_factor = material_strength / stress
+        
+        fig, ax = plt.subplots()
+        ax.bar(["Stress", "Material Strength"], [stress, material_strength])
+        ax.set_ylabel("Stress (MPa)")
+        ax.set_title("Stress Analysis")
+        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
 
 # Function for Tool Optimization
 def optimize_tool(speed, feed_rate, depth_of_cut, material):
-    tool_life = 1000 / (speed * feed_rate * depth_of_cut)
-    recommended_speed = 0.8 * speed
-    recommended_feed_rate = 0.9 * feed_rate
-    
-    return {
-        "Estimated Tool Life (hrs)": round(tool_life, 2),
-        "Recommended Speed (m/min)": round(recommended_speed, 2),
-        "Recommended Feed Rate (mm/rev)": round(recommended_feed_rate, 2)
-    }
+    try:
+        tool_life = 1000 / (speed * feed_rate * depth_of_cut)
+        recommended_speed = 0.8 * speed
+        recommended_feed_rate = 0.9 * feed_rate
+        
+        return {
+            "Estimated Tool Life (hrs)": round(tool_life, 2),
+            "Recommended Speed (m/min)": round(recommended_speed, 2),
+            "Recommended Feed Rate (mm/rev)": round(recommended_feed_rate, 2)
+        }
+    except Exception as e:
+        return {"Error": str(e)}
 
 # Gradio interface functions
 def progressive_die_interface(length, width, thickness):