Update for Browser Storage

app.py

@@ -142,16 +142,31 @@ def generate_session_id():
     return str(uuid4())
 
 
+# def render(session_id: str):
+#     """Render the session ID."""
+#     return session_id
+
+
+def on_load(local_storage):
+    """Update the visibility of the sign up and chat interfaces."""
+    signed_up = local_storage[1]
+    return gr.update(visible=not signed_up), gr.update(visible=signed_up)
+
+
 with gr.Blocks() as demo:
-    random_session_id = gr.Textbox(
-        value=generate_session_id,
-        visible=False,
-        interactive=False,
-        label="Session ID",
-    )
+
+    session = gr.BrowserState(["", False])
+    # gr.Textbox(
+    #     value=render,
+    #     visible=False,
+    #     interactive=False,
+    #     label="Session ID",
+    #     inputs=[session],
+    # )
 
     with gr.Accordion(
         'Sign Up',
+        visible=lambda session: not session[1],
     ) as signup:
 
         with gr.Column():
@@ -168,7 +183,7 @@ with gr.Blocks() as demo:
             call_travel_ai,
             type='messages',
             title="AI Travel Partner",
-            additional_inputs=[random_session_id],
+            additional_inputs=[session],
             autofocus=True,
             fill_height=True,
         )
@@ -177,20 +192,31 @@ with gr.Blocks() as demo:
         _name: str,
         _email: str,
         _phone: str,
-
+        local_storage: str
     ):
         """Update the visibility of the sign up and chat interfaces."""
+        print(local_storage)
+        # session_id, signed_up = local_storage
+        local_storage[0] = local_storage[0] or generate_session_id()
         _status = submit_signup(_email, _name, _phone)
         gr.Info(_status, duration=5, title='Sign Up Status')
+        if _status == 'Success':
+            local_storage[1] = True
 
         return (
-            gr.update(visible=_status != 'Success'),
-            gr.update(visible=_status == 'Success')
+            gr.update(visible=not local_storage[1]),
+            gr.update(visible=local_storage[1]),
+            local_storage
         )
 
     submit_button.click(  # pylint: disable=no-member
         update_visibility,
-        [name, email, phone],
+        [name, email, phone, session],
+        outputs=[signup, chat, session],
+    )
+    demo.load(  # pylint: disable=no-member
+        on_load,
+        [session],
         outputs=[signup, chat],
     )