Update for Browser Storage

app.py

@@ -89,7 +89,7 @@ def submit_signup(_email: str, _name: str, _contact: str):
 def call_travel_ai(
     message: str,
     history: Sequence[Tuple[str, str]],
-    session_id: str,
+    local_storage: str,
     endpoint: str = ENDPOINT,
     output_type: str = "chat",
     input_type: str = "chat",
@@ -108,7 +108,7 @@ def call_travel_ai(
     api_url = f"{BASE_API_URL}/lf/{LANGFLOW_ID}/api/v1/run/{endpoint}"
 
     payload = {
-        'session_id': session_id,
+        'session_id': local_storage[0],
         "input_value": message,
         "output_type": output_type,
         "input_type": input_type,
@@ -129,10 +129,14 @@ def call_travel_ai(
     )
     response = response.json()
     resp_message = ''
-    for resp in response['outputs']:
-        for _resp in resp['outputs']:
-            for message in _resp['messages']:
-                resp_message = message['message']
+    try:
+        for resp in response['outputs']:
+            for _resp in resp['outputs']:
+                for message in _resp['messages']:
+                    resp_message = message['message']
+    except KeyError as e:
+        print(e)
+        print(response)
 
     return resp_message
 
@@ -195,7 +199,6 @@ with gr.Blocks() as demo:
         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)