import streamlit as st
from dotenv import load_dotenv
from utils import query_agent

load_dotenv()

# Direct CSV file 
data = pd.read_csv('Tickets_cleaned_v2.csv')

st.title("Let's do some analysis on your CSV")
st.header("Please upload your CSV file here:")



query = st.text_area("Enter your query")
button = st.button("Generate Response")

if button:
    # Get Response
    answer =  query_agent(data,query)
    st.write(answer)