Update app.py

app.py

@@ -15,6 +15,7 @@ from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_openai import OpenAIEmbeddings, ChatOpenAI
 import faiss
+import tempfile
 
 # Load environment variables
 os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
@@ -36,6 +37,11 @@ if uploaded_file:
         st.write("Preview of uploaded data:")
         st.dataframe(data)
 
+        # Create a temporary file for the uploaded CSV
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w") as temp_file:
+            data.to_csv(temp_file.name, index=False)
+            temp_file_path = temp_file.name
+
         # Tabs
         tab1, tab2 = st.tabs(["Chat w CSV using LangChain", "Chat w CSV using LlamaIndex"])
 
@@ -43,8 +49,8 @@ if uploaded_file:
         with tab1:
             st.subheader("LangChain Query")
             try:
-                # Use CSVLoader with the uploaded file
-                loader = CSVLoader(file_path=uploaded_file)
+                # Use CSVLoader with the temporary file path
+                loader = CSVLoader(file_path=temp_file_path)
                 docs = loader.load_and_split()
 
                 # Preview the first document
@@ -80,7 +86,6 @@ if uploaded_file:
                 if query:
                     answer = langchain_rag_chain.invoke({"input": query})
                     st.write(f"Answer: {answer['answer']}")
-
             except Exception as e:
                 st.error(f"Error processing with LangChain: {e}")
 
@@ -91,7 +96,7 @@ if uploaded_file:
                 # Use PagedCSVReader for LlamaIndex
                 csv_reader = PagedCSVReader()
                 reader = SimpleDirectoryReader(
-                    input_files=[uploaded_file.name],
+                    input_files=[temp_file_path],
                     file_extractor={".csv": csv_reader},
                 )
                 docs = reader.load_data()
@@ -120,5 +125,9 @@ if uploaded_file:
                     st.write(f"Answer: {response.response}")
             except Exception as e:
                 st.error(f"Error processing with LlamaIndex: {e}")
+            finally:
+                # Clean up the temporary file
+                if 'temp_file_path' in locals() and os.path.exists(temp_file_path):
+                    os.remove(temp_file_path)
     except Exception as e:
         st.error(f"Error reading uploaded file: {e}")