Update app.py

app.py

@@ -351,8 +351,8 @@ def generate_answer(message, choice, retrieval_mode):
     elif "restaurant" in message.lower() and "birmingham" in message.lower():
         intro = "Here are the top Restaurants in Birmingham:\n\n"
         response = fetch_yelp_restaurants()
-    elif "flight" in message.lower() and "birmingham" in message.lower():
-        # intro = "Here are some available flights for today:\n\n"
+    elif "flight" in message.lower() and ("JFK" in message or "BHM" in message):
+        intro = "Here are some available flights for today:\n\n"
         response = fetch_google_flights()
     else:
         prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
@@ -375,14 +375,22 @@ def generate_answer(message, choice, retrieval_mode):
     # Assign numbers to each item in the response and format it
     response_lines = response.splitlines()
     formatted_response = ""
+    item_counter = 1
+
     for i in range(0, len(response_lines), 7):  # Assuming 7 lines per restaurant/hotel/flight entry
-        formatted_response += f"{i//7 + 1}. {response_lines[i]}\n"  # Number and name
-        formatted_response += f"   {response_lines[i + 1]}\n"  # Link and location
-        formatted_response += f"   {response_lines[i + 2]}\n"  # Contact number
-        formatted_response += f"   {response_lines[i + 3]}\n"  # Rating
-        formatted_response += f"   {response_lines[i + 4]}\n"  # Snippet
-        formatted_response += f"   {response_lines[i + 5]}\n"  # Divider
+        formatted_response += f"{item_counter}. {response_lines[i]}\n"  # Number and name
+        if i + 1 < len(response_lines):
+            formatted_response += f"   {response_lines[i + 1]}\n"  # Link and location
+        if i + 2 < len(response_lines):
+            formatted_response += f"   {response_lines[i + 2]}\n"  # Contact number
+        if i + 3 < len(response_lines):
+            formatted_response += f"   {response_lines[i + 3]}\n"  # Rating
+        if i + 4 < len(response_lines):
+            formatted_response += f"   {response_lines[i + 4]}\n"  # Snippet
+        if i + 5 < len(response_lines):
+            formatted_response += f"   {response_lines[i + 5]}\n"  # Divider
         formatted_response += "\n"  # Extra line between items
+        item_counter += 1
 
     return intro + formatted_response.strip(), extract_addresses(response)
 
@@ -395,6 +403,7 @@ def generate_answer(message, choice, retrieval_mode):
 
 
 
+
 # def bot(history, choice, tts_choice, retrieval_mode):
 #     if not history:
 #         return history