Update app.py

app.py

@@ -278,7 +278,7 @@ def generate_answer(message, choice, retrieval_mode):
         response = fetch_yelp_restaurants()
         return response, extract_addresses(response)
     # Check if the question is about flights
-    if "flight" in message.lower() and "birmingham" in message.lower():
+    if "flight" in message.lower() or "flights" in message.lower() and "birmingham" in message.lower():
         response = fetch_google_flights()
         return response, extract_addresses(response)
 
@@ -353,7 +353,13 @@ def extract_addresses(response):
         r'([A-Z].*\s\d{2},\sBirmingham,\sAL\s\d{5})',
         r'([a-zA-Z]\s Birmingham)',
         r'([a-zA-Z].*,\sBirmingham,\sAL)',
-        r'(^Birmingham,AL$)'
+        r'(^Birmingham,AL$)',
+        r'((.*)(Stadium|Field),.*,AL$)',
+        r'((.*)(Stadium|Field),.*,FL$)',
+        r'((.*)(Stadium|Field),.*,MS$)',
+        r'((.*)(Stadium|Field),.*,AR$)',
+        r'((.*)(Stadium|Field),.*,KY$)'
+        
     ]
     addresses = []
     for pattern in address_patterns: