| import streamlit as st |
|
|
| from langchain.agents import create_pandas_dataframe_agent |
| from langchain.chat_models import ChatOpenAI |
| from langchain.agents.agent_types import AgentType |
| from langchain.chat_models.azure_openai import AzureChatOpenAI |
|
|
| import pandas as pd |
|
|
| form = st.sidebar.form(key='my_form') |
| user_api_key = form.text_input( |
| label="#### Your OpenAI API key 👇", |
| placeholder="Paste your openAI API key, sk-", |
| type="password") |
| uploaded_file = form.file_uploader("Choose a file") |
| text_prompt = form.text_area('Text Prompt', value='Hello, how are you?') |
| submit_button = form.form_submit_button(label='Submit') |
|
|
| if submit_button : |
| df = pd.read_csv(uploaded_file) |
|
|
| st.title("OpenAI Chatbot 🤖" |
| ) |
| st.subheader("This is a simple chatbot that uses OpenAI's GPT-3 model to generate responses to your text prompt. ") |
| |
| st.subheader("Your data:") |
| st.write(df) |
| |
| |
| st.markdown("""---""") |
|
|
| st.subheader("AI generated response:") |
|
|
| chat_model = AzureChatOpenAI( |
| deployment_name="chatgpt-4-32k", |
| openai_api_version="2023-03-15-preview", |
| openai_api_type="azure", |
| openai_api_base="https://api.hku.hk", |
| openai_api_key="", |
| ) |
| |
| agent = create_pandas_dataframe_agent( |
| llm=chat_model, |
| df=df, |
| verbose=True, |
| max_tokens_limit=2048, |
| handle_parsing_errors="Check your output and make sure it conforms!", |
| ) |
| |
| result = agent.run(text_prompt) |
| st.write(result) |
| |
| |
| |
