Update app.py

app.py

@@ -5,9 +5,18 @@ import requests
 # Replace with your actual Groq API endpoint URL
 GROQ_API_URL = "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"
 
+
 def process_model(model_path):
     # Load the 3D model using trimesh
-    mesh = trimesh.load(model_path)
+    try:
+        mesh = trimesh.load(model_path)
+    except FileNotFoundError:
+        st.error(f"Error: File not found at path '{model_path}'.")
+        return None
+    except (trimesh.exceptions.ImportError, ValueError) as e:
+        # Handle potential format-specific errors (e.g., STEP format issues)
+        st.error(f"Error loading model: {e}")
+        return None
 
     # Extract features (replace with relevant features for material estimation)
     features = {
@@ -31,19 +40,18 @@ def process_model(model_path):
         st.error(error_message)
         return None
 
+
 def main():
-    st.title("QUICK Cad Model Analyzer App")
+    st.title("RAG-based Material Estimation App (Groq API)")
 
-    uploaded_file = st.file_uploader("Upload 3D CAD Model (Supported formats: .stl, .obj, etc.)")
+    uploaded_file = st.file_uploader("Upload 3D CAD Model (Supported formats: .stl, .obj, .stp, etc.)")
 
     if uploaded_file is not None:
         model_path = uploaded_file.name
     else:
-        st.stop()
+        model_path = None  # Set model_path to None if no file is uploaded
 
-    sheet_length = st.number_input("Steel Sheet Length (mm)", min_value=0.0)
-    sheet_width = st.number_input("Steel Sheet Width (mm)", min_value=0.0)
-    quantity = st.number_input("Quantity to Manufacture", min_value=1)
+    # ... (rest of your code for sheet dimensions and quantity)
 
     if st.button("Estimate Material Requirements"):
         if model_path:
@@ -51,5 +59,6 @@ def main():
             if material_requirements:
                 st.success(f"Estimated Material Requirements:\n{material_requirements}")
 
+
 if __name__ == "__main__":
     main()