|
from tempfile import NamedTemporaryFile |
|
from langchain.agents import create_csv_agent |
|
from langchain.llms import OpenAI |
|
from dotenv import load_dotenv |
|
import os |
|
import streamlit as st |
|
|
|
|
|
|
|
def main(): |
|
load_dotenv() |
|
|
|
|
|
api_key = os.getenv("OPENAI_API_KEY") |
|
if api_key is None or api_key == "": |
|
st.error("OPENAI_API_KEY is not set") |
|
return |
|
|
|
st.set_page_config(page_title="Insightly") |
|
st.sidebar.image("/home/oem/Downloads/insightly_wbg.png", use_column_width=True) |
|
st.header("Data Analysis 📈") |
|
|
|
csv_file = st.file_uploader("Upload a CSV file", type="csv") |
|
if csv_file: |
|
with NamedTemporaryFile(delete=False) as f: |
|
f.write(csv_file.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) |
|
|
|
|
|
if __name__ == "__main__": |
|
main() |
