Update app.py

app.py

@@ -147,18 +147,22 @@ with gr.Blocks(css=CSS) as demo:
         system_msg = gr.Textbox(
             start_message, label="System Message", interactive=True, visible=True, placeholder="System prompt. Provide instructions which you want the model to remember.", lines=5)
 
-    chat_history_state = gr.State()
-    clear.click(clear_chat, inputs=[chat_history_state, message], outputs=[chat_history_state, message], queue=False)
+    chat_history_state1 = gr.State()
+    chat_history_state2 = gr.State()
+    clear.click(clear_chat, inputs=[chat_history_state1, message], outputs=[chat_history_state1, message], queue=False)
+    clear.click(clear_chat, inputs=[chat_history_state2, message], outputs=[chat_history_state2, message], queue=False)
     clear.click(lambda: None, None, chatbot1, queue=False)
     clear.click(lambda: None, None, chatbot2, queue=False)
 
     submit_click_event1 = submit.click(
-        fn=user, inputs=[message, chat_history_state], outputs=[message, chat_history_state], queue=True
+        fn=user, inputs=[message, chat_history_state1], outputs=[message, chat_history_state1], queue=True
     ).then(
-        fn=chat1, inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot1, chat_history_state, message], queue=True
+        fn=chat1, inputs=[chat_history_state1, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot1, chat_history_state1, message], queue=True
     )
     submit_click_event2 = submit.click(
-        fn=chat2, inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot1, chat_history_state, message], queue=True
+        fn=user, inputs=[message, chat_history_state2], outputs=[message, chat_history_state2], queue=True
+    ).then(
+        fn=chat2, inputs=[chat_history_state2, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot1, chat_history_state2, message], queue=True
     )
     stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_click_event1, submit_click_event2], queue=False)