Update app.py

app.py

@@ -361,28 +361,44 @@ def main():
 
     if uploaded_files:
         # 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}")
-
-                # 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}")
+        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())
+                st.write(f"loaded_documents: {loaded_documents}")
+                # # 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}")
 
         st.write("Chat with the Document:")
         query = st.text_input("Ask a question:")