Update app.py

app.py

@@ -104,28 +104,28 @@ def gen(input):
             out = remove_caseifer(generated_text)
             return input + out
             
-
-def chatbot_func(user_message, history):
-    history = history + [[user_message, None]]
-    bot_message = gen(user_message)
-
-    for character in bot_message:
-        history[-1][1] += character
-        time.sleep(0.05)
-
-    return history
-
-def clear_chatbot(clear, history):
-    return [] if clear else history
-
+            
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(callback=chatbot_func, default=[[None, "Hi! I'm your ChatGPT bot. How can I assist you today?"]])
-    clear = gr.Button("Clear", label="Clear Chat")
-    
-    clear.set_observer(clear_chatbot, [clear, chatbot])
-
-demo.add(chatbot, clear)
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    def user(user_message, history):
+        return gr.update(value="", interactive=False), history + [[user_message, None]]
+
+    def bot(history):
+        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+        history[-1][1] = ""
+        for character in bot_message:
+            history[-1][1] += character
+            time.sleep(0.05)
+            yield history
+
+    response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+demo.queue()
 demo.launch()
-
-
-