Update app.py

app.py

@@ -355,40 +355,30 @@ 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:
         # Create a temporary directory
         with tempfile.TemporaryDirectory() as td:
-            # Move the uploaded files to the temporary directory
+            # Move the uploaded files to the temporary directory and process them
             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()
+                st.write(f"Uploaded: {uploaded_file.name}")
+                ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
+                st.write(f"Uploaded: {ext}")
+
+                # Check if the extension is in FILE_LOADER_MAPPING
                 if ext in FILE_LOADER_MAPPING:
                     loader_class, loader_args = FILE_LOADER_MAPPING[ext]
-                    with open(os.path.join(td, file), 'wb') as temp_file:
+
+                    # Save the uploaded file to the temporary directory
+                    file_path = os.path.join(td, uploaded_file.name)
+                    with open(file_path, 'wb') as temp_file:
                         temp_file.write(uploaded_file.read())
-                    loader = loader_class(os.path.join(td, file), **loader_args)
+
+                    # Use Langchain loader to process the file
+                    loader = loader_class(file_path, **loader_args)
                     loaded_documents.extend(loader.load())
 
         st.write("Chat with the Document:")