Update app.py

app.py

@@ -57,7 +57,6 @@ Polish Terms:
 
 If there is no booking information available for a booking ID, just say so and politely suggest they contact **[email protected]**
 to enquire further.
-By default, you respond using Norwegian bokmål.
 
 Maintain a conversational and professional tone, **reflecting the warmth of
 a female customer support representative archetype.** 
@@ -72,10 +71,37 @@ daysoff_assistant_prompt = PromptTemplate(
     template=daysoff_assistant_template,
 )
 
-# -- async wrapper for requests.post
-async def async_post_request(url, headers, data):
-    return await asyncio.to_thread(requests.post, url, headers=headers, json=data)
 
+# =============================================================================================================
+class APIConnectionError(Exception):
+    """Raised when API connection fails"""
+    pass
+
+class APIResponseError(Exception):
+    """Raised when API returns invalid response"""
+    pass
+
+class BookingNotFoundError(Exception):
+    """Raised when booking ID is not found"""
+    pass
+# =============================================================================================================
+
+#async def async_post_request(url, headers, data):
+    #return await asyncio.to_thread(requests.post, url, headers=headers, json=data)
+
+# =============================================================================================================
+async def async_post_request(url, headers, data):
+    try:
+        response = await asyncio.to_thread(requests.post, url, headers=headers, json=data)
+        response.raise_for_status()
+        return response
+    except requests.ConnectionError:
+        raise APIConnectionError("Failed to connect to booking service")
+    except requests.Timeout:
+        raise APIConnectionError("Request timed out")
+    except requests.RequestException as e:
+        raise APIResponseError(f"API request failed: {str(e)}")
+# =============================================================================================================
 
 @cl.on_chat_start
 def setup_multiple_chains():
@@ -125,49 +151,48 @@ async def handle_message(message: cl.Message):
             response.raise_for_status()
             booking_data = response.json()
 
-            if "booking_id" in booking_data:
-                try:
-                    # --markdown_table
-                    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')} |"
-                    )
-
-                    # --send both as combined_message
-                    combined_message = f"### Informasjon for Bestillingskode:\n\n{table}"
-                    await cl.Message(content=combined_message).send()
-
-                except Exception as e:
-                    await cl.Message(content=f"Error processing booking data: {str(e)}").send()
-
-            else:
-                await cl.Message(content="Booking not found").send()
-
+            if not booking_data:
+                raise BookingNotFoundError("No booking data returned")
+            
+            if "error" in booking_data:
+                raise APIResponseError(booking_data["error"])
+
+            # --markdown_table
+            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 for Bestillingskode:\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\nKontakt [email protected] for assistanse.").send()
+            return None
         except requests.exceptions.RequestException as e:
-            await cl.Message(content=f"Request failed: {str(e)}").send()
+            await cl.Message(content="En uventet feil oppstod. Vennligst kontakt [email protected]").send()
+            return None
 
     else:
         try:
-            # --invoke LLM w/ user_message
             response = await llm_chain.ainvoke({
                 "question": user_message,
                 "chat_history": ""
             }, callbacks=[cl.AsyncLangchainCallbackHandler()])
-
-            # Send the LLM response as a message
-            #await cl.Message(content=response.get("text")).send()
             await cl.Message(content=response["text"]).send()
-
         except Exception as e:
             await cl.Message(content=f"Error: {str(e)}").send()
-
-