up

app.py

@@ -1,30 +1,16 @@
 import streamlit as st
-from langchain_experimental.agents import create_pandas_dataframe_agent
-import pandas as pd
-from langchain_openai import OpenAI
-import os
-from langchain_openai import ChatOpenAI
-
-
-
+from utils import query_agent
 
 st.title("Let's do some analysis on your CSV")
 st.header("Please upload your CSV file here:")
 
 # Capture the CSV file
 data = st.file_uploader("Upload CSV file",type="csv")
-df = pd.read_csv(data)
-llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
-agent_executor = create_pandas_dataframe_agent(
-    llm,
-    df,
-    agent_type="tool-calling",
-    verbose=True
-)
+
 query = st.text_area("Enter your query")
 button = st.button("Generate Response")
 
 if button:
     # Get Response
-    answer =  agent_executor.invoke(data,query)
+    answer =  query_agent(data,query)
     st.write(answer)