Update app.py

app.py

@@ -0,0 +1,171 @@
+import gradio as gr
+import numpy as np
+import matplotlib.pyplot as plt
+import cadquery as cq
+import pyvista as pv
+from ansys.mapdl.core import launch_mapdl
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+import os
+
+# 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 to generate and display combined graph for multiple parameters
+def generate_combined_graph(force, die_width, die_height, material_strength, punch_size):
+    try:
+        # Stress = Force / Area (Die Width * Die Height)
+        stress = force / (die_width * die_height)
+        safety_factor = material_strength / stress  # Safety Factor = Material Strength / Stress
+
+        # Tool life calculation (simplified estimation)
+        tool_life = 1000 / (force * punch_size)
+
+        # Data for plotting (Multiple parameters)
+        x = np.array([force, die_width, die_height, punch_size])
+        y_stress = np.array([stress, safety_factor, tool_life])
+        
+        # Multiple parameters (force vs material strength, safety factor vs stress, etc.)
+        plt.figure(figsize=(10, 6))
+        
+        # Plot multiple curves
+        plt.plot(x, y_stress, label="Stress vs Force and Tool Life")
+        plt.plot(x, np.array([safety_factor] * len(x)), label="Safety Factor vs Stress", linestyle='--')
+        
+        # Labels and Titles
+        plt.xlabel("Force, Die Width, Die Height, Punch Size")
+        plt.ylabel("Values")
+        plt.title("Combined Simulation Parameters for Die Design and Stress Analysis")
+        
+        # Adding legend
+        plt.legend()
+        
+        # Display the graph
+        plt.tight_layout()
+        plt.close()
+        
+        return "Graph generated successfully."
+
+    except Exception as e:
+        return f"Error in generating graph: {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 for Progressive Die Design and Stress Analysis
+def stress_analysis_interface(force, die_width, die_height, material_strength, simulation_tool):
+    if simulation_tool == "Python":
+        # Generate and plot combined graph
+        result = generate_combined_graph(force, die_width, die_height, material_strength, punch_size=10)
+        return result  # Show graph or return result for Gradio output
+    
+    elif simulation_tool == "ANSYS":
+        # Run ANSYS-based simulation
+        result = run_ansys_simulation(force, die_width, die_height, material_strength)
+        return result
+
+    elif simulation_tool == "SolidWorks":
+        # Run SolidWorks-based simulation
+        result = solidworks_stress_analysis(force, die_width, die_height, material_strength)
+        return result
+
+    else:
+        return "Invalid simulation tool selected", None
+
+# Gradio Interface for Press Tool AI Suite
+with gr.Blocks() as app:
+    gr.Markdown("## Press Tool AI Suite")
+    
+    with gr.Tabs():
+        with gr.Tab("Progressive Die Design"):
+            # Inputs for Progressive Die Design
+            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"):
+            # Inputs for Stress Analysis
+            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)
+            simulation_tool = gr.Dropdown(choices=["Python", "ANSYS", "SolidWorks"], label="Simulation Tool", value="Python")
+            
+            stress_output = gr.Textbox(label="Simulation Result")
+            stress_chart = gr.Plot()  # This will display the combined graph
+            stress_button = gr.Button("Analyze Stress and Visualize")
+            stress_button.click(
+                stress_analysis_interface,
+                inputs=[force, die_width, die_height, material_strength, simulation_tool],
+                outputs=[stress_output, stress_chart],
+            )
+
+        with gr.Tab("Tool Optimization"):
+            # Inputs 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()