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

load_dotenv()

# get the API key from the environment
api_key = os.getenv("OPENAI_API_KEY")
st.write(api_key)

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")

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

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