fix booking issues

app.py

@@ -8,7 +8,14 @@ from openai.types.responses import ResponseTextDeltaEvent
 
 
 
+# @cl.on_message
+# async def main(message: cl.Message):
+#     # Your custom logic goes here...
 
+#     # Send a response back to the user
+#     await cl.Message(
+#         content=f"Received: {message.content}",
+#     ).send()
 
 CSV_PATH = "availablity.csv"
 
@@ -47,18 +54,18 @@ def get_availability(date: str) -> str:
 
 # Book a room if available
 @function_tool
-def book_room(date: str, room_type: str, guests: int) -> str:
+def book_room(date: str, room_type: str, guests: int, num_rooms: int = 1) -> str:
     """
-    Attempts to book a room of the specified type on a given date.
+    Attempts to book a specified number of rooms of the given type on a date.
 
-Args:
-    date (str): The booking date (format: YYYY-MM-DD).
-    room_type (str): The type of room to book (e.g., 'single', 'double', 'suite').
-    guests (int): Number of guests for the booking.
+    Args:
+        date (str): The booking date (format: YYYY-MM-DD).
+        room_type (str): The type of room to book (e.g., 'single', 'double', 'suite').
+        guests (int): Number of guests for the booking.
+        num_rooms (int, optional): Number of rooms to book. Defaults to 1.
 
-Returns:
-    str: A confirmation message if booking is successful,
-            or an error message if the room is unavailable or does not exist.
+    Returns:
+        str: A confirmation message if booking is successful, or an error message.
     """
     df = load_data()
     match = (df['date'] == date) & (df['room_type'].str.lower() == room_type.lower())
@@ -68,14 +75,12 @@ Returns:
         return f"No {room_type} room available on {date}."
 
     idx = room_row.index[0]
-    if df.at[idx, 'available'] > 0:
-        df.at[idx, 'available'] -= 1
+    if df.at[idx, 'available'] >= num_rooms:
+        df.at[idx, 'available'] -= num_rooms
         save_data(df)
-        return f"✅ Booked a {room_type.title()} room for {guests} guest(s) on {date}."
+        return f"✅ Booked {num_rooms} {room_type.title()} room(s) for {guests} guest(s) on {date}."
     else:
-        return f"❌ No {room_type.title()} rooms left on {date}."
-
-
+        return f"❌ Only {df.at[idx, 'available']} {room_type.title()} rooms left on {date}."
 
 load_dotenv(find_dotenv())
 api_key = os.getenv('OpenAI_api_key')