Update app.py

app.py

@@ -5,7 +5,10 @@ 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
 
 # Groq API endpoint and headers
 GROQ_API_URL = "https://api.groq.com/v1/query"
@@ -33,10 +36,22 @@ def generate_report(cad_data):
         st.error(f"Error in report generation: {response.text}")
         return "Error generating report."
 
-# Function to parse and extract relevant data from CAD files
+# Function to parse STEP CAD files using pythonOCC
 def parse_cad_file(file_path):
-    # Placeholder: Extract relevant data from CAD file (This can be improved with actual CAD parsing libraries)
-    cad_data = f"Extracted information from {file_path.name}: \n - Dimensions: 100mm x 50mm\n - Material: Steel\n - Weight: 1.5kg"
+    # Read the STEP file and extract data using pythonOCC
+    step_reader = STEPControl_Reader()
+    status = step_reader.ReadFile(str(file_path))
+    
+    if status != 1:  # If the file couldn't be read properly
+        st.error("Failed to read CAD file.")
+        return None
+    
+    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."
     return cad_data
 
 # Function to create a PDF report
@@ -74,26 +89,27 @@ if uploaded_file is not None:
     # Parse the CAD file and extract relevant data
     cad_data = parse_cad_file(temp_file_path)
 
-    # Display extracted data as a preview
-    st.write("Extracted Data from CAD File:")
-    st.text(cad_data)
+    if cad_data:
+        # Display extracted data as a preview
+        st.write("Extracted Data from CAD File:")
+        st.text(cad_data)
 
-    # Generate report using Groq API
-    report_text = generate_report(cad_data)
-    st.write("Generated Report:")
-    st.text(report_text)
+        # Generate report using Groq API
+        report_text = generate_report(cad_data)
+        st.write("Generated Report:")
+        st.text(report_text)
 
-    # Create a PDF report
-    pdf_file_path = create_pdf_report(report_text)
+        # Create a PDF report
+        pdf_file_path = create_pdf_report(report_text)
 
-    # Provide a download link for the PDF report
-    with open(pdf_file_path, "rb") as f:
-        pdf_bytes = f.read()
-        st.download_button(
-            label="Download Technical Report",
-            data=pdf_bytes,
-            file_name="technical_report.pdf",
-            mime="application/pdf"
-        )
+        # Provide a download link for the PDF report
+        with open(pdf_file_path, "rb") as f:
+            pdf_bytes = f.read()
+            st.download_button(
+                label="Download Technical Report",
+                data=pdf_bytes,
+                file_name="technical_report.pdf",
+                mime="application/pdf"
+            )
 else:
     st.write("Please upload a CAD file to begin.")