Update app.py

app.py

@@ -39,9 +39,22 @@ def create_vector_db(uploaded_files):
             model_kwargs={'device': 'cpu'}
         )
 
+        # Create the vectorstore directory if it doesn't exist
+        if not os.path.exists(DB_FAISS_PATH):
+            try:
+                os.makedirs(DB_FAISS_PATH)
+                print(f"Created directory {DB_FAISS_PATH}")
+            except Exception as e:
+                print(f"Error creating directory: {e}")
+
         # Create and save FAISS database
         db = FAISS.from_documents(texts, embeddings)
-        db.save_local(DB_FAISS_PATH)
+        try:
+            db.save_local(DB_FAISS_PATH)
+            print(f"FAISS database saved to {DB_FAISS_PATH}")
+        except Exception as e:
+            print(f"Error saving FAISS database: {e}")
+
         return True
 
 def main():
@@ -63,4 +76,4 @@ def main():
                 st.error(f"An error occurred: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()