Update app.py

app.py

@@ -11,7 +11,7 @@ def generate_die(length, width, thickness):
         die = plate.cut(punch)
         filename = "progressive_die.step"
         cq.exporters.export(die, filename)
-        return f"Progressive die design saved as {filename}"
+        return f"Progressive die design saved as {filename}. You can download the file."
     except Exception as e:
         return f"Error generating die: {str(e)}"
 
@@ -25,7 +25,9 @@ def stress_analysis(force, die_width, die_height, material_strength):
         ax.bar(["Stress", "Material Strength"], [stress, material_strength])
         ax.set_ylabel("Stress (MPa)")
         ax.set_title("Stress Analysis")
-        plt.close(fig)
+        
+        plt.tight_layout()  # Ensure the plot is displayed correctly
+        plt.close(fig)  # Close the plot so it doesn't render twice
         
         return f"Safety Factor: {round(safety_factor, 2)}", fig
     except Exception as e: