Update app.py

app.py

@@ -43,9 +43,7 @@ def parse_codeblock(text):
             if i > 0:
                 lines[i] = "<br/>" + line.replace("<", "&lt;").replace(">", "&gt;")
     return "".join(lines)
-
-def reset_textbox():
-    return gr.update(value='', interactive=False), gr.update(interactive=False)
+=
 
 def random_sample():
     global SAMPLE_IDX
@@ -64,9 +62,12 @@ def random_sample():
     print(CURRENT_POSITION)
     return np.array(new_image)
 
-def reset_sample():
+def reset_sample(*agents):
     image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
-    return image
+    new_agents = []
+    for k in range(TOTAL_K):
+        new_agents.append('')
+    return image, *[gr.Textbox.update(value=answer) for answer in new_agents]
 
 
 title = """<h1 align="center">Courtroom with AI juries</h1>"""
@@ -164,7 +165,7 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
     accept_checkbox.change(fn=enable_inputs, inputs=[], outputs=[user_consent_block, main_block], queue=False)
 
     b0.click(random_sample, inputs = [], outputs = [image])
-    b2.click(reset_sample, inputs = [], outputs = [image])
+    b2.click(reset_sample, inputs = [], outputs = [image, *agents)
 
     b1.click(submit, inputs = [image, *agents], outputs = [image, *agents])