Working immediate input from chat

app.py

@@ -72,7 +72,6 @@ def cut_dialogue_history(history_memory, keep_last_n_words=400):
 
 class ConversationBot:
     def __init__(self, load_dict):
-        # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
         print(f"Initializing VisualChatGPT, load_dict={load_dict}")
         if 'ImageCaptioning' not in load_dict:
             raise ValueError("You have to load ImageCaptioning as a basic function for VisualChatGPT")
@@ -94,7 +93,9 @@ class ConversationBot:
         res = self.agent({"input": text})
         res['output'] = res['output'].replace("\\", "/")
         response = re.sub('(image/\S*png)', lambda m: f'![](/file={m.group(0)})*{m.group(0)}*', res['output'])
+
         state = state + [(text, response)]
+
         print(f"\nProcessed run_text, Input text: {text}\nCurrent state: {state}\n"
               f"Current Memory: {self.agent.memory.buffer}")
         return state, state
@@ -197,19 +198,21 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
         inputs=txt
     )
 
-    gr.HTML('''<br><br><br><center>You can duplicate this Space to skip the queue:
-                <a href="https://huggingface.co/spaces/microsoft/visual_chatgpt?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a><br>
-            </center>''')
-
-
-
     if not OPENAI_API_KEY:
         openai_api_key_textbox.submit(bot.init_agent, [openai_api_key_textbox], [input_raws])
     
+    def update_text(text, state):
+        chat = state + [(text, None)]
+        return chat
+
+    txt.submit(update_text, [txt, state], [chatbot])
     txt.submit(bot.run_text, [txt, state], [chatbot, state])
     txt.submit(lambda: "", None, txt)
+
+    run.click(update_text, [txt, state], [chatbot])
     run.click(bot.run_text, [txt, state], [chatbot, state])
     run.click(lambda: "", None, txt)
+
     btn.upload(bot.run_image, [btn, state, txt], [chatbot, state, txt])
     clear.click(bot.memory.clear)
     clear.click(lambda: [], None, chatbot)