Update app.py

app.py

@@ -4,11 +4,7 @@ import tempfile
 from io import BytesIO
 from fpdf import FPDF
 import requests
-import json
-import pythonocc
-from pathlib import Path
-from OCC.Extend.DataExchange import read_step_file
-from OCC.Core.STEPControl import STEPControl_Reader
+from stl import mesh
 
 # Groq API endpoint and headers
 GROQ_API_URL = "https://api.groq.com/v1/query"
@@ -36,22 +32,16 @@ def generate_report(cad_data):
         st.error(f"Error in report generation: {response.text}")
         return "Error generating report."
 
-# Function to parse STEP CAD files using pythonOCC
-def parse_cad_file(file_path):
-    # Read the STEP file and extract data using pythonOCC
-    step_reader = STEPControl_Reader()
-    status = step_reader.ReadFile(str(file_path))
+# Function to parse STL CAD files using pySTL
+def parse_stl_file(file_path):
+    # Load the STL file and extract basic information
+    cad_mesh = mesh.Mesh.from_file(file_path)
     
-    if status != 1:  # If the file couldn't be read properly
-        st.error("Failed to read CAD file.")
-        return None
+    # Example: Get the number of faces and volume
+    num_faces = cad_mesh.vectors.shape[0]
+    volume = cad_mesh.get_mass_properties()[0]  # Get volume from mass properties
     
-    step_reader.TransferRoot()
-    shape = step_reader.Shape()
-    
-    # Extract basic information from the shape
-    # Here we can extract dimension, material properties, and more based on the available data
-    cad_data = f"Extracted information from {file_path.name}: \n - Shape contains {shape.NbChildren()} components.\n - Additional analysis can be performed."
+    cad_data = f"Extracted information from {file_path.name}:\n - Number of Faces: {num_faces}\n - Volume: {volume} cubic units."
     return cad_data
 
 # Function to create a PDF report
@@ -78,7 +68,7 @@ def create_pdf_report(report_text):
 st.title("CAD File Analysis and Technical Report Generation")
 
 # File uploader
-uploaded_file = st.file_uploader("Upload a CAD file", type=["stl", "step", "dxf", "iges"])
+uploaded_file = st.file_uploader("Upload a CAD file", type=["stl"])
 
 if uploaded_file is not None:
     # Save uploaded file temporarily
@@ -86,8 +76,8 @@ if uploaded_file is not None:
     with open(temp_file_path, "wb") as f:
         f.write(uploaded_file.getbuffer())
 
-    # Parse the CAD file and extract relevant data
-    cad_data = parse_cad_file(temp_file_path)
+    # Parse the STL file and extract relevant data
+    cad_data = parse_stl_file(temp_file_path)
 
     if cad_data:
         # Display extracted data as a preview