Update app.py

app.py

@@ -42,7 +42,7 @@ def visualize_die(length, width, thickness):
         return f"Error visualizing die: {str(e)}"
 
 
-# Function for Stress Analysis (including thermal stress and fatigue strength)
+# Stress Analysis Function
 def stress_analysis(force, die_width, die_height, material_strength, temperature_change=50, alpha=1e-5, elastic_modulus=200000, fatigue_strength=150):
     try:
         # Mechanical stress
@@ -86,25 +86,26 @@ def stress_analysis(force, die_width, die_height, material_strength, temperature
         return f"Error in stress analysis: {str(e)}", None
 
 
-# Function to generate PDF report
-def generate_pdf_report(data, filename="report.pdf"):
+# Tool Optimization Function
+def optimize_tool(speed, feed_rate, depth_of_cut, material):
+    """
+    Optimizes machining parameters for maximum tool life.
+    """
     try:
-        c = canvas.Canvas(filename, pagesize=letter)
-        c.drawString(100, 750, "Simulation Report")
-        c.drawString(100, 730, f"Max Stress: {data.get('stress', 'N/A')} MPa")
-        c.drawString(100, 710, f"Safety Factor: {data.get('safety_factor', 'N/A')}")
-        c.save()
-        return filename
+        # Simple formula for tool life estimation
+        tool_life = 1000 / (speed * feed_rate * depth_of_cut)
+
+        # Recommend adjustments for better tool life
+        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 f"Error generating report: {str(e)}"
-
-
-# Gradio interface functions
-def stress_analysis_interface(force, die_width, die_height, material_strength, temperature_change, alpha, elastic_modulus, fatigue_strength):
-    safety_factor, fig = stress_analysis(force, die_width, die_height, material_strength, temperature_change, alpha, elastic_modulus, fatigue_strength)
-    data = {"stress": force / (die_width * die_height), "safety_factor": safety_factor}
-    pdf_filename = generate_pdf_report(data)
-    return safety_factor, fig, pdf_filename
+        return {"Error": str(e)}
 
 
 # Create Gradio App
@@ -140,12 +141,25 @@ with gr.Blocks() as app:
 
             safety_factor_output = gr.Textbox(label="Safety Factor")
             stress_chart = gr.Plot()
-            pdf_file = gr.File(label="Download Report")
             stress_button = gr.Button("Analyze Stress")
             stress_button.click(
-                stress_analysis_interface,
+                lambda f, dw, dh, ms, tc, a, em, fs: stress_analysis(f, dw, dh, ms, tc, a, em, fs),
                 inputs=[force, die_width, die_height, material_strength, temperature_change, alpha, elastic_modulus, fatigue_strength],
-                outputs=[safety_factor_output, stress_chart, pdf_file],
+                outputs=[safety_factor_output, stress_chart],
+            )
+
+        with gr.Tab("Tool Optimization"):
+            gr.Markdown("### Enter Machining Parameters for Tool Optimization")
+            speed = gr.Number(label="Cutting Speed (m/min)", value=100)
+            feed_rate = gr.Number(label="Feed Rate (mm/rev)", value=0.2)
+            depth_of_cut = gr.Number(label="Depth of Cut (mm)", value=1.0)
+            material = gr.Dropdown(choices=["Steel", "Aluminum", "Titanium"], label="Material", value="Steel")
+            optimization_results = gr.JSON(label="Optimization Results")
+            optimize_button = gr.Button("Optimize Tool")
+            optimize_button.click(
+                lambda s, fr, dc, m: optimize_tool(s, fr, dc, m),
+                inputs=[speed, feed_rate, depth_of_cut, material],
+                outputs=optimization_results,
             )
 
 # Launch the app