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
+    elif "hotels" in message.lower() and "birmingham" in message.lower():
+        response = fetch_hotels_birmingham()
+        return response, extract_addresses(response)
+
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
     if retrieval_mode == "VDB":
@@ -332,6 +337,7 @@ def generate_answer(message, choice, retrieval_mode):
         return "Invalid retrieval mode selected.", []
 
 
+
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
@@ -911,87 +917,7 @@ def handle_retrieval_mode_change(choice):
         return gr.update(interactive=True), gr.update(interactive=True)
 
 
-# def fetch_yelp_restaurants():
-#     from serpapi.google_search import GoogleSearch
-#     import os
-
-#     params = {
-#       "engine": "yelp",
-#       "find_desc": "Restaurant",
-#       "find_loc": "Birmingham, AL, USA",
-#       "api_key": os.getenv("SERP_API")
-#     }
-
-#     search = GoogleSearch(params)
-#     results = search.get_dict()
-#     organic_results = results.get("organic_results", [])
-
-#     yelp_html = """
-#     <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Top Restaurants</h2>
-#     <style>
-#         table {
-#             font-family: 'Verdana', sans-serif;
-#             color: #333;
-#             border-collapse: collapse;
-#             width: 100%;
-#         }
-#         th, td {
-#             border: 1px solid #fff !important;
-#             padding: 8px;
-#         }
-#         th {
-#             background-color: #f2f2f2;
-#             color: #333;
-#             text-align: left;
-#         }
-#         tr:hover {
-#             background-color: #f5f5f5;
-#         }
-#         .restaurant-link {
-#             color: #1E90FF;
-#             text-decoration: none;
-#         }
-#         .restaurant-link:hover {
-#             text-decoration: underline;
-#         }
-#     </style>
-#     <table>
-#         <tr>
-#             <th>Name</th>
-#             <th>Rating</th>
-#             <th>Reviews</th>
-#             <th>Phone</th>
-#             <th>Snippet</th>
-#             <th>Services</th>
-#         </tr>
-#     """
-#     for result in organic_results:
-#         name = result.get("title", "No name")
-#         rating = result.get("rating", "No rating")
-#         reviews = result.get("reviews", "No reviews")
-#         phone = result.get("phone", "Not Available")
-#         snippet = result.get("snippet", "No Available")
-#         services = result.get("service_options", "Not Known")
-
-#         if isinstance(services, list):
-#             services = ", ".join(services)
-#         elif isinstance(services, dict):
-#             services = ", ".join([f"{key}: {value}" for key, value in services.items()])
-
-#         link = result.get("link", "#")
-
-#         yelp_html += f"""
-#         <tr>
-#             <td><a class='restaurant-link' href='{link}' target='_blank'>{name}</a></td>
-#             <td>{rating}</td>
-#             <td>{reviews}</td>
-#             <td>{phone}</td>
-#             <td>{snippet}</td>
-#             <td>{services}</td>
-#         </tr>
-#         """
-#     yelp_html += "</table>"
-#     return yelp_html
+
 def fetch_yelp_restaurants():
     from serpapi.google_search import GoogleSearch
     import os
@@ -1016,7 +942,7 @@ def fetch_yelp_restaurants():
 
     response_text = ""
     
-    for result in organic_results[:10]:  # Limiting to top 10 restaurants
+    for result in organic_results[:5]:  # Limiting to top 10 restaurants
         name = result.get("title", "No name")
         rating = result.get("rating", "No rating")
         reviews = result.get("reviews", "No reviews")
@@ -1039,6 +965,66 @@ def fetch_yelp_restaurants():
     return response_text
 
 
+from serpapi import GoogleSearch
+from datetime import datetime, timedelta
+
+def fetch_hotels_birmingham():
+    current_date = datetime.now().strftime("%Y-%m-%d")
+
+    params = {
+        "engine": "google_hotels",
+        "q": "Birmingham,AL",
+        "check_in_date": current_date,
+        "check_out_date": (datetime.now() + timedelta(days=2)).strftime("%Y-%m-%d"),
+        "vacation_rentals": "true",
+        "amenities": "2,4,10,15,16,20,21,24,29,32",
+        "rating": "7,8,9",
+        "hotel_class": "2,3,4,5",
+        "currency": "USD",
+        "gl": "us",
+        "hl": "en",
+        "api_key": "74972e6cc9f9e092bd86764a78f7d18c1d71b156eca8b1fe7fd65c2406cfb588"
+    }
+
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    hotels = results.get("hotels_results", [])
+
+    def star_rating(rating):
+        full_stars = int(float(rating))
+        half_star = 1 if (float(rating) - full_stars) >= 0.5 else 0
+        empty_stars = 5 - full_stars - half_star
+        stars = "★" * full_stars + "½" * half_star + "☆" * empty_stars
+        return stars
+
+    response_text = ""
+    
+    for hotel in hotels[:5]:  # Limiting to top 10 hotels
+        name = hotel.get("title", "No name")
+        rating = hotel.get("rating", "No rating")
+        reviews = hotel.get("reviews", "No reviews")
+        location = f"{name}, Birmingham, AL"
+        price = hotel.get("price", "Price not available")
+        link = hotel.get("link", "#")
+        hotel_class = hotel.get("hotel_class", "Not available")
+        amenities = hotel.get("amenities", "Amenities not available")
+
+        if isinstance(amenities, list):
+            amenities = ", ".join(amenities)
+
+        # Format the output for each hotel
+        response_text += f"[{name}]({link})\n"  # Name with clickable link, no bold
+        response_text += f"*{location}*\n"  # Location with hotel name and "Birmingham, AL"
+        response_text += f"**Price:** {price}\n"
+        response_text += f"**Hotel Class:** {hotel_class} stars\n"
+        response_text += f"**Amenities:** {amenities}\n"
+        response_text += f"**Rating:** {star_rating(rating)} ({rating} stars, {reviews} reviews)\n"
+        response_text += "-" * 50 + "\n"
+
+    return response_text
+
+
+