app.py

import streamlit as st
from ai_agent import pandas_agent

# Get user input
file = st.file_uploader("Upload CSV file")
agent_input = st.text_input("Enter question")

if st.button('Execute'):
    if file is not None and agent_input != "":
        # Create a loading spinner
        with st.spinner('Running the agent...'):
            # Call the pandas_agent function
            output = pandas_agent(file, agent_input)

        # Display the output
        st.write(output)
    else:
        st.write("Please upload a file and enter a question.")