Update app.py

app.py

@@ -114,6 +114,8 @@ 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
@@ -137,7 +139,7 @@ async def rollback_last(
     ppm = res[idx]
     ppm.replace_last_pong(response_txt)
     
-    return "", ppm.build_uis(), str(res), gr.update(interactive=True), "off"
+    yield "", 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]