Update app.py

app.py

@@ -1,32 +1,42 @@
+
 import streamlit as st
-from langchain.llms import OpenAI
+from langchain.chat_models import ChatOpenAI
+from langchain.schema import SystemMessage, HumanMessage, AIMessage
+
+# From here down is all the StreamLit UI.
+st.set_page_config(page_title="Entz's LLM LangChain-OpenAI", page_icon=":ant:")
+st.header("Role Play: A 5-Years Old Cute Girl")
+
+# put a presumptions for ai to the streamlit session state
+# st.session_state provides a way to store and persist data between reruns,
+# effectively allowing the app to remember information like user inputs, selections, variables
+if "presumptions" not in st.session_state:
+     st.session_state.presumptions = [
+        SystemMessage(content="You are a 5 years old girl, who can only speak simple words, and is a huge fan of Barbie and toy kitchen sets.")
+    ]
 
-#Function to return the response
 def load_answer(question):
-    llm = OpenAI(model_name="gpt-3.5-turbo",temperature=0)
-    answer=llm(question)
-    return answer
 
+    st.session_state.presumptions.append(HumanMessage(content=question))
+    assistant_answer  = chat(st.session_state.presumptions )
+    # store the new answer the presumption list
+    st.session_state.presumptions.append(AIMessage(content=assistant_answer.content))
+    return assistant_answer.content
 
-#App UI starts here
-st.set_page_config(page_title="Entz's LLM LangChain-gpt-3.5-turbo", page_icon=":ant:")
-st.header("Entz's LLM LangChain Base Model")
 
-#Gets the user input
-def question():
-    input_text = st.text_input("Enter your question: ", key="input")
+def get_text():
+    input_text = st.text_input("Ask me question please~ : ", key= input)
     return input_text
 
+chat = ChatOpenAI(temperature=0)
 
-user_input = question()
-response = load_answer(user_input)
+user_input=get_text()
+submit = st.button('Little girl answers: ')  
 
-submit = st.button('Entz AI Answer')  
-
-#If generate button is clicked
 if submit:
+    
+    response = load_answer(user_input)
+    st.subheader("Answer:")
 
-    st.subheader("According to Entz AI:")
-
-    st.write(response)
+    st.write(response,key= 1)