Update app.py

app.py

@@ -307,16 +307,16 @@ chain_neo4j = (
 def generate_answer(message, choice, retrieval_mode):
     logging.debug(f"generate_answer called with choice: {choice} and retrieval_mode: {retrieval_mode}")
 
+    # Check if the question is about hotels
+    if "hotel" in message.lower() and "birmingham" in message.lower():
+        response = fetch_google_hotels()
+        return response, extract_addresses(response)
+    
     # Check if the question is about restaurants
     if "restaurant" in message.lower() and "birmingham" in message.lower():
         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":
@@ -341,6 +341,7 @@ def generate_answer(message, choice, retrieval_mode):
 
 
 
+
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
@@ -967,60 +968,38 @@ def fetch_yelp_restaurants():
 
     return response_text
 
-def fetch_and_display_hotels():
+def fetch_google_hotels(query="Birmingham hotels", check_in="2024-08-14", check_out="2024-08-15", adults=2):
     params = {
-      "engine": "google_hotels",
-      "q": "Bali Resorts",
-      "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": query,
+        "check_in_date": check_in,
+        "check_out_date": check_out,
+        "adults": str(adults),
+        "currency": "USD",
+        "gl": "us",
+        "hl": "en",
+        "api_key": os.getenv("SERP_API")
     }
-
+    
     search = GoogleSearch(params)
     results = search.get_dict()
-    hotels = results.get("organic_results", [])
-
-    # Create an HTML table
-    table_html = """
-    <table style="width:100%; border-collapse: collapse; font-family: Arial, sans-serif;">
-        <tr style="background-color: #f2f2f2;">
-            <th style="padding: 8px; border: 1px solid #ddd;">Name</th>
-            <th style="padding: 8px; border: 1px solid #ddd;">Location</th>
-            <th style="padding: 8px; border: 1px solid #ddd;">Rating</th>
-            <th style="padding: 8px; border: 1px solid #ddd;">Reviews</th>
-            <th style="padding: 8px; border: 1px solid #ddd;">Price</th>
-            <th style="padding: 8px; border: 1px solid #ddd;">Link</th>
-        </tr>
-    """
-
-    for hotel in hotels:
+    hotel_results = results.get("hotel_results", [])
+    
+    response_text = ""
+    for hotel in hotel_results[:5]:  # Limiting to top 5 hotels
         name = hotel.get("title", "No name")
+        address = hotel.get("address", "No address")
         rating = hotel.get("rating", "No rating")
-        reviews = hotel.get("reviews", "No reviews")
         price = hotel.get("price", "No price")
         link = hotel.get("link", "#")
-        location = f"{name}, Birmingham, AL"
 
-        # Add each hotel as a row in the table
-        table_html += f"""
-        <tr>
-            <td style="padding: 8px; border: 1px solid #ddd;">{name}</td>
-            <td style="padding: 8px; border: 1px solid #ddd;">{location}</td>
-            <td style="padding: 8px; border: 1px solid #ddd;">{rating}</td>
-            <td style="padding: 8px; border: 1px solid #ddd;">{reviews}</td>
-            <td style="padding: 8px; border: 1px solid #ddd;">{price}</td>
-            <td style="padding: 8px; border: 1px solid #ddd;"><a href="{link}" target="_blank">View</a></td>
-        </tr>
-        """
-
-    table_html += "</table>"
-
-    return table_html
+        response_text += f"[{name}]({link})\n"
+        response_text += f"*{address}*\n"
+        response_text += f"**Rating:** {rating} stars\n"
+        response_text += f"**Price:** {price}\n"
+        response_text += "-" * 50 + "\n"
 
+    return response_text
 
 
 
@@ -1090,7 +1069,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             events_output = gr.HTML(value=fetch_local_events())
             # restaurant_output=gr.HTML(value=fetch_yelp_restaurants())
         
-            hotel_output = gr.HTML(value=fetch_and_display_hotels())
+          
         
 
         with gr.Column():