Update app.py

app.py

@@ -47,8 +47,9 @@ if uploaded_file:
             st.subheader("LangChain Query")
             try:
                 # Save the uploaded file to a temporary file for LangChain
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as temp_file:
-                    temp_file.write(uploaded_file.getvalue())
+                with tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w") as temp_file:
+                    # Write the DataFrame to the temp file
+                    data.to_csv(temp_file.name, index=False)
                     temp_file_path = temp_file.name
 
                 # Use CSVLoader with the temporary file path
@@ -100,10 +101,15 @@ if uploaded_file:
         with tab2:
             st.subheader("LlamaIndex Query")
             try:
+                # Save uploaded file content to a temporary CSV file for LlamaIndex
+                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
+
                 # 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()
@@ -132,5 +138,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}")