Update src/rag_engine.py

src/rag_engine.py

@@ -146,4 +146,32 @@ def list_documents(username: str = "default") -> List[str]:
         if f.lower().endswith(('.pdf', '.txt', '.md')):
             files.append(f)
             
-    return files
+    return files
+
+def save_uploaded_file(uploaded_file, username: str = "default") -> str:
+    """
+    Saves a StreamlitUploadedFile to a temporary location on disk.
+    Returns the absolute path to the saved file.
+    """
+    try:
+        # Define the directory where files will be stored
+        # You can customize "source_documents" to match your preferred structure
+        base_dir = "source_documents"
+        user_dir = os.path.join(base_dir, username)
+        
+        # Create the directory if it doesn't exist
+        os.makedirs(user_dir, exist_ok=True)
+        
+        # Create the full file path
+        file_path = os.path.join(user_dir, uploaded_file.name)
+        
+        # Write the file content
+        with open(file_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+            
+        logger.info(f"File saved successfully at: {file_path}")
+        return file_path
+        
+    except Exception as e:
+        logger.error(f"Error saving uploaded file: {e}")
+        return None