Update app.py

app.py

@@ -312,6 +312,11 @@ def generate_answer(message, choice, retrieval_mode):
         response = fetch_yelp_restaurants()
         return response, extract_addresses(response)
 
+    # Check if the question is about hotels
+    if "hotel" in message.lower() and "birmingham" in message.lower():
+        response = fetch_hotels_in_birmingham()
+        return response, extract_addresses(response)
+
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
     if retrieval_mode == "VDB":
@@ -962,6 +967,41 @@ def fetch_yelp_restaurants():
 
     return response_text
 
+def fetch_hotels_in_birmingham():
+    params = {
+      "engine": "google_hotels",
+      "q": "Hotels in Birmingham, AL",
+      "check_in_date": "2024-08-14",
+      "check_out_date": "2024-08-15",
+      "adults": "2",
+      "currency": "USD",
+      "gl": "us",
+      "hl": "en",
+      "api_key": os.getenv("SERP_API")
+    }
+
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    hotels = results.get("organic_results", [])
+
+    response_text = ""
+
+    for hotel in hotels[:5]:  # Limiting to top 5 hotels
+        name = hotel.get("title", "No name")
+        rating = hotel.get("rating", "No rating")
+        reviews = hotel.get("reviews", "No reviews")
+        price = hotel.get("price", "No price")
+        location = f"{name}, Birmingham, AL"
+        link = hotel.get("link", "#")
+
+        # Format the output for each hotel
+        response_text += f"[{name}]({link})\n"
+        response_text += f"*{location}*\n"
+        response_text += f"**Price:** {price}\n"
+        response_text += f"**Rating:** {rating} ({reviews} reviews)\n"
+        response_text += "-" * 50 + "\n"
+
+    return response_text