Update app.py

app.py

@@ -262,7 +262,7 @@ def add_message(history, message):
         
         print("### Not bigger than one history", history)
         print("### Not bigger than one conv", our_chatbot.conversation)
-        return history, gr.MultimodalTextbox(value=None, interactive=False)
+        return history, gr.MultimodalTextbox(value=None, interactive=False), our_chatbot
     else:
         for x in message["files"]:
             history.append(((x,), None))
@@ -271,7 +271,7 @@ def add_message(history, message):
 
         print("### Bigger than one history", history)
         print("### Bigger than one conv", our_chatbot.conversation)
-        return history, gr.MultimodalTextbox(value=None, interactive=False)
+        return history, gr.MultimodalTextbox(value=None, interactive=False), our_chatbot
     
 
 @spaces.GPU
@@ -549,10 +549,13 @@ with gr.Blocks(
         gr.Markdown(bibtext)
 
     # chat_msg = 
+    # chat_input.submit(
+    #     add_message, [chatbot, chat_input], [chatbot, chat_input]
+    # ).then(bot, [chatbot,temperature, top_p, max_output_tokens], chatbot, api_name="bot_response").then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
     chat_input.submit(
-        add_message, [chatbot, chat_input], [chatbot, chat_input]
-    ).then(bot, [chatbot,temperature, top_p, max_output_tokens], chatbot, api_name="bot_response").then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
-
+    add_message, [chatbot, chat_input], [chatbot, chat_input, 'state']
+    ).then(bot, ['state', chatbot, temperature, top_p, max_output_tokens], chatbot).then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
+    
     # chatbot.like(print_like_dislike, None, None)
     clear_btn.click(
         fn=clear_history, inputs=[chatbot], outputs=[chatbot], api_name="clear_all"