from langchain.agents import create_pandas_dataframe_agent
#from langchain_experimental.agents.agent_toolkits.csv.base import create_csv_agent
from langchain.llms import OpenAI
from langchain.chat_models import ChatOpenAI

import pandas as pd

# Parse the CSV file and create a Pandas DataFrame from its contents.
df = pd.read_csv("Tickets_cleaned_v2.csv")

def query_agent(query):
   
    llm = ChatOpenAI(
        model_name="gpt-4-32k",
        temperature=0.2,
    )
    
    
    # Create a Pandas DataFrame agent.
    agent = create_pandas_dataframe_agent(llm, df, verbose=True)

    #Python REPL: A Python shell used to evaluating and executing Python commands. 
    #It takes python code as input and outputs the result. The input python code can be generated from another tool in the LangChain
    return agent.run(query)