Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import os
 from ansys.mapdl.core import launch_mapdl  # For ANSYS integration
 
+
 # Function for Progressive Die Design
 def generate_die(length, width, thickness):
     try:
@@ -20,6 +21,7 @@ def generate_die(length, width, thickness):
     except Exception as e:
         return f"Error generating die: {str(e)}"
 
+
 # Function to visualize die in 3D
 def visualize_die(length, width, thickness):
     try:
@@ -40,31 +42,71 @@ def visualize_die(length, width, thickness):
     except Exception as e:
         return f"Error visualizing die: {str(e)}"
 
-# Function for Python-based Stress Analysis with a combined graph
+
+# Function for Python-based Stress Analysis
 def stress_analysis(force, die_width, die_height, material_strength):
     try:
-        # Calculate parameters
-        stress = force / (die_width * die_height)  # Stress = Force / Area
-        safety_factor = material_strength / stress  # Safety Factor = Material Strength / Stress
-
-        # Data for plotting
-        x = ["Stress", "Material Strength", "Safety Factor"]
-        y_stress = [stress, material_strength, safety_factor]
+        stress = force / (die_width * die_height)
+        safety_factor = material_strength / stress
 
-        # Create a combined graph
         fig, ax = plt.subplots()
-        ax.plot(x, y_stress, marker='o', label="Simulation Parameters")
-        ax.axhline(y=1, color='red', linestyle='--', label="Critical Safety Factor (1)")
-        ax.set_ylabel("Value")
-        ax.set_title("Stress Analysis Parameters")
-        ax.legend()
-        plt.tight_layout()
+        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
 
+
+# ANSYS Integration for Stress Analysis
+def run_ansys_simulation(force, die_width, die_height, material_strength):
+    try:
+        # Launch ANSYS MAPDL instance
+        mapdl = launch_mapdl()
+        mapdl.prep7()  # Enter pre-processing module
+
+        # Define geometry and material properties
+        mapdl.rectng(0, die_width, 0, die_height)
+        mapdl.mp('EX', 1, material_strength)
+        mapdl.et(1, 'PLANE183')
+
+        # Apply loads
+        mapdl.nsel('S', 'LOC', 'X', 0)
+        mapdl.d('ALL', 'UX', 0)
+        mapdl.f('ALL', 'FY', -force)
+
+        # Solve
+        mapdl.run('/SOLU')
+        mapdl.solve()
+        mapdl.finish()
+
+        # Post-processing
+        mapdl.post1()
+        stress = mapdl.get_value('NODE', 1, 'S', 'EQV')  # Retrieve max equivalent stress
+        mapdl.exit()
+
+        return f"Max Stress: {stress:.2f} MPa"
+    except Exception as e:
+        return f"Error running ANSYS simulation: {str(e)}"
+
+
+# SolidWorks Placeholder for Stress Analysis
+def solidworks_stress_analysis(force, die_width, die_height, material_strength):
+    # Replace this function with SolidWorks API or external script
+    try:
+        output_file = "/path/to/solidworks/output.txt"  # Replace with actual path
+        if os.path.exists(output_file):
+            with open(output_file, "r") as file:
+                result = file.read()
+            return result.strip()
+        else:
+            return "SolidWorks simulation output not found."
+    except Exception as e:
+        return f"Error running SolidWorks simulation: {str(e)}"
+
+
 # Function to generate PDF report
 def generate_pdf_report(data, filename="report.pdf"):
     try:
@@ -77,18 +119,46 @@ def generate_pdf_report(data, filename="report.pdf"):
     except Exception as e:
         return f"Error generating report: {str(e)}"
 
+
+# Tool Optimization Function
+def optimize_tool(speed, feed_rate, depth_of_cut, material):
+    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 stress_analysis_interface(force, die_width, die_height, material_strength, simulation_tool):
     if simulation_tool == "Python":
-        # Python-based stress analysis with combined graph
+        # Python-based stress analysis
         safety_factor, fig = stress_analysis(force, die_width, die_height, material_strength)
         data = {"stress": force / (die_width * die_height), "safety_factor": safety_factor}
         pdf_filename = generate_pdf_report(data)
         return safety_factor, fig, pdf_filename
 
+    elif simulation_tool == "ANSYS":
+        # Run ANSYS-based simulation
+        result = run_ansys_simulation(force, die_width, die_height, material_strength)
+        return result, None, None
+
+    elif simulation_tool == "SolidWorks":
+        # Run SolidWorks-based simulation
+        result = solidworks_stress_analysis(force, die_width, die_height, material_strength)
+        return result, None, None
+
     else:
         return "Invalid simulation tool selected", None, None
 
+
 # Create Gradio App
 with gr.Blocks() as app:
     gr.Markdown("## Press Tool AI Suite")
@@ -112,7 +182,7 @@ with gr.Blocks() as app:
         with gr.Tab("Stress Analysis"):
             gr.Markdown("### Select Simulation Tool and Enter Parameters for Stress Analysis")
             simulation_tool = gr.Dropdown(
-                choices=["Python"],
+                choices=["Python", "ANSYS", "SolidWorks"],
                 label="Simulation Tool",
                 value="Python",
             )
@@ -129,6 +199,65 @@ with gr.Blocks() as app:
                 inputs=[force, die_width, die_height, material_strength, simulation_tool],
                 outputs=[safety_factor_output, stress_chart, pdf_file],
             )
+            # Function for Python-based Stress Analysis with a combined graph
+def stress_analysis(force, die_width, die_height, material_strength):
+    try:
+        # Calculate parameters
+        stress = force / (die_width * die_height)  # Stress = Force / Area
+        safety_factor = material_strength / stress  # Safety Factor = Material Strength / Stress
+
+        # Data for plotting
+        x = ["Stress", "Material Strength", "Safety Factor"]
+        y_stress = [stress, material_strength, safety_factor]
+
+        # Create a combined graph
+        fig, ax = plt.subplots()
+        ax.plot(x, y_stress, marker='o', label="Simulation Parameters")
+        ax.axhline(y=1, color='red', linestyle='--', label="Critical Safety Factor (1)")
+        ax.set_ylabel("Value")
+        ax.set_title("Stress Analysis Parameters")
+        ax.legend()
+        plt.tight_layout()
+        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
+def stress_analysis_interface(force, die_width, die_height, material_strength, simulation_tool):
+    if simulation_tool == "Python":
+        # Python-based stress analysis with combined graph
+        safety_factor, fig = stress_analysis(force, die_width, die_height, material_strength)
+        data = {"stress": force / (die_width * die_height), "safety_factor": safety_factor}
+        pdf_filename = generate_pdf_report(data)
+        return safety_factor, fig, pdf_filename
+
+    elif simulation_tool == "ANSYS":
+        # Run ANSYS-based simulation
+        result = run_ansys_simulation(force, die_width, die_height, material_strength)
+        return result, None, None
+
+    elif simulation_tool == "SolidWorks":
+        # Run SolidWorks-based simulation
+        result = solidworks_stress_analysis(force, die_width, die_height, material_strength)
+        return result, None, None
+
+    else:
+        return "Invalid simulation tool selected", None, None
+
+
+        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(
+                optimize_tool,
+                inputs=[speed, feed_rate, depth_of_cut, material],
+                outputs=optimization_results,
+            )
 
 # Launch the app
 app.launch()