Update app.py

app.py

@@ -314,7 +314,7 @@ def generate_answer(message, choice, retrieval_mode):
 
     # Check if the question is about hotels
     elif "hotels" in message.lower() and "birmingham" in message.lower():
-        response = fetch_hotels_birmingham()
+        response = fetch_google_hotels()
         return response, extract_addresses(response)
 
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
@@ -968,33 +968,27 @@ def fetch_yelp_restaurants():
 from serpapi.google_search import GoogleSearch
 from datetime import datetime, timedelta
 
-def fetch_hotels_birmingham():
+def fetch_google_hotels():
+   
+    import os
+
     params = {
-        "engine": "google_hotels",
-        "q": "Birmingham Resorts,AL",
-        "check_in_date": "2024-08-14",
-        "check_out_date": "2024-08-15",
-        "adults": "2",
-        "currency": "USD",
-        "gl": "us",
-        "hl": "en",
-        "api_key": "74972e6cc9f9e092bd86764a78f7d18c1d71b156eca8b1fe7fd65c2406cfb588"
+      "engine": "google_hotels",
+      "q": "Birmingham Resorts",
+      "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")  # Ensure you use your SERP API key
     }
 
-    # Send the request to the Google Hotels API
     search = GoogleSearch(params)
     results = search.get_dict()
-
-    # Extract the hotel results from the response
-    hotels = results.get("hotels_results", [])
-
-    # Check if any hotels were found
-    if not hotels:
-        print("No resorts found in the response.")
-        return "No resort information available."
+    hotel_results = results.get("hotels_results", [])
 
     def star_rating(rating):
-        # Convert the rating to a star format with half-stars if necessary
         full_stars = int(float(rating))
         half_star = 1 if (float(rating) - full_stars) >= 0.5 else 0
         empty_stars = 5 - full_stars - half_star
@@ -1003,28 +997,21 @@ def fetch_hotels_birmingham():
 
     response_text = ""
     
-    # Iterate through the top 5 resorts and format the response
-    for hotel in hotels[:5]:  # Limiting to top 5 resorts
+    for hotel in hotel_results[:5]:  # Limiting to top 5 hotels
         name = hotel.get("title", "No name")
         rating = hotel.get("rating", "No rating")
         reviews = hotel.get("reviews", "No reviews")
-        location = f"{name}, Bali"
-        price = hotel.get("price", "Price not available")
+        address = hotel.get("address", "Not Available")
+        price = hotel.get("price", "Not Available")
         link = hotel.get("link", "#")
-        amenities = hotel.get("amenities", "Amenities not available")
-
-        if isinstance(amenities, list):
-            amenities = ", ".join(amenities)
 
-        # Format the output for each resort
-        response_text += f"[{name}]({link})\n"  # Name with clickable link
-        response_text += f"*{location}*\n"  # Location with resort name and "Bali"
-        response_text += f"**Price:** {price}\n"
-        response_text += f"**Amenities:** {amenities}\n"
+        # Format the output for each hotel
+        response_text += f"[{name}]({link})\n"  # Name with clickable link, no bold
+        response_text += f"*{name}, {address}*\n"  # Location with hotel name and address
         response_text += f"**Rating:** {star_rating(rating)} ({rating} stars, {reviews} reviews)\n"
+        response_text += f"**Price:** {price}\n"
         response_text += "-" * 50 + "\n"
 
-    # Return the formatted response
     return response_text
 
 
@@ -1035,6 +1022,7 @@ def fetch_hotels_birmingham():
 
 
 
+
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     with gr.Row():
         with gr.Column():