Update rag.py

rag.py

@@ -1,5 +1,5 @@
 import os
-import shutil
+import shutil  # <-- Added this missing import
 from llama_index.core import (
     VectorStoreIndex,
     SimpleDirectoryReader,
@@ -60,6 +60,11 @@ def add_document(file_path: str):
     
     # Move the uploaded file to the persistent data folder
     dest = os.path.join(DATA_DIR, os.path.basename(file_path))
+    
+    # Ensure the destination doesn't exist or handle it (shutil.move requires this)
+    if os.path.exists(dest):
+        os.remove(dest)
+        
     shutil.move(file_path, dest)
     
     # Reload all documents and rebuild index