Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import pandas as pd
 import streamlit as st
 from dotenv import load_dotenv
+import tempfile
 
 def main():
     # Load environment variables from .env file (if you're using one)
@@ -18,7 +19,7 @@ def main():
         st.error("OpenAI API Key is not set in the environment. Please check your environment variables.")
         return
     else:
-        st.success("")
+        st.success("OpenAI API Key loaded successfully!")
 
     # Upload the CSV file
     csv_file = st.file_uploader("Upload a CSV file", type="csv")
@@ -26,7 +27,7 @@ def main():
     # If a file is uploaded
     if csv_file is not None:
         try:
-            # Read the CSV without saving to disk
+            # Read the CSV file to show a preview
             df = pd.read_csv(csv_file, encoding='ISO-8859-1')
             st.write("Preview of the CSV file:")
             st.dataframe(df.head())  # Display the first few rows of the DataFrame
@@ -34,10 +35,15 @@ def main():
             st.error(f"Error reading the CSV file: {str(e)}")
             return
 
+        # Save the uploaded file to a temporary location
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmp_file:
+            tmp_file.write(csv_file.getbuffer())
+            tmp_file_path = tmp_file.name
+
         # Create the agent
         agent = create_csv_agent(
             ChatOpenAI(temperature=0, model="gpt-4-turbo"),
-            csv_file,
+            tmp_file_path,  # Pass the path of the saved CSV file
             verbose=True,
             agent_type=AgentType.OPENAI_FUNCTIONS,
         )