Update app.py

app.py

@@ -2,6 +2,11 @@ import cadquery as cq
 import numpy as np
 import matplotlib.pyplot as plt
 import gradio as gr
+import os
+
+# Ensure there's a directory for saving files
+if not os.path.exists("generated_files"):
+    os.makedirs("generated_files")
 
 # Function for Progressive Die Design
 def generate_die(length, width, thickness):
@@ -12,17 +17,21 @@ def generate_die(length, width, thickness):
         die = plate.cut(punch)
         
         # Save the die to a file
-        filename = "progressive_die.step"
+        filename = "generated_files/progressive_die.step"
         cq.exporters.export(die, filename)
         
-        # Return success message
-        return f"Progressive die design saved as {filename}. You can download the file."
+        # Return success message and file path for download
+        return f"Progressive die design saved. [Download the file](/{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):
     try:
+        # Validate inputs to avoid invalid calculations
+        if force <= 0 or die_width <= 0 or die_height <= 0 or material_strength <= 0:
+            return "All values must be positive."
+
         # Calculate stress and safety factor
         stress = force / (die_width * die_height)
         safety_factor = material_strength / stress
@@ -43,8 +52,10 @@ def stress_analysis(force, die_width, die_height, material_strength):
 # Function for Tool Optimization
 def optimize_tool(speed, feed_rate, depth_of_cut, material):
     try:
-        # Estimate tool life and recommend adjustments
-        tool_life = 1000 / (speed * feed_rate * depth_of_cut)
+        # Adjust tool life estimate based on material
+        material_factor = {"Steel": 1.0, "Aluminum": 0.8, "Titanium": 1.2}
+        tool_life = 1000 / (speed * feed_rate * depth_of_cut * material_factor.get(material, 1.0))
+        
         recommended_speed = 0.8 * speed
         recommended_feed_rate = 0.9 * feed_rate