Update app.py

app.py

@@ -1,68 +1,62 @@
-
-
 from langchain.agents import create_csv_agent
 from langchain.chat_models import ChatOpenAI
 from langchain.agents.agent_types import AgentType
-#from dotenv import load_dotenv
 import os
 import pandas as pd
 import streamlit as st
-
+from dotenv import load_dotenv
 
 def main():
-    # set the page title
-    
-    # text input to ask for openai api key
-    # then hide the input
-    openai_api_key = st.text_input("Enter your  key", type="password")
+    # Load environment variables from .env file (if you're using one)
+    load_dotenv()
 
-    # set this key as an environment variable
-    os.environ["OPENAI_API_KEY"] = openai_api_key
+    # Load the OpenAI API key from the environment
+    openai_api_key = os.getenv("OPENAI_API_KEY")
 
-    # load the api key from the .env file
-    #load_dotenv()
-    
-    # inform the user that the api key is loaded
-    if os.getenv("OPENAI_API_KEY") is None or os.getenv("OPENAI_API_KEY") == "":
-        print("OPENAI_API_KEY is not set")
-        exit(1)
+    # Check if the OpenAI API key is set
+    if openai_api_key is None or openai_api_key == "":
+        st.error("OpenAI API Key is not set in the environment. Please check your environment variables.")
+        return
     else:
-        print("OPENAI_API_KEY is set")
-        
-    # inform user that the model being used is the turbo model
-    st.write("Using the key")
-
+        st.success("OpenAI API Key is set from the backend.")
 
     # Upload the CSV file
     csv_file = st.file_uploader("Upload a CSV file", type="csv")
 
-    # if the user has uploaded a csv file then save it to the current directory
+    # If a file is uploaded
     if csv_file is not None:
-        with open(os.path.join(os.getcwd(), csv_file.name), "wb") as f:
-            f.write(csv_file.getbuffer())
-        st.write("CSV file uploaded to: ", os.path.join(os.getcwd(), csv_file.name))
-
-        # see a preview of the csv file
-        st.write("Preview of the CSV file:")
-         # see a preview of the csv file
-        df = pd.read_csv(os.path.join(os.getcwd(), csv_file.name))
-        st.dataframe(df.head())  # Display the first few rows of the DataFrame
-
-        # create the agent
+        try:
+            # Read the CSV without saving to disk
+            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
+        except Exception as e:
+            st.error(f"Error reading the CSV file: {str(e)}")
+            return
+
+        # Create the agent
         agent = create_csv_agent(
             ChatOpenAI(temperature=0, model="gpt-4-turbo"),
-            os.path.join(os.getcwd(), csv_file.name),
+            csv_file,
             verbose=True,
             agent_type=AgentType.OPENAI_FUNCTIONS,
         )
 
-        # ask the user for a question
-        user_question = st.text_input("Ask a question: ")
+        # Ask the user for a question
+        user_question = st.text_input("Ask a question:")
 
-        # if the user has asked a question then run the agent
-        if user_question is not None and user_question != "":
+        # If a question is entered, process it with the agent
+        if user_question and user_question.strip():
             with st.spinner(text="In progress..."):
-                st.write(agent.run(user_question))
+                try:
+                    response = agent.run(user_question)
+                    st.write(response)
+                except Exception as e:
+                    st.error(f"Error while processing your request: {str(e)}")
+        else:
+            st.warning("Please enter a valid question.")
 
 if __name__ == "__main__":
     main()
+
+