Create app.py

app.py

@@ -0,0 +1,134 @@
+import cadquery as cq
+import numpy as np
+import matplotlib.pyplot as plt
+import pyvista as pv
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+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:
+        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)
+        filename = "progressive_die.step"
+        cq.exporters.export(die, filename)
+        return filename
+    except Exception as e:
+        return f"Error generating die: {str(e)}"
+
+# Function to visualize die in 3D
+def visualize_die(length, width, thickness):
+    try:
+        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)
+
+        # Export to STL for visualization
+        cq.exporters.exportShape(die.val(), "STL", "progressive_die.stl")
+
+        # Visualize with PyVista
+        mesh = pv.read("progressive_die.stl")
+        plotter = pv.Plotter(off_screen=True)
+        plotter.add_mesh(mesh, color="blue")
+        screenshot = "progressive_die_visualization.png"
+        plotter.screenshot(screenshot)
+        return screenshot
+    except Exception as e:
+        return f"Error visualizing die: {str(e)}"
+
+# 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
+
+# Function to generate PDF report
+def generate_pdf_report(data, filename="report.pdf"):
+    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
+    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, 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
+
+    else:
+        return "Invalid simulation tool selected", None, None
+
+# Create Gradio App
+with gr.Blocks() as app:
+    gr.Markdown("## Press Tool AI Suite")
+    gr.Markdown("Select a tool below to get started:")
+
+    with gr.Tabs():
+        with gr.Tab("Progressive Die Design"):
+            gr.Markdown("### Enter Dimensions for Progressive Die")
+            length = gr.Number(label="Length (mm)", value=100)
+            width = gr.Number(label="Width (mm)", value=50)
+            thickness = gr.Number(label="Thickness (mm)", value=10)
+            die_output = gr.Textbox(label="Die Output File")
+            visualization_output = gr.Image(label="3D Visualization")
+            die_button = gr.Button("Generate Die")
+            die_button.click(
+                lambda l, w, t: (generate_die(l, w, t), visualize_die(l, w, t)),
+                inputs=[length, width, thickness],
+                outputs=[die_output, visualization_output],
+            )
+
+        with gr.Tab("Stress Analysis"):
+            gr.Markdown("### Select Simulation Tool and Enter Parameters for Stress Analysis")
+            simulation_tool = gr.Dropdown(
+                choices=["Python"],
+                label="Simulation Tool",
+                value="Python",
+            )
+            force = gr.Number(label="Force (N)", value=10000)
+            die_width = gr.Number(label="Width (m)", value=0.05)
+            die_height = gr.Number(label="Height (m)", value=0.01)
+            material_strength = gr.Number(label="Material Strength (MPa)", value=250)
+            safety_factor_output = gr.Textbox(label="Safety Factor or Simulation Result")
+            stress_chart = gr.Plot()
+            pdf_file = gr.File(label="Download Report (Python Only)")
+            stress_button = gr.Button("Analyze Stress")
+            stress_button.click(
+                stress_analysis_interface,
+                inputs=[force, die_width, die_height, material_strength, simulation_tool],
+                outputs=[safety_factor_output, stress_chart, pdf_file],
+            )
+
+# Launch the app
+app.launch()