Update app.py

app.py

@@ -1,13 +1,13 @@
 import os
 import time
-import openai
+from openai import OpenAI
 import numpy as np
 import streamlit as st
 import tensorflow as tf
 import tensorflow_text
 # import plotly.graph_objects as go
 # from dotenv import load_dotenv
-from langchain.llms import OpenAI
+from langchain_openai import OpenAI as OpenAiLC
 from langchain.memory import ConversationSummaryMemory, ChatMessageHistory
 from llm import sys_instruction
 
@@ -72,13 +72,20 @@ openai.api_key = st.session_state.key
 
 # gpt llm
 if 'llm' not in st.session_state:
-    st.session_state.llm = OpenAI(
+    st.session_state.llm = OpenAiLC(
         temperature=0.2, openai_api_key=st.session_state.key)
 
 # model name
 if "openai_model" not in st.session_state:
     st.session_state["openai_model"] = "gpt-3.5-turbo-instruct"
 
+# openai client 
+# model name
+if "client" not in st.session_state:
+    st.session_state["client"]  = OpenAI(
+    api_key=st.session_state.key
+)
+    
 # st chat history
 if "message_history" not in st.session_state:
     st.session_state.message_history = []
@@ -190,7 +197,7 @@ if prompt := st.chat_input("What is up?"):
         full_response = ""
 
         # get response
-        for response in openai.ChatCompletion.create(
+        for chunk in st.session_state.client.chat.completions.create(
                 model=st.session_state["openai_model"],
                 messages=[
                     {"role": "system", "content": st.session_state.sys_inst.format(
@@ -200,8 +207,10 @@ if prompt := st.chat_input("What is up?"):
                 stream=True):
 
             # render gpt response in realtime
-            full_response += response.choices[0].delta.get("content", "")
-            message_placeholder.markdown(full_response + "▌")
+            if chunk.choices[0].delta.content:
+                # print(chunk.choices[0].delta.content)   
+                full_response += chunk.choices[0].delta.content
+                message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
 
     # add to st history