Update app.py

app.py

@@ -277,30 +277,110 @@ def generate_answer(message, choice, retrieval_mode):
     else:
         return "Invalid retrieval mode selected.", []
 
+# def bot(history, choice, tts_choice, retrieval_mode):
+#     if not history:
+#         return history
+    
+#     response, addresses = generate_answer(history[-1][0], choice, retrieval_mode)
+#     history[-1][1] = ""
+
+#     with concurrent.futures.ThreadPoolExecutor() as executor:
+#         if tts_choice == "Alpha":
+#             audio_future = executor.submit(generate_audio_elevenlabs, response)
+#         elif tts_choice == "Beta":
+#             audio_future = executor.submit(generate_audio_parler_tts, response)
+#         elif tts_choice == "Gamma":
+#             audio_future = executor.submit(generate_audio_mars5, response)
+
+#         for character in response:
+#             history[-1][1] += character
+#             time.sleep(0.05)
+#             yield history, None
+
+#         audio_path = audio_future.result()
+#         yield history, audio_path
+
+#     history.append([response, None])  # Ensure the response is added in the correct format
+
+
 def bot(history, choice, tts_choice, retrieval_mode):
     if not history:
         return history
     
-    response, addresses = generate_answer(history[-1][0], choice, retrieval_mode)
-    history[-1][1] = ""
+    user_message = history[-1][0].lower()
 
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        if tts_choice == "Alpha":
-            audio_future = executor.submit(generate_audio_elevenlabs, response)
-        elif tts_choice == "Beta":
-            audio_future = executor.submit(generate_audio_parler_tts, response)
-        elif tts_choice == "Gamma":
-            audio_future = executor.submit(generate_audio_mars5, response)
+    # Check if the query is related to restaurants
+    if "restaurant" in user_message:
+        response = process_restaurant_query(user_message)
+    else:
+        # Continue with the normal process if not a restaurant query
+        response, addresses = generate_answer(user_message, choice, retrieval_mode)
+        history[-1][1] = ""
+
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            if tts_choice == "Alpha":
+                audio_future = executor.submit(generate_audio_elevenlabs, response)
+            elif tts_choice == "Beta":
+                audio_future = executor.submit(generate_audio_parler_tts, response)
+            elif tts_choice == "Gamma":
+                audio_future = executor.submit(generate_audio_mars5, response)
+
+            for character in response:
+                history[-1][1] += character
+                time.sleep(0.05)
+                yield history, None
+
+            audio_path = audio_future.result()
+            yield history, audio_path
 
-        for character in response:
-            history[-1][1] += character
-            time.sleep(0.05)
-            yield history, None
+    history.append([response, None])  # Ensure the response is added in the correct format
 
-        audio_path = audio_future.result()
-        yield history, audio_path
 
-    history.append([response, None])  # Ensure the response is added in the correct format
+def fetch_yelp_restaurants(location="Birmingham, AL, USA", term="Restaurant"):
+    params = {
+        "engine": "yelp",
+        "find_desc": term,
+        "find_loc": location,
+        "api_key": os.getenv("SERP_API")
+    }
+
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    organic_results = results.get("organic_results", [])
+
+    yelp_data = []
+    for result in organic_results:
+        restaurant_info = {
+            "name": result.get("title", "No name"),
+            "rating": result.get("rating", "No rating"),
+            "reviews": result.get("reviews", "No reviews"),
+            "phone": result.get("phone", "Not Available"),
+            "snippet": result.get("snippet", "Not Available"),
+            "services": result.get("service_options", "Not Known"),
+            "link": result.get("link", "#")
+        }
+        yelp_data.append(restaurant_info)
+    return yelp_data
+
+def format_restaurant_info(restaurant_info):
+    formatted_info = []
+    for restaurant in restaurant_info:
+        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"
+        )
+    return "\n\n".join(formatted_info)
+
+def process_restaurant_query(query):
+    location = "Birmingham, AL, USA"  # Default location, can be extracted from query if needed
+    restaurant_info = fetch_yelp_restaurants(location=location, term=query)
+    formatted_info = format_restaurant_info(restaurant_info)
+    return formatted_info