Update app.py

app.py

@@ -6,11 +6,16 @@ import gradio as gr
 # Function for Progressive Die Design
 def generate_die(length, width, thickness):
     try:
+        # Generate the die design
         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)
+        
+        # Save the die to a file
         filename = "progressive_die.step"
         cq.exporters.export(die, filename)
+        
+        # Return success message
         return f"Progressive die design saved as {filename}. You can download the file."
     except Exception as e:
         return f"Error generating die: {str(e)}"
@@ -18,17 +23,19 @@ def generate_die(length, width, thickness):
 # Function for Stress Analysis
 def stress_analysis(force, die_width, die_height, material_strength):
     try:
+        # Calculate stress and safety factor
         stress = force / (die_width * die_height)
         safety_factor = material_strength / stress
         
+        # Create stress analysis plot
         fig, ax = plt.subplots()
         ax.bar(["Stress", "Material Strength"], [stress, material_strength])
         ax.set_ylabel("Stress (MPa)")
         ax.set_title("Stress Analysis")
-        
         plt.tight_layout()  # Ensure the plot is displayed correctly
         plt.close(fig)  # Close the plot so it doesn't render twice
         
+        # Return the results and plot
         return f"Safety Factor: {round(safety_factor, 2)}", fig
     except Exception as e:
         return f"Error in stress analysis: {str(e)}", None
@@ -36,10 +43,12 @@ 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)
         recommended_speed = 0.8 * speed
         recommended_feed_rate = 0.9 * feed_rate
         
+        # Return optimization results
         return {
             "Estimated Tool Life (hrs)": round(tool_life, 2),
             "Recommended Speed (m/min)": round(recommended_speed, 2),
@@ -64,6 +73,7 @@ with gr.Blocks() as app:
     gr.Markdown("Select a tool below to get started:")
     
     with gr.Tabs():
+        # Tab for Progressive Die Design
         with gr.Tab("Progressive Die Design"):
             gr.Markdown("### Enter Dimensions for Progressive Die")
             length = gr.Number(label="Length (mm)", value=100)
@@ -73,6 +83,7 @@ with gr.Blocks() as app:
             die_button = gr.Button("Generate Die")
             die_button.click(progressive_die_interface, inputs=[length, width, thickness], outputs=die_output)
         
+        # Tab for Stress Analysis
         with gr.Tab("Stress Analysis"):
             gr.Markdown("### Enter Parameters for Stress Analysis")
             force = gr.Number(label="Force (N)", value=10000)
@@ -84,6 +95,7 @@ with gr.Blocks() as app:
             stress_button = gr.Button("Analyze Stress")
             stress_button.click(stress_analysis_interface, inputs=[force, die_width, die_height, material_strength], outputs=[safety_factor_output, stress_chart])
         
+        # Tab for Tool Optimization
         with gr.Tab("Tool Optimization"):
             gr.Markdown("### Enter Machining Parameters for Tool Optimization")
             speed = gr.Number(label="Cutting Speed (m/min)", value=100)