Update app.py

app.py

@@ -41,13 +41,30 @@ with gr.Blocks(css = "custom.css") as demo:
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
-    def bot_api(history):
+    def bot_api(user_message):
+        l = user_message.split("[SEP]")
+        history = []
+        for ele in history:
+            if not history:
+                history.append(
+                    [ele]
+                )
+            else:
+                if len(history[-1]) == 2:
+                    history.append([ele])
+                else:
+                    history[-1].append(ele)
         history[-1][1] = ""
-        user_message = history[-1][0]
-        pred_iter = predict(user_message ,history)
+        user_message_ = history[-1][0]
+        pred_iter = predict(user_message_ ,history)
         for ele in pred_iter:
             history[-1][1] = ele
-        return history
+        user_message_out = []
+        for ele in history:
+            for e in ele:
+                user_message_out.append(e)
+        user_message_out = "[SEP]".join(user_message_out)
+        return user_message_out
 
     def bot(history):
         '''
@@ -80,6 +97,7 @@ with gr.Blocks(css = "custom.css") as demo:
             interactive=True,
             #info = "Generate by Click, and can edit by yourself, look up Examples below"
         )
+        msg_out = gr.Textbox(visible = False)
         sub_button = gr.Button("Submit")
     clear = gr.Button("Clear")
     api_button = gr.Button("api", visible = False)
@@ -89,7 +107,7 @@ with gr.Blocks(css = "custom.css") as demo:
     )
 
     api_button.click(
-        bot_api, chatbot, chatbot
+        bot_api, msg, msg_out
     )
 
     clear.click(lambda: None, None, chatbot, queue=False)