Update app.py

app.py

@@ -362,46 +362,29 @@ def main():
     loaded_documents = []
 
     if uploaded_files:
-        if not uploaded_files:
-            st.warning("Please upload some documents.")
         # Create a temporary directory
-        # with tempfile.TemporaryDirectory() as td:
-        #     # Move the uploaded files to the temporary directory and process them
-        #     for uploaded_file in uploaded_files:
-        #         st.write(f"Uploaded: {uploaded_file.name}")
-        #         ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
-        #         st.write(f"Uploaded: {ext}")
-        for uploaded_file in uploaded_files:
-            # Check if the extension is in FILE_LOADER_MAPPING
-            ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
-            if ext in FILE_LOADER_MAPPING:
-                loader_class, loader_args = FILE_LOADER_MAPPING[ext]
-                st.write(f"loader_class: {loader_class}")
-                # Save the uploaded file to the temporary directory
-                # file_path = os.path.join(td, uploaded_file.name)
-                with open(os.path.join(td, uploaded_file), 'wb') as temp_file:
-                    temp_file.write(uploaded_file.read())
-                     # with open(file_path, 'wb') as temp_file:
-                         # temp_file.write(uploaded_file.read())
-
-                # Use Langchain loader to process the file
-                loader = loader_class(os.path.join(td, file), **loader_args)
-                loaded_documents.extend(loader.load())
-                # # Check if the extension is in FILE_LOADER_MAPPING
-                # if ext in FILE_LOADER_MAPPING:
-                #     loader_class, loader_args = FILE_LOADER_MAPPING[ext]
-                #     st.write(f"loader_class: {loader_class}")
-                #     # Save the uploaded file to the temporary directory
-                #     # file_path = os.path.join(td, uploaded_file.name)
-                #     with open(os.path.join(td, uploaded_file), 'wb') as temp_file:
-                #         temp_file.write(uploaded_file.read())
-                #         # with open(file_path, 'wb') as temp_file:
-                #             # temp_file.write(uploaded_file.read())
-
-                #     # Use Langchain loader to process the file
-                #     loader = loader_class(os.path.join(td, file), **loader_args)
-                #     loaded_documents.extend(loader.load())
-                #     st.write(f"loaded_documents: {loaded_documents}")
+        with tempfile.TemporaryDirectory() as td:
+            # Move the uploaded files to the temporary directory and process them
+            for uploaded_file in uploaded_files:
+                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]
+
+                    # 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())
+
+                    # Use Langchain loader to process the file
+                    loader = loader_class(file_path, **loader_args)
+                    loaded_documents.extend(loader.load())
+                else:
+                    st.warning(f"Unsupported file extension: {ext}")
+
         st.write(f"loaded_documents: {loaded_documents}")  
         st.write("Chat with the Document:")
         query = st.text_input("Ask a question:")