Update app.py

app.py

@@ -176,31 +176,38 @@ async def handle_message(message: cl.Message):
             "url": endpoint_url,
             "body": { 
                 "booking_id": bestillingskode
+            }
         }
-    }
 
         response = await post_tool.arun(
             json.dumps(post_data),
             callbacks=[cl.AsyncLangchainCallbackHandler()]
         )
 
+        # --degug!
+        print(f"Raw response: {response}")
+
         if response:
-            
-            booking_data = json.loads(response)
-            booking_data = f"""
-            | Field | Value |
-            |-------|--------|
-            | Booking ID | {booking_data['booking_id']} |
-            | Name | {booking_data['full_name']} |
-            | Amount | {booking_data['amount']} kr |
-            | Check-in | {booking_data['checkin']} |
-            | Check-out | {booking_data['checkout']} |
-            | Address | {booking_data['address']} |
-            | User ID | {booking_data['user_id']} |
-            | Info | {booking_data['infotext']} |
-            | Included | {booking_data['included']} |
-            """
-            await cl.Message(booking_data).send()
+            try:
+                booking_data = json.loads(response)
+        
+                booking_data = f"""
+                | Field | Value |
+                |-------|--------|
+                | Booking ID | {booking_data.get('booking_id', 'N/A')} |
+                | Name | {booking_data.get('full_name', 'N/A')} |
+                | Amount | {booking_data.get('amount', 'N/A')} kr |
+                | Check-in | {booking_data.get('checkin', 'N/A')} |
+                | Check-out | {booking_data.get('checkout', 'N/A')} |
+                | Address | {booking_data.get('address', 'N/A')} |
+                | User ID | {booking_data.get('user_id', 'N/A')} |
+                | Info | {booking_data.get('infotext', 'N/A')} |
+                | Included | {booking_data.get('included', 'N/A')} |
+                """
+                await cl.Message(booking_data).send()
+        
+            except json.JSONDecodeError:
+                await cl.Message(f"Error: Non-JSON response received. Raw response: {response}").send()
 
     else:
         response = await llm_chain.ainvoke(