Update app.py

app.py

@@ -316,6 +316,11 @@ def generate_answer(message, choice, retrieval_mode):
     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_google_hotels()
+        return response, extract_addresses(response)
 
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
@@ -339,6 +344,8 @@ def generate_answer(message, choice, retrieval_mode):
 
 
 
+
+
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
@@ -970,7 +977,6 @@ from serpapi.google_search import GoogleSearch
 from datetime import datetime, timedelta
 
 def fetch_google_hotels():
-   
     import os
 
     params = {