Update app.py

app.py

@@ -540,6 +540,7 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
         "chat_id": None,
         "session": None,
         "opt_out": DEFAULT_OPT_OUT_VALUE,
+        "agreed": False,
     })  # Store session state as a dictionary
 
     gr.HTML(f"""
@@ -632,11 +633,12 @@ Because we do not track user identities, individual submissions cannot be delete
                   )
 
 
-    def agree_to_terms():
+    def agree_to_terms(state):
         log_info("Privacy agreement accepted by user")
-        return gr.update(visible=False)
+        state["agreed"] = True
+        return gr.update(visible=False), state
 
-    agree_btn.click(agree_to_terms, None, [agreement_overlay])
+    agree_btn.click(agree_to_terms, inputs=[session_state], outputs=[agreement_overlay, session_state], queue=False)
 
     gr.on(
         triggers=[send_btn.click, user_input.submit],