Update app.py

app.py

@@ -243,8 +243,8 @@ examples = [
 ]
 ex_btns = []
 
-def reset_chat(idx, ld, state):
-    res = [state["ppmanager_type"].from_json(json.dumps(ppm_str)) for ppm_str in ld]
+def reset_chat(idx, ld):
+    res = [GradioAlpacaChatPPManager.from_json(json.dumps(ppm)) for ppm in ld]
     res[idx].pingpongs = []
         
     return (
@@ -269,19 +269,14 @@ That said, the assistant is practical and really does its best, and doesn’t le
     return CtxLastWindowStrategy(3)(dummy_ppm)
 
 def add_pingpong(idx, ld, ping):
-    res = [
-        GradioAlpacaChatPPManager.from_json(json.dumps(ppm))
-        for ppm in ld
-    ]
-
+    res = [GradioAlpacaChatPPManager.from_json(json.dumps(ppm)) for ppm in ld]
     ppm = res[idx]
+    
     ppm.add_pingpong(
         PingPong(ping, "")
     )
-    prompt = build_prompts(ppm)
-    print(prompt)
-
-    prompt = tf.constant(prompt)
+    
+    prompt = tf.constant(build_prompts(ppm))
     max_length = tf.constant(256, dtype="int64")
     
     result = gpt_lm_predict_fn(
@@ -397,7 +392,7 @@ with gr.Blocks(css=STYLE, elem_id='container-col') as demo:
 
     clean.click(
         reset_chat,
-        [idx, local_data, chat_state],
+        [idx, local_data],
         [instruction_txtbox, chatbot, local_data, example_block, regenerate]
     ).then(
         None, local_data, None,