create_csv_agent is expecting the Path to a csv file, so you need to provide it a path, rather than an uploaded object itself. The easiest way to do this is by creating a NamedTemporaryFile, writing the uploaded content to that file, and pass that fileβs path to create_csv_agent:
from tempfile import NamedTemporaryFile
import streamlit as st
from dotenv import load_dotenv
from langchain.agents import create_csv_agent
from langchain.llms import OpenAI
def main():
load_dotenv()
st.set_page_config(page_title="Ask your CSV")
st.header("Ask your CSV")
file = st.file_uploader("upload file", type="csv")
if file:
with NamedTemporaryFile() as f: # Create temporary file
f.write(file.getvalue()) # Save uploaded contents to file
llm = OpenAI(temperature=0)
user_input = st.text_input("Question here:")
agent = create_csv_agent(llm, f.name, verbose=True) # Pass temporary filename to create_csv_agent
if user_input:
response = agent.run(user_input)
st.write(response)
if __name__ == "__main__":
main()
Hi @blackary,
One thing β You might need to flush() after writing to the file.
This is working for me!
if user_csv is not None:
with NamedTemporaryFile(mode='w+b', suffix=".csv") as f:
f.write(user_csv.getvalue())
f.flush()
llm = OpenAI(temperature=0)
user_input = st.text_input("Question here:")
agent = create_csv_agent(llm, f.name, verbose=True)
if user_input:
response = agent.run(user_input)
st.write(response)
Thanks for stopping by! We use cookies to help us understand how you interact with our website.
By clicking βAccept allβ, you consent to our use of cookies. For more information, please see our privacy policy.
Cookie settings
Strictly necessary cookies
These cookies are necessary for the website to function and cannot be switched off. They are usually only set in response to actions made by you which amount to a request for services, such as setting your privacy preferences, logging in or filling in forms.
Performance cookies
These cookies allow us to count visits and traffic sources so we can measure and improve the performance of our site. They help us understand how visitors move around the site and which pages are most frequently visited.
Functional cookies
These cookies are used to record your choices and settings, maintain your preferences over time and recognize you when you return to our website. These cookies help us to personalize our content for you and remember your preferences.
Targeting cookies
These cookies may be deployed to our site by our advertising partners to build a profile of your interest and provide you with content that is relevant to you, including showing you relevant ads on other websites.