Update app.py

app.py

@@ -116,20 +116,21 @@ def load_model():
 
 def load_document(
     # file_path: str,
-    uploaded_files: str,
+    uploaded_files: list,
     mapping: dict = FILE_LOADER_MAPPING,
     default_loader: BaseLoader = UnstructuredFileLoader,
 ) -> Document:
-    # Choose loader from mapping, load default if no match found
-    # ext = "." + uploaded_files.rsplit(".", 1)[-1]
-    ext = os.path.splitext(uploaded_files.name)[-1][1:].lower()
-    if ext in mapping:
-        loader_class, loader_args = mapping[ext]
-        loader = loader_class(uploaded_files, **loader_args)
-    else:
-        loader = default_loader(uploaded_files)
     loaded_documents = []
-    loaded_documents.extend(loader.load())
+    for uploaded_file in uploaded_files:
+        # Choose loader from mapping, load default if no match found
+        # ext = "." + uploaded_files.rsplit(".", 1)[-1]
+        ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
+        if ext in mapping:
+            loader_class, loader_args = mapping[ext]
+            loader = loader_class(uploaded_file, **loader_args)
+        else:
+            loader = default_loader(uploaded_file)
+        loaded_documents.extend(loader.load())
     return loaded_documents
 
 def create_vector_database(loaded_documents):