Update app.py

app.py

@@ -95,6 +95,9 @@ async def set_starters():
 #------?-----@cl.Context
 
 @cl.on_chat_start
+async def start():
+    cl.user_session.set("socket_auth", True) 
+    
 async def setup():
     asyncio.create_task(keep_alive())
 
@@ -127,102 +130,112 @@ def setup_multiple_chains():
     )
 
     cl.user_session.set("llm_chain", llm_chain)
+    
 
 @cl.on_message
 async def handle_message(message: cl.Message):
-    user_message = message.content
-    llm_chain = cl.user_session.get("llm_chain")
-    #languages = cl.user_session.get("languages")
-
-    if sys.version_info >= (3, 11):
-        from asyncio import timeout
-    else:
-        from async_timeout import timeout
-
-    booking_pattern = r'\b[A-Z]{6}\d{6}\b'
-    match = re.search(booking_pattern, user_message)
-
-    if match:
-        bestillingskode = match.group()
-        headers = {
-            "Authorization": auth_token,
-            "Content-Type": "application/json"
-        }
-        payload = {"booking_id": bestillingskode}
-
-        try:
-            
-            response = await async_post_request(API_URL, headers, payload)
-            response.raise_for_status()
-            booking_data = response.json()
 
-            if not booking_data:
-                raise BookingNotFoundError("No booking data returned")
+    try:
+        if not cl.user_session.get("socket_auth"):
+            await cl.Message(content="Session authentication failed. Please refresh the page.").send()
+            return
             
-            if "error" in booking_data:
-                raise APIResponseError(booking_data["error"])
-
-            #table = (
-                #"| 𝑭𝒊𝒆𝒍𝒅           | 𝗜𝗻𝗳𝗼                                      |\n"
-                #"|:----------------|:-------------------------------------------|\n"
-                #f"| 𝙊𝙧𝙙𝙚𝙧 𝙉𝙪𝙢𝙗𝙚𝙧   | {booking_data.get('order_number', 'N/A')}        |\n"
-                #f"| 𝘼𝙘𝙘𝙚𝙥𝙩𝙚𝙙       | {booking_data.get('accepted', 'N/A')}           |\n"
-                #f"| 𝙂𝙪𝙚𝙨𝙩𝙨         | {booking_data.get('guests', 'N/A')}             |\n"
-                #f"| 𝙏𝙞𝙢𝙚 𝙁𝙧𝙤𝙢     | {booking_data.get('time_from', 'N/A')}          |\n"
-                #f"| 𝙏𝙞𝙢𝙚 𝙏𝙤       | {booking_data.get('time_to', 'N/A')}            |\n"
-                #f"| 𝘾𝙖𝙣𝙘𝙚𝙡𝙡𝙚𝙙     | {'Yes' if booking_data.get('cancelled', 0) else 'No'} |\n"
-                #f"| 𝘿𝙚𝙨𝙘𝙧𝙞𝙥𝙩𝙞𝙤𝙣   | {booking_data.get('description', 'N/A')}        |\n"
-                #f"| 𝘼𝙧𝙧𝙞𝙫𝙖𝙡       | {booking_data.get('arrival', 'N/A')}            |\n"
-                #f"| 𝘿𝙚𝙥𝙖𝙧𝙩𝙪𝙧𝙚     | {booking_data.get('departure', 'N/A')}          |\n"
-                #f"| 𝘽𝙚𝙙𝙙𝙞𝙣𝙜       | {booking_data.get('bedding', 'N/A')}            |\n"
-                #f"| 𝙇𝙤𝙘𝙖𝙩𝙞𝙤𝙣      | Lat: {booking_data.get('lat', 'N/A')}, Long: {booking_data.get('long', 'N/A')} |\n"
-                #f"| 𝙉𝙖𝙢𝙚          | {booking_data.get('name', 'N/A')}                |\n"
-                #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙉𝙖𝙢𝙚 | {booking_data.get('contactname', 'N/A')}         |\n"
-                #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙀𝙢𝙖𝙞𝙡| {booking_data.get('contactemail', 'N/A')}        |\n"
-                #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙋𝙝𝙤𝙣𝙚| {booking_data.get('contactphone', 'N/A')}        |\n"
-                #f"| 𝘾𝙤𝙪𝙣𝙩𝙧𝙮 𝘾𝙤𝙙𝙚 | {booking_data.get('country_code', 'N/A')}        |\n"
-            #)
-            #combined_message = f"### Informasjon om booking:\n\n{table}"
-            #await cl.Message(content=combined_message).send()
-
-            table = (
-                "| 𝑭𝒊𝒆𝒍𝒅      | 𝗜𝗻𝗳𝗼                |\n"
-                "|:-----------|:---------------------|\n"
-                f"| 𝙱𝚎𝚜𝚝𝚒𝚕𝚕𝚒𝚗𝚐𝚜𝚔𝚘𝚍𝚎 | {booking_data.get('booking_id', 'N/A')} |\n"
-                f"| 𝙁𝙪𝙡𝙡 𝙉𝙖𝙢𝙚  | {booking_data.get('full_name', 'N/A')} |\n"
-                f"| 𝘼𝙢𝙤𝙪𝙣𝙩     | {booking_data.get('amount', 0)} kr |\n"
-                f"| 𝘾𝙝𝙚𝙘𝙠-𝙞𝙣   | {booking_data.get('checkin', 'N/A')} |\n"
-                f"| 𝘾𝙝𝙚𝙘𝙠-𝙤𝙪𝙩  | {booking_data.get('checkout', 'N/A')} |\n"
-                f"| 𝘼𝙙𝙙𝙧𝙚𝙨𝙨    | {booking_data.get('address', 'N/A')} |\n"
-                f"| 𝙐𝙨𝙚𝙧 𝙄𝘿    | {booking_data.get('user_id', 0)} |\n"
-                f"| 𝙄𝙣𝙛𝙤 𝙏𝙚𝙭𝙩  | {booking_data.get('infotext', 'N/A')} |\n"
-                f"| 𝙄𝙣𝙘𝙡𝙪𝙙𝙚𝙙   | {booking_data.get('included', 'N/A')} |"
-            )
-
-            combined_message = f"### Informasjon om booking:\n\n{table}"
-            await cl.Message(content=combined_message).send()
-
-        except (APIConnectionError, APIResponseError, BookingNotFoundError) as e:
-            error_messages = {
-                APIConnectionError: "Kunne ikke koble til bookingsystemet. Prøv igjen senere.",
-                APIResponseError: "Det oppstod en feil ved henting av bookingdata.",
-                BookingNotFoundError: "Ingen booking funnet med denne koden."
+        user_message = message.content
+        llm_chain = cl.user_session.get("llm_chain")
+        #languages = cl.user_session.get("languages")
+    
+        if sys.version_info >= (3, 11):
+            from asyncio import timeout
+        else:
+            from async_timeout import timeout
+    
+        booking_pattern = r'\b[A-Z]{6}\d{6}\b'
+        match = re.search(booking_pattern, user_message)
+    
+        if match:
+            bestillingskode = match.group()
+            headers = {
+                "Authorization": auth_token,
+                "Content-Type": "application/json"
             }
-            await cl.Message(content=f"❌ {error_messages[type(e)]}\n\nPrøv igjen, kanskje du feilstavet eller glemte ett siffer eller en bokstav?\n\nHvis du ser denne feilmedlingen gjentatte ganger, vennligst kontakt kundeservice@daysoff.no så de kan hjelpe deg.").send()
-            return None
-        except requests.exceptions.RequestException as e:
-            await cl.Message(content="En uventet feil oppstod. Vennligst kontakt kundeservice@daysoff.no").send()
-            return None
-
-    else:
-        try:
-            async with timeout(30): 
-                response = await llm_chain.ainvoke({
-                    "question": user_message,
-                    "chat_history": ""
-                }, callbacks=[cl.AsyncLangchainCallbackHandler()])
-            await cl.Message(content=response["text"]).send()
-        except asyncio.TimeoutError:
-            await cl.Message(content="Request timed out. Please try again.").send()
-        except Exception as e:
-            await cl.Message(content=f"Error: {str(e)}").send()
+            payload = {"booking_id": bestillingskode}
+    
+            try:
+                
+                response = await async_post_request(API_URL, headers, payload)
+                response.raise_for_status()
+                booking_data = response.json()
+    
+                if not booking_data:
+                    raise BookingNotFoundError("No booking data returned")
+                
+                if "error" in booking_data:
+                    raise APIResponseError(booking_data["error"])
+    
+                #table = (
+                    #"| 𝑭𝒊𝒆𝒍𝒅           | 𝗜𝗻𝗳𝗼                                      |\n"
+                    #"|:----------------|:-------------------------------------------|\n"
+                    #f"| 𝙊𝙧𝙙𝙚𝙧 𝙉𝙪𝙢𝙗𝙚𝙧   | {booking_data.get('order_number', 'N/A')}        |\n"
+                    #f"| 𝘼𝙘𝙘𝙚𝙥𝙩𝙚𝙙       | {booking_data.get('accepted', 'N/A')}           |\n"
+                    #f"| 𝙂𝙪𝙚𝙨𝙩𝙨         | {booking_data.get('guests', 'N/A')}             |\n"
+                    #f"| 𝙏𝙞𝙢𝙚 𝙁𝙧𝙤𝙢     | {booking_data.get('time_from', 'N/A')}          |\n"
+                    #f"| 𝙏𝙞𝙢𝙚 𝙏𝙤       | {booking_data.get('time_to', 'N/A')}            |\n"
+                    #f"| 𝘾𝙖𝙣𝙘𝙚𝙡𝙡𝙚𝙙     | {'Yes' if booking_data.get('cancelled', 0) else 'No'} |\n"
+                    #f"| 𝘿𝙚𝙨𝙘𝙧𝙞𝙥𝙩𝙞𝙤𝙣   | {booking_data.get('description', 'N/A')}        |\n"
+                    #f"| 𝘼𝙧𝙧𝙞𝙫𝙖𝙡       | {booking_data.get('arrival', 'N/A')}            |\n"
+                    #f"| 𝘿𝙚𝙥𝙖𝙧𝙩𝙪𝙧𝙚     | {booking_data.get('departure', 'N/A')}          |\n"
+                    #f"| 𝘽𝙚𝙙𝙙𝙞𝙣𝙜       | {booking_data.get('bedding', 'N/A')}            |\n"
+                    #f"| 𝙇𝙤𝙘𝙖𝙩𝙞𝙤𝙣      | Lat: {booking_data.get('lat', 'N/A')}, Long: {booking_data.get('long', 'N/A')} |\n"
+                    #f"| 𝙉𝙖𝙢𝙚          | {booking_data.get('name', 'N/A')}                |\n"
+                    #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙉𝙖𝙢𝙚 | {booking_data.get('contactname', 'N/A')}         |\n"
+                    #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙀𝙢𝙖𝙞𝙡| {booking_data.get('contactemail', 'N/A')}        |\n"
+                    #f"| 𝘾𝙤𝙣𝙩𝙖𝙘𝙩 𝙋𝙝𝙤𝙣𝙚| {booking_data.get('contactphone', 'N/A')}        |\n"
+                    #f"| 𝘾𝙤𝙪𝙣𝙩𝙧𝙮 𝘾𝙤𝙙𝙚 | {booking_data.get('country_code', 'N/A')}        |\n"
+                #)
+                #combined_message = f"### Informasjon om booking:\n\n{table}"
+                #await cl.Message(content=combined_message).send()
+    
+                table = (
+                    "| 𝑭𝒊𝒆𝒍𝒅      | 𝗜𝗻𝗳𝗼                |\n"
+                    "|:-----------|:---------------------|\n"
+                    f"| 𝙱𝚎𝚜𝚝𝚒𝚕𝚕𝚒𝚗𝚐𝚜𝚔𝚘𝚍𝚎 | {booking_data.get('booking_id', 'N/A')} |\n"
+                    f"| 𝙁𝙪𝙡𝙡 𝙉𝙖𝙢𝙚  | {booking_data.get('full_name', 'N/A')} |\n"
+                    f"| 𝘼𝙢𝙤𝙪𝙣𝙩     | {booking_data.get('amount', 0)} kr |\n"
+                    f"| 𝘾𝙝𝙚𝙘𝙠-𝙞𝙣   | {booking_data.get('checkin', 'N/A')} |\n"
+                    f"| 𝘾𝙝𝙚𝙘𝙠-𝙤𝙪𝙩  | {booking_data.get('checkout', 'N/A')} |\n"
+                    f"| 𝘼𝙙𝙙𝙧𝙚𝙨𝙨    | {booking_data.get('address', 'N/A')} |\n"
+                    f"| 𝙐𝙨𝙚𝙧 𝙄𝘿    | {booking_data.get('user_id', 0)} |\n"
+                    f"| 𝙄𝙣𝙛𝙤 𝙏𝙚𝙭𝙩  | {booking_data.get('infotext', 'N/A')} |\n"
+                    f"| 𝙄𝙣𝙘𝙡𝙪𝙙𝙚𝙙   | {booking_data.get('included', 'N/A')} |"
+                )
+    
+                combined_message = f"### Informasjon om booking:\n\n{table}"
+                await cl.Message(content=combined_message).send()
+    
+            except (APIConnectionError, APIResponseError, BookingNotFoundError) as e:
+                error_messages = {
+                    APIConnectionError: "Kunne ikke koble til bookingsystemet. Prøv igjen senere.",
+                    APIResponseError: "Det oppstod en feil ved henting av bookingdata.",
+                    BookingNotFoundError: "Ingen booking funnet med denne koden."
+                }
+                await cl.Message(content=f"❌ {error_messages[type(e)]}\n\nPrøv igjen, kanskje du feilstavet eller glemte ett siffer eller en bokstav?\n\nHvis du ser denne feilmedlingen gjentatte ganger, vennligst kontakt kundeservice@daysoff.no så de kan hjelpe deg.").send()
+                return None
+            except requests.exceptions.RequestException as e:
+                await cl.Message(content="En uventet feil oppstod. Vennligst kontakt kundeservice@daysoff.no").send()
+                return None
+    
+        else:
+            try:
+                async with timeout(30): 
+                    response = await llm_chain.ainvoke({
+                        "question": user_message,
+                        "chat_history": ""
+                    }, callbacks=[cl.AsyncLangchainCallbackHandler()])
+                await cl.Message(content=response["text"]).send()
+            except asyncio.TimeoutError:
+                await cl.Message(content="Request timed out. Please try again.").send()
+            except Exception as e:
+                await cl.Message(content=f"Error: {str(e)}").send()
+
+    except Exception as e:
+        await cl.Message(content="Connection error. WTF yo...").send()