Update app.py

app.py

@@ -121,20 +121,24 @@ response_container = st.container()
 # container for text box
 container = st.container()
 
+@st.cache_resource
+def query(payload):
+    for output in graph.stream([HumanMessage(content=payload], 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.content})
+          st.session_state['past'].append(user_input)
+          st.session_state['generated'].append(value.content)
+                      
 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:
-                for output in graph.stream([HumanMessage(content=user_input)], 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.content})
-                      st.session_state['past'].append(user_input)
-                      st.session_state['generated'].append(value.content)
+            query(user_input)
                   
 if st.session_state['generated']:
     with response_container: