Update app.py

app.py

@@ -41,6 +41,14 @@ with gr.Blocks(css = "custom.css") as demo:
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
+    def bot_api(history):
+        history[-1][1] = ""
+        user_message = history[-1][0]
+        pred_iter = predict(user_message ,history)
+        for ele in pred_iter:
+            history[-1][1] = ele
+        return history
+
     def bot(history):
         '''
         bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
@@ -81,7 +89,7 @@ with gr.Blocks(css = "custom.css") as demo:
     )
 
     api_button.click(
-        bot, chatbot, chatbot
+        bot_api, chatbot, chatbot
     )
 
     clear.click(lambda: None, None, chatbot, queue=False)