updating code

app.py

@@ -1,13 +1,15 @@
+from dotenv import load_dotenv
+import gradio as gr
+import requests
+from retry import retry
+from requests.exceptions import ConnectTimeout, ReadTimeout
+import os
+from uuid import uuid4
+from typing import Optional, Sequence, Tuple
+
+
 """gradio app for Chat Interface for DataStax Langflow calls"""
 
-from typing import Optional, Sequence, Tuple
-from uuid import uuid4
-import os
-from requests.exceptions import ConnectTimeout, ReadTimeout
-from retry import retry
-import requests
-import gradio as gr
-from dotenv import load_dotenv
 
 load_dotenv()
 
@@ -35,7 +37,7 @@ TWEAKS = {
 
 
 @retry((ConnectTimeout, ReadTimeout), tries=3, delay=2)
-def call_travel_ai(
+def call_ai(
     message: str,
     history: Sequence[Tuple[str, str]],
     local_storage: str,
@@ -77,7 +79,7 @@ def call_travel_ai(
         headers=headers,
         timeout=120,
     )
-    print(response.content)
+    print(response, response.content)
     response = response.json()
     resp_message = ''
     try:
@@ -101,25 +103,52 @@ 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:
         chat_interface = gr.ChatInterface(
-            call_travel_ai,
+            call_ai,
             type='messages',
-            title="AI Travel Partner",
+            title="AI Beauty Partner",
             additional_inputs=[session],
             autofocus=True,
             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,)