Update app.py

app.py

@@ -3,19 +3,9 @@ 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
-from langchain.llms import AzureOpenAI
-
-import os
-
-os.environ["OPENAI_API_TYPE"] = "azure"
-os.environ["OPENAI_API_VERSION"] = "2023-03-15-preview"
-os.environ["OPENAI_API_BASE"] = "https://api.hku.hk"
-os.environ["OPENAI_API_KEY"] = ""
-
-# Import Azure OpenAI
-from langchain.llms import AzureOpenAI
 
 form = st.sidebar.form(key='my_form')
 user_api_key = form.text_input(
@@ -42,16 +32,16 @@ if submit_button :
 
     st.subheader("AI generated response:")
 
-
-    llm = AzureOpenAI(
-        deployment_name="chatgpt",
-        model_name="chatgpt",
+    chat_model = AzureChatOpenAI(
+        deployment_name="chatgpt-4",
+        openai_api_version="2023-03-15-preview",
+        openai_api_type="azure",
+        openai_api_base="https://api.hku.hk",
         openai_api_key=user_api_key,
     )
-
     
     agent = create_pandas_dataframe_agent(
-        llm,
+        chat_model,
         df,
         verbose=True,
         agent_type=AgentType.OPENAI_FUNCTIONS,