Update app.py

app.py

@@ -337,19 +337,26 @@ def bot(history, choice, tts_choice, retrieval_mode):
 
 
 def format_restaurant_info(restaurant_info):
+    if not isinstance(restaurant_info, list):  # Check if it's not a list
+        print("Error: restaurant_info is not a list")  # Add a helpful error message
+        return "Error: Unable to retrieve restaurant information."
+
     formatted_info = []
     for restaurant in restaurant_info:
+        if not isinstance(restaurant, dict):  # Ensure each item is a dictionary
+            continue
         formatted_info.append(
-            f"**Name:** {restaurant['name']}\n"
-            f"**Rating:** {restaurant['rating']} stars\n"
-            f"**Reviews:** {restaurant['reviews']}\n"
-            f"**Phone:** {restaurant['phone']}\n"
-            f"**Snippet:** {restaurant['snippet']}\n"
-            f"**Services:** {restaurant['services']}\n"
-            f"**Yelp URL:** [Link]({restaurant['link']})\n"
+            f"**Name:** {restaurant.get('name', 'N/A')}\n"
+            f"**Rating:** {restaurant.get('rating', 'N/A')} stars\n"
+            f"**Reviews:** {restaurant.get('reviews', 'N/A')}\n"
+            f"**Phone:** {restaurant.get('phone', 'N/A')}\n"
+            f"**Snippet:** {restaurant.get('snippet', 'N/A')}\n"
+            f"**Services:** {restaurant.get('services', 'N/A')}\n"
+            f"**Yelp URL:** [Link]({restaurant.get('link', '#')})\n"
         )
     return "\n\n".join(formatted_info)
 
+
 def fetch_yelp_restaurants():
     params = {
         "engine": "yelp",
@@ -377,7 +384,8 @@ def fetch_yelp_restaurants():
     return yelp_data
 
 def process_restaurant_query(query):
-    restaurant_info = fetch_yelp_restaurants()  # No arguments passed now
+    restaurant_info = fetch_yelp_restaurants()
+    print("DEBUG: restaurant_info =", restaurant_info)  # Add this line for debugging
     formatted_info = format_restaurant_info(restaurant_info)
     return formatted_info
 
@@ -386,7 +394,6 @@ def process_restaurant_query(query):
 
 
 
-
 def add_message(history, message):
     history.append((message, None))
     return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)