Update app.py

app.py

@@ -1,11 +1,3 @@
-import subprocess
-import sys
-
-def install(package):
-    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
-
-install('crewai')
-
 import gradio as gr
 import requests
 import os
@@ -311,51 +303,99 @@ def bot(history, choice, tts_choice, retrieval_mode):
     history.append([response, None])  # Ensure the response is added in the correct format
 
 import os
-import gradio as gr
-from crewai import Agent
-from langchain.agents import Tool
-from langchain.utilities import GoogleSerperAPIWrapper
-
-# Setup API keys
-api_key=os.environ["SERP_API"] 
-
-# Initialize the Google Serper API Wrapper
-search = GoogleSerperAPIWrapper()
-
-# Define a tool specifically for restaurant searches in Birmingham
-def restaurant_search(query):
-    birmingham_query = f"{query} in Birmingham, AL"
-    return search.run(birmingham_query)
-
-# Create and assign the search tool to the agent
-serper_tool = Tool(
-  name="Restaurant Search",
-  func=restaurant_search,
-  description="Search for restaurants in Birmingham, AL",
-)
+from serpapi.google_search import GoogleSearch
 
-# Create an agent with the search tool
-restaurant_agent = Agent(
-  role='Restaurant Search Agent',
-  goal='Find and recommend the best restaurants in Birmingham, AL',
-  backstory='An expert on local eateries with a focus on Birmingham, AL.',
-  tools=[serper_tool]
-)
 
-# Function to handle restaurant-related queries
+def fetch_restaurant_details(location):
+    params = {
+        "engine": "yelp",
+        "find_desc": "Restaurant",
+        "find_loc": location,
+        "api_key": os.getenv("SERP_API")
+    }
+
+    search = GoogleSearch(params)
+    results = search.get_dict()
+    organic_results = results.get("organic_results", [])
+
+    yelp_html = """
+    <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Top Restaurants in {}</h2>
+    <style>
+        table {{
+            font-family: 'Verdana', sans-serif;
+            color: #333;
+            border-collapse: collapse;
+            width: 100%;
+        }}
+        th, td {{
+            border: 1px solid #fff !important;
+            padding: 8px;
+        }}
+        th {{
+            background-color: #f2f2f2;
+            color: #333;
+            text-align: left;
+        }}
+        tr:hover {{
+            background-color: #f5f5f5;
+        }}
+        .restaurant-link {{
+            color: #1E90FF;
+            text-decoration: none;
+        }}
+        .restaurant-link:hover {{
+            text-decoration: underline;
+        }}
+    </style>
+    <table>
+        <tr>
+            <th>Name</th>
+            <th>Rating</th>
+            <th>Reviews</th>
+            <th>Phone</th>
+            <th>Snippet</th>
+            <th>Services</th>
+        </tr>
+    """.format(location)
+
+    for result in organic_results:
+        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", "No Available")
+        services = result.get("service_options", "Not Known")
+
+        if isinstance(services, list):
+            services = ", ".join(services)
+        elif isinstance(services, dict):
+            services = ", ".join([f"{key}: {value}" for key, value in services.items()])
+
+        link = result.get("link", "#")
+
+        yelp_html += f"""
+        <tr>
+            <td><a class='restaurant-link' href='{link}' target='_blank'>{name}</a></td>
+            <td>{rating}</td>
+            <td>{reviews}</td>
+            <td>{phone}</td>
+            <td>{snippet}</td>
+            <td>{services}</td>
+        </tr>
+        """
+
+    yelp_html += "</table>"
+    return yelp_html
+
 def restaurant_search_agent(query):
-    if "restaurant" in query.lower() or "eatery" in query.lower():
-        return restaurant_agent.run(query)
+    if "restaurant" in query.lower():
+        location = "Birmingham, AL, USA"  # You can customize this based on user input or other logic
+        return fetch_restaurant_details(location)
     else:
         return "This agent only handles restaurant queries."
 
-# General response function for the chatbot
-def respond_to_query(query):
-    if "restaurant" in query.lower() or "eatery" in query.lower():
-        return restaurant_search_agent(query)
-    else:
-        # Handle other queries (existing logic)
-        return "Handling other queries here"
+
+
 
 
 def add_message(history, message):