Update app.py

app.py

@@ -31,10 +31,17 @@ def recommend(features, feature_list):
 # Function to save uploaded file
 def save_uploaded_file(uploaded_file):
     try:
-        with open(os.path.join('uploads', uploaded_file.name), 'wb') as f:
+        # Ensure the uploads directory exists
+        if not os.path.exists('uploads'):
+            os.makedirs('uploads')
+        
+        file_path = os.path.join('uploads', uploaded_file.name)
+        with open(file_path, 'wb') as f:
             f.write(uploaded_file.getbuffer())
+        st.success(f"File saved to {file_path}")
         return True
-    except:
+    except Exception as e:
+        st.error(f"Error saving file: {e}")
         return False
 
 # Function to show dashboard content
@@ -49,8 +56,12 @@ def show_dashboard():
         GlobalMaxPooling2D()
     ])
 
-    feature_list = np.array(pickle.load(open('embeddings.pkl', 'rb')))
-    filenames = pickle.load(open('filenames.pkl', 'rb'))
+    try:
+        feature_list = np.array(pickle.load(open('embeddings.pkl', 'rb')))
+        filenames = pickle.load(open('filenames.pkl', 'rb'))
+    except Exception as e:
+        st.error(f"Error loading pickle files: {e}")
+        return
 
     # File upload section
     uploaded_file = st.file_uploader("Choose an image")
@@ -78,9 +89,8 @@ def show_dashboard():
                 st.image(filenames[indices[0][3]])
             with col5:
                 st.image(filenames[indices[0][4]])
-
         else:
-            st.header("Some error occurred in file upload")
+            st.error("Some error occurred in file upload")
 
     # Chatbot section
     user_question = st.text_input("Ask a question:")