Update app.py

app.py

@@ -8,45 +8,58 @@ import matplotlib.pyplot as plt
 import json
 import os
 
-# 2D View Function
-def generate_2d_view(step_file):
-    try:
-        mesh = trimesh.load(step_file.name)
-        projection = mesh.section(plane_origin=[0, 0, 0], plane_normal=[0, 0, 1])
-        projection_img = projection.to_image() if projection else None
-
-        fig, ax = plt.subplots()
-        ax.imshow(projection_img, cmap="gray")
-        ax.axis("off")
-
-        buffer = BytesIO()
-        plt.savefig(buffer, format="png")
-        buffer.seek(0)
-        return buffer
-    except Exception as e:
-        return f"Error in generating 2D view: {e}"
+# Wrapper function to generate both 2D and 3D views
+def generate_views(step_file):
+    # Call the individual functions and collect their outputs
+    two_d_view = generate_2d_view(step_file)
+    three_d_view = generate_3d_view(step_file)
+    return two_d_view, three_d_view
+
+# Updated Gradio Interface
+def main():
+    with gr.Blocks() as app:
+        gr.Markdown("# Press Tool Design and APDL Script Generator")
 
-# 3D View Function
-def generate_3d_view(step_file):
-    try:
-        model = cq.importers.importStep(step_file.name)
-        stl_file = "temp_model.stl"
-        exporters.export(model, stl_file)
+        with gr.Row():
+            step_input = gr.File(label="Upload STEP File")
 
-        mesh = pv.read(stl_file)
-        plotter = pv.Plotter(off_screen=True)
-        plotter.add_mesh(mesh, color="lightblue")
-        screenshot = plotter.screenshot(asarray=True)
-
-        buffer = BytesIO()
-        plt.imshow(screenshot)
-        plt.axis("off")
-        plt.savefig(buffer, format="png")
-        buffer.seek(0)
-        return buffer
-    except Exception as e:
-        return f"Error in generating 3D view: {e}"
+        with gr.Row():
+            two_d_output = gr.Image(label="2D View")
+            three_d_output = gr.Image(label="3D View")
+            generate_views_btn = gr.Button("Generate 2D & 3D Views")
+
+        with gr.Row():
+            force_input = gr.Number(label="Press Force (N)", value=1000)
+            material_input = gr.Textbox(
+                label="Material Properties (JSON)",
+                value='{"elastic_modulus": 2e11, "poisson": 0.3}'
+            )
+
+        with gr.Row():
+            script_output = gr.Textbox(label="Generated APDL Script", interactive=False)
+            download_button = gr.File(label="Download Script")
+            generate_apdl_btn = gr.Button("Generate APDL Script")
+
+        # Events for Visualization
+        generate_views_btn.click(
+            fn=generate_views,  # Use the wrapper function
+            inputs=step_input,
+            outputs=[two_d_output, three_d_output]
+        )
+
+        # Events for APDL Script
+        def handle_download(apdl_script, apdl_file_path):
+            if apdl_file_path and os.path.exists(apdl_file_path):
+                return apdl_file_path
+            return None
+
+        generate_apdl_btn.click(
+            fn=generate_apdl_script,
+            inputs=[step_input, force_input, material_input],
+            outputs=[script_output, download_button]
+        )
 
+    
 # APDL Script Generation
 def generate_apdl_script(file, press_force, material_json):
     try: