Update app.py

app.py

@@ -22,6 +22,39 @@ with gr.Blocks() as app:
                 inputs=[length, width, thickness],
                 outputs=[die_output, visualization_output],
             )
+        import gradio as gr
+from design import generate_die, visualize_die
+
+# Gradio UI for Progressive Die Design
+with gr.Blocks() as app:
+    gr.Markdown("## Progressive Die Designer")
+
+    # User input fields for die dimensions
+    length = gr.Number(label="Length (mm)", value=100)
+    width = gr.Number(label="Width (mm)", value=50)
+    thickness = gr.Number(label="Thickness (mm)", value=10)
+
+    # Outputs: STEP file location and 3D visualization image
+    die_output = gr.Textbox(label="STEP File Location")
+    visualization_output = gr.Image(label="3D Visualization")
+
+    # Button to generate die and visualization
+    die_button = gr.Button("Generate Die")
+
+    # Connect the button to both functions
+    def generate_outputs(l, w, t):
+        step_file = generate_die(l, w, t)
+        visualization_file = visualize_die(l, w, t)
+        return step_file, visualization_file
+
+    die_button.click(
+        generate_outputs,
+        inputs=[length, width, thickness],
+        outputs=[die_output, visualization_output],
+    )
+
+# Launch the app
+app.launch()
 
         with gr.Tab("Stress Analysis"):
             gr.Markdown("### Select Simulation Tool and Enter Parameters for Stress Analysis")