Update app.py

app.py

@@ -121,20 +121,21 @@ 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:
-            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)
+while True:
+    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)
                   
 if st.session_state['generated']:
     with response_container: