Update app.py

app.py

@@ -4,18 +4,28 @@
 import streamlit as st
 
 
-from langchain.llms import OpenAI
+from langchain.chat_models import ChatOpenAI
 
-#Function to return the response
-def load_answer(question):
-    llm = OpenAI(model_name="text-davinci-003",temperature=0)
-    answer=llm(question)
-    return answer
+from langchain.schema import (AIMessage,HumanMessage,SystemMessage)
 
 
 #App UI starts here
 st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
-st.header("LangChain Demo")
+st.header("GetRoastedGPT")
+
+if "sessionMessages" not in st.session_state:
+    st.session_state.sessionMessages = [
+        SystemMessage(content= "You are an assistant which roasts every input")
+    ]
+
+def load_answer(question):
+
+    st.session_state.sessionMessages.append(HumanMessage(content=question))
+
+    AIans= chat(st.session_state.sessionMessages)
+    st.session_state.sessionMessages.append(AIMessage(content=AIans.content))
+
+    return AIans.content
 
 #Gets the user input
 def get_text():
@@ -24,14 +34,11 @@ def get_text():
 
 
 user_input=get_text()
-response = load_answer(user_input)
-
-submit = st.button('Generate')  
-
+submit = st.button('Roast me')  
 #If generate button is clicked
 if submit:
-
+    response= load_answer(user_input)
     st.subheader("Answer:")
 
-    st.write(response)
+    st.write(response,key=1)