Langchain Pandas Dataframe Agent

pandas.py

"""
Streamlit application for querying invoice data using LangChain's pandas DataFrame agent.
Provides a simple interface for natural language queries on invoice data.
"""

import os
import pandas as pd
import streamlit as st
from langchain_experimental.agents import create_pandas_dataframe_agent
from langchain_openai import OpenAI
from dotenv import load_dotenv

load_dotenv()

os.getenv("OPENAI_API_KEY")
llm = OpenAI()

st.title("Querying Incoices")

# data = st.file_uploader("Upload CSV file", type="csv")

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


def get_response(dataframe):
    """
    Create and query a pandas dataframe agent.
    
    Args:
        dataframe: pandas DataFrame to query
        
    Returns:
        dict: Agent's response to the query
    """
    agent = create_pandas_dataframe_agent(llm, dataframe, verbose=True)
    return agent.invoke(query)


if button:
    df = pd.read_csv("data/invoices.csv")
    response = get_response(df)
    st.write(response["output"])