Update app.py

app.py

@@ -1,10 +1,12 @@
+import openai
+import streamlit as st
+from streamlit_chat import message
 from langchain_core.messages import SystemMessage
 from langchain_openai import ChatOpenAI
 from langchain_core.pydantic_v1 import BaseModel, Field
 from langgraph.graph import MessageGraph, END
 from langgraph.checkpoint.sqlite import SqliteSaver
 from langchain_core.messages import HumanMessage
-import streamlit as st
 from typing import List
 import os
 import uuid
@@ -22,6 +24,11 @@ If you're asking anything please be friendly and comment on any of the info you
  
 After you are able to discerne all the information, call the relevant tool"""
 
+
+OPENAI_API_KEY='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'
+os.environ['OPENAI_API_KEY'] = OPENAI_API_KEY
+
+
 llm = ChatOpenAI(temperature=0)
 
 def get_messages_info(messages):
@@ -88,45 +95,54 @@ graph = workflow.compile(checkpointer=memory)
 
 config = {"configurable": {"thread_id": str(uuid.uuid4())}}
 
-# Helper function to execute the LangChain logic
-def execute_langchain(user_input):
-    output_list = []
-    for output in graph.stream([HumanMessage(content=user_input)], config=config):
-        if "__end__" in output:
-            continue
-        for key, value in output.items():
-            output_list.append((key, value))
-    return output_list
-
 # Streamlit app layout
 st.title("LangChain Chat")
 
-conversation_history = []
+# Initialise session state variables
+if 'generated' not in st.session_state:
+    st.session_state['generated'] = []
+if 'past' not in st.session_state:
+    st.session_state['past'] = []
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = [
+        {"role": "system", "content": "You are a helpful assistant."}
+    ]
+ 
 
-message = st.chat_message("assistant")
-message.write("Good to see you. Please tell me your latest career")
+# reset everything
+if clear_button:
+    st.session_state['generated'] = []
+    st.session_state['past'] = []
+    st.session_state['messages'] = [
+        {"role": "system", "content": "You are a helpful assistant."}
+    ]
+ 
 
- # Display chat history
-for message in st.session_state["chat_history"]:
-    st.chat_message(message["message"], key=message["message"])
-    
-user_input = st.text_input("Enter your Response")
 
-if "chat_history" not in st.session_state:
-    st.session_state["chat_history"] = []
-    
-if st.button("Send"):
-    if user_input:
-        # Execute LangChain logic
-        outputs = execute_langchain(user_input)         
-        # Update chat history
-        st.session_state["chat_history"].append({"message": user_input, "role": "user"})
-        st.session_state["chat_history"].append({"message": outputs, "role": "bot"})
+# container for chat history
+response_container = st.container()
+# container for text box
+container = st.container()
 
+with container:
+    with st.form(key='my_form', clear_on_submit=True):
+        user_input = st.text_area("You:", key='input', height=100)
+        submit_button = st.form_submit_button(label='Send')
 
+    if submit_button and user_input:
+            st.session_state['messages'].append({"role": "user", "content": user_input})
     
-# Allow the user to quit the chat
-if st.button("Quit"):
-    st.text("Bot: Byebye")
-
-    
+            for output in graph.stream([HumanMessage(content=st.session_state['messages'])], config=config):
+              if "__end__" in output:
+                  continue
+              # stream() yields dictionaries with output keyed by node name
+              for key, value in output.items():
+                  st.session_state['messages'].append({"role": "assistant", "content": value})
+                  st.session_state['past'].append(user_input)
+                  st.session_state['generated'].append(value)
+
+if st.session_state['generated']:
+    with response_container:
+        for i in range(len(st.session_state['generated'])):
+            message(st.session_state["past"][i], is_user=True, key=str(i) + '_user')
+            message(st.session_state["generated"][i], key=str(i))