Update app.py

app.py

@@ -354,23 +354,40 @@ def retrieve_bot_answer(query):
 def main():
     st.title("Docuverse")
 
+    # # Upload files
+    # uploaded_files = st.file_uploader("Upload your documents", type=["pdf", "md", "txt", "csv", "py", "epub", "html", "ppt", "pptx", "doc", "docx", "odt", "ipynb"], accept_multiple_files=True)
+    # loaded_documents = []
+    # if uploaded_files:
+    #     # Process uploaded files
+    #     for uploaded_file in uploaded_files:
+    #         st.write(f"Uploaded: {uploaded_file}")
+    #         st.write(f"Uploaded: {uploaded_file.name}")
+    #         st.write(f"Uploaded: {type(uploaded_file.name)}")
+    #         ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
+    #         st.write(f"Uploaded: {ext}")
+    #         if ext in FILE_LOADER_MAPPING:
+    #             loader_class, loader_args = FILE_LOADER_MAPPING[ext]
+    #             loader = loader_class(uploaded_file, **loader_args)
+    #         else:
+    #             loader = UnstructuredFileLoader(uploaded_file)
+    #         loaded_documents.extend(loader.load())
     # Upload files
     uploaded_files = st.file_uploader("Upload your documents", type=["pdf", "md", "txt", "csv", "py", "epub", "html", "ppt", "pptx", "doc", "docx", "odt", "ipynb"], accept_multiple_files=True)
     loaded_documents = []
     if uploaded_files:
-        # Process uploaded files
-        for uploaded_file in uploaded_files:
-            st.write(f"Uploaded: {uploaded_file}")
-            st.write(f"Uploaded: {uploaded_file.name}")
-            st.write(f"Uploaded: {type(uploaded_file.name)}")
-            ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
-            st.write(f"Uploaded: {ext}")
-            if ext in FILE_LOADER_MAPPING:
-                loader_class, loader_args = FILE_LOADER_MAPPING[ext]
-                loader = loader_class(uploaded_file, **loader_args)
-            else:
-                loader = UnstructuredFileLoader(uploaded_file)
-            loaded_documents.extend(loader.load())
+        # Create a temporary directory
+        with tempfile.TemporaryDirectory() as td:
+            # Move the uploaded files to the temporary directory
+            for uploaded_file in uploaded_files:
+                uploaded_file.save(os.path.join(td, uploaded_file.name))
+
+            # Process the files in the temporary directory
+            for file in os.listdir(td):
+                ext = os.path.splitext(file)[-1][1:].lower()
+                if ext in FILE_LOADER_MAPPING:
+                    loader_class, loader_args = FILE_LOADER_MAPPING[ext]
+                    loader = loader_class(os.path.join(td, file), **loader_args)
+                    loaded_documents.extend(loader.load())
 
         st.write("Chat with the Document:")
         query = st.text_input("Ask a question:")