Update app.py

app.py

@@ -114,8 +114,6 @@ async def rollback_last(
     prompt = build_prompts(ppm, global_context, ctx_num_lconv)
     print(prompt)
 
-    yield "", ppm.build_uis(), str(res), gr.update(interactive=True), "off"
-
     #######
     # if internet_option:
     #     search_prompt = None
@@ -139,7 +137,7 @@ async def rollback_last(
     ppm = res[idx]
     ppm.replace_last_pong(response_txt)
     
-    yield "", ppm.build_uis(), str(res), gr.update(interactive=True), "off"
+    return "", ppm.build_uis(), str(res), gr.update(interactive=True), "off"
 
 def reset_chat(idx, ld, state, palm_if):
     res = [state["ppmanager_type"].from_json(json.dumps(ppm_str)) for ppm_str in ld]
@@ -401,6 +399,10 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
     # )
 
     regen_event = regenerate.click(
+        lambda history: history[:-1]
+        [chat_state],
+        [chatbot]
+    ).then(
         rollback_last,
         [idx, local_data, chat_state,
          global_context, res_temp, res_topk, res_topp, ctx_num_lconv,