Update app.py

app.py

@@ -307,20 +307,10 @@ 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 flights
-    if "flight" in message.lower() and ("to" in message.lower() or "from" in message.lower()):
-        response = fetch_flight_info(message)
-        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_google_hotels()
-        return response, extract_addresses(response)
 
     prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
 
@@ -973,108 +963,7 @@ def fetch_yelp_restaurants():
     return response_text
 
 
-from serpapi.google_search import GoogleSearch
-from datetime import datetime, timedelta
-
-def fetch_google_hotels():
-    import os
-
-    params = {
-        "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
-    }
-
-    search = GoogleSearch(params)
-    results = search.get_dict()
-
-    # Debugging: Print the entire response to check the content
-    print("Google Hotels API Response:", results)
-
-    hotel_results = results.get("hotels_results", [])
-
-    if not hotel_results:
-        print("No hotel results found.")
-        return "No hotel information available."
-
-    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 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")
-        address = hotel.get("address", "Not Available")
-        price = hotel.get("price", "Not Available")
-        link = hotel.get("link", "#")
-
-        # 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"
-
-    # Debugging: Print the generated response text
-    print("Generated Hotel Response Text:", response_text)
-
-    return response_text
-
-
-def fetch_flight_info(message):
-    
-    departure_id = "JFK"  # Replace with parsed value from the message
-    arrival_id = "BHM"    # Replace with parsed value from the message
-    outbound_date = "2024-08-14"  # Replace with parsed date from the message
-    return_date = "2024-08-20"    # Replace with parsed date if applicable
-    
-    params = {
-        "engine": "google_flights",
-        "departure_id": departure_id,
-        "arrival_id": arrival_id,
-        "outbound_date": outbound_date,
-        "return_date": return_date,
-        "currency": "USD",
-        "hl": "en",
-        "api_key": os.getenv("SERP_API")
-    }
-
-    search = GoogleSearch(params)
-    results = search.get_dict()
-    flights_results = results.get("flights_results", [])
 
-    response_text = ""
-    
-    for flight in flights_results[:5]:  # Limiting to top 5 flight options
-        price = flight.get("price", "No price")
-        duration = flight.get("duration", "No duration")
-        stops = flight.get("stops", "Direct flight")
-        departure_time = flight.get("departure_time", "No departure time")
-        arrival_time = flight.get("arrival_time", "No arrival time")
-        airlines = flight.get("airlines", ["No airline"])
-
-        # Format the output for each flight
-        response_text += f"Flight: {', '.join(airlines)}\n"
-        response_text += f"Price: {price}\n"
-        response_text += f"Duration: {duration}\n"
-        response_text += f"Stops: {stops}\n"
-        response_text += f"Departure: {departure_time}\n"
-        response_text += f"Arrival: {arrival_time}\n"
-        response_text += "-" * 50 + "\n"
-
-    return response_text