reset Gradio textbox on_next round

app.py

@@ -970,7 +970,7 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
     validate_btn.click(on_validate, inputs=[round_id_box, lat_box, lng_box, user_profile], outputs=[popup_html, ai_chat])
 
     def on_next(r_state: list, idx: int, profile: dict, request: gr.Request):
-        if not profile: return idx, gr.update(), gr.update(), gr.update()
+        if not profile: return idx, gr.update(), gr.update(), gr.update(), gr.update()
         username = profile["name"]
         idx += 1
         sess = user_sessions.get(username)
@@ -997,12 +997,12 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
                 <h3>{winner_message}</h3>
             </div>
             """
-            return idx, gr.update(value=summary_html), gr.update(value=""), gr.update(value="")
+            return idx, gr.update(value=summary_html), gr.update(value=""), gr.update(value=""), gr.update(value="")
         r = sess['rounds'][idx]
         s_html = build_street_html(r['image_url'])
-        return idx, gr.update(value=s_html), gr.update(value=r['id']), gr.update(value="")
+        return idx, gr.update(value=s_html), gr.update(value=r['id']), gr.update(value=""), gr.update(value="")
 
-    next_btn.click(on_next, inputs=[rounds_state, idx_state, user_profile], outputs=[idx_state, street_html, round_id_box, popup_html])
+    next_btn.click(on_next, inputs=[rounds_state, idx_state, user_profile], outputs=[idx_state, street_html, round_id_box, popup_html, ai_chat])
 
     # Inject boot JS using load(js=callable) compatible format
     demo.load(fn=lambda: None, inputs=None, outputs=None, js=APP_BOOT_JS)