Update app.py

app.py

@@ -9,6 +9,29 @@ import json
 import atexit
 import glob
 
+def process_cad_file(file):
+    # Check if the file exists and is not empty
+    if not os.path.exists(file.name):
+        raise ValueError("Uploaded file is missing or does not exist.")
+    if os.path.getsize(file.name) == 0:
+        raise ValueError("Uploaded file is empty.")
+    
+    # Validate file format
+    file_extension = os.path.splitext(file.name)[-1].lower()
+    if file_extension not in [".stp", ".step", ".iges", ".igs"]:
+        raise ValueError(f"Unsupported file format: {file_extension}")
+    
+    # Process the valid CAD file
+    try:
+        if file_extension in [".stp", ".step"]:
+            return cq.importers.importStep(file.name)
+        elif file_extension in [".iges", ".igs"]:
+            return cq.importers.importStep(file.name)  # Same handler for now
+        else:
+            raise ValueError("Invalid file type detected during processing.")
+    except Exception as e:
+        raise RuntimeError(f"Error processing the CAD file: {str(e)}")
+
 # Cleanup function to remove temporary files
 def cleanup_temp_files():
     temp_files = glob.glob("/tmp/*.svg") + glob.glob("/tmp/*.obj") + glob.glob("/tmp/*.stl") + glob.glob("/tmp/*.inp")