updating code

app.py

@@ -101,15 +101,22 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.Image(
             "./aem_logo.jpeg",
-            width=300, label="AEM", show_download_button=False,
+            width=300,
+            label="AEM",
+            show_download_button=False,
             show_fullscreen_button=False,
             show_label=False,
             interactive=False,
             show_share_button=False,
         )
 
-    session = gr.BrowserState(["", False])
-
+    # session = gr.State(["", ], time_to_live=24 * 60*60)
+    session = gr.BrowserState(None, storage_key="session_id")
+    show_session = gr.Textbox(
+        visible=False,
+        interactive=False,
+        label="Session ID",
+    )
     with gr.Accordion(
         'Chat With Agent',
     ) as chat:
@@ -122,4 +129,24 @@ with gr.Blocks() as demo:
             fill_height=True,
         )
 
-demo.launch(share=False, debug=True,)
+    def initialize_session(existing_id):
+        """
+        Runs ONCE per browser session on page load.
+        Checks if an ID already exists in BrowserState (local storage).
+        If not, generates a new one.
+        Returns the ID to populate the BrowserState and other components.
+        """
+        if existing_id:
+            print(f"Found existing ID in BrowserState: {existing_id}")
+            return existing_id, existing_id
+        else:
+            new_id = generate_session_id()
+            print(f"No existing ID found, returning NEW ID: {new_id}")
+            return new_id, new_id
+
+    demo.load(  # pylint: disable=no-member
+        initialize_session,
+        [session],
+        outputs=[session, show_session],
+    )
+    demo.launch(share=False, debug=True,)