Update app.py

app.py

@@ -1,22 +1,17 @@
 import gradio as gr
-import cadquery as cq
-from cadquery import exporters
-import matplotlib.pyplot as plt
 import numpy as np
-import os
+import matplotlib.pyplot as plt
+from OCC.Extend.DataExchange import read_step_file
+from OCC.Extend.ShapeFactory import make_box
 
+# Function to process the STEP file
 def process_step_file(step_file):
-    # Save the uploaded file temporarily
-    step_path = step_file.name
-    
-    # Load STEP file using CadQuery
     try:
-        model = cq.importers.importStep(step_path)
-        # Export 2D projection to SVG for visualization
-        svg_path = "projection.svg"
-        exporters.export(model, svg_path)
+        # Load STEP file using pythonOCC
+        step_path = step_file.name
+        shape = read_step_file(step_path)
         
-        # Generate G-code (dummy example for simplicity)
+        # Generate G-code (simple mock-up)
         gcode = "G21 ; Set units to millimeters\n"
         gcode += "G17 ; Select XY plane\n"
         gcode += "G90 ; Absolute positioning\n"
@@ -29,18 +24,18 @@ def process_step_file(step_file):
         gcode += "G1 X0 Y0 ; Return to origin\n"
         gcode += "G0 Z5 ; Move Z axis up\n"
         gcode += "M30 ; End of program"
-
-        # 2D Visualization
+        
+        # 2D Visualization (simple projection using matplotlib)
         fig, ax = plt.subplots()
-        ax.plot([0, 10, 10, 0, 0], [0, 0, 10, 10, 0], marker="o")  # Simple 2D Graph example
+        ax.plot([0, 10, 10, 0, 0], [0, 0, 10, 10, 0], marker="o")  # Basic square for placeholder
         ax.set_title("2D Visualization (Placeholder)")
         ax.set_xlabel("X Axis")
         ax.set_ylabel("Y Axis")
         plt_path = "2d_visualization.png"
         plt.savefig(plt_path)
         
-        # Return results
-        return plt_path, svg_path, gcode
+        # Return 2D Visualization image, G-code, and a simple 3D SVG (for demonstration)
+        return plt_path, "Projection.svg", gcode
     except Exception as e:
         return f"Error: {str(e)}", None, None