Update app.py

app.py

@@ -243,6 +243,18 @@ examples = [
 ]
 ex_btns = []
 
+def reset_chat(idx, ld, state):
+    res = [state["ppmanager_type"].from_json(json.dumps(ppm_str)) for ppm_str in ld]
+    res[idx].pingpongs = []
+        
+    return (
+        "",
+        [],
+        str(res),
+        gr.update(visible=True),
+        gr.update(interactive=False),
+    )
+
 def build_prompts(ppmanager):
     dummy_ppm = copy.deepcopy(ppmanager)
     
@@ -343,8 +355,8 @@ with gr.Blocks(css=STYLE, elem_id='container-col') as demo:
 
             with gr.Column(elem_id="aux-btns-popup", visible=True):
                 with gr.Row():
-                    stop = gr.Button("Stop", elem_classes=["aux-btn"])
-                    regenerate = gr.Button("Regenerate", elem_classes=["aux-btn"])
+                    # stop = gr.Button("Stop", elem_classes=["aux-btn"])
+                    # regenerate = gr.Button("Regenerate", elem_classes=["aux-btn"])
                     clean = gr.Button("Clean", elem_classes=["aux-btn"])
 
             chatbot = gr.Chatbot(elem_id='chatbot')
@@ -383,6 +395,15 @@ with gr.Blocks(css=STYLE, elem_id='container-col') as demo:
         _js="(v)=>{ setStorage('local_data',v) }"
     )
 
+    clean.click(
+        reset_chat,
+        [idx, local_data, chat_state],
+        [instruction_txtbox, chatbot, local_data, example_block, regenerate]
+    ).then(
+        None, local_data, None, 
+        _js="(v)=>{ setStorage('local_data',v) }"
+    )
+    
     demo.load(
         None,
         inputs=None,