Update app.py

app.py

@@ -14,6 +14,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")
@@ -35,6 +36,11 @@ if uploaded_file:
         st.write("Preview of uploaded data:")
         st.dataframe(data)
 
+        # Save the uploaded file to a temporary location
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w") as temp_file:
+            temp_file_path = temp_file.name
+            data.to_csv(temp_file_path, index=False)
+
         # Tabs for LangChain and LlamaIndex
         tab1, tab2 = st.tabs(["LangChain", "LlamaIndex"])
 
@@ -42,8 +48,8 @@ if uploaded_file:
         with tab1:
             st.subheader("LangChain Query")
             try:
-                # Use CSVLoader with the uploaded DataFrame
-                loader = CSVLoader(data=data)
+                # Use CSVLoader with the temporary file path
+                loader = CSVLoader(file_path=temp_file_path)
                 docs = loader.load_and_split()
 
                 # Preview the first document chunk
@@ -86,9 +92,9 @@ if uploaded_file:
         with tab2:
             st.subheader("LlamaIndex Query")
             try:
-                # Use PagedCSVReader directly on the uploaded DataFrame
+                # Use PagedCSVReader directly on the uploaded file
                 csv_reader = PagedCSVReader()
-                docs = csv_reader.load_from_dataframe(data)
+                docs = csv_reader.load_from_file(temp_file_path)
 
                 # Preview the first document chunk
                 if docs:
@@ -114,6 +120,10 @@ 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}")