Update app.py

app.py

@@ -109,103 +109,104 @@ def get_weather_data(city: str):
     print(f"The warmest city is {warmest_city} with an average of {avg_temps[warmest_city]:.1f}°C")  
 
  
-
-
-
 @tool
-def hotel_recommendation_tool(city: str,price: str): str
-"""This is a hotel recommendation tool that web scrapes foursquare.com, and yelp.com for hotel ratings, user comments, and hotel information.
-        Args:
-             city: A string representing a valid city (e.g., 'Dubai, Singapore')
-        Args:
-            price: A string representing a valid price for the hotel rooms (e.g., '$160/night, $100/day')
-
-        Returns: A hotel name in a particular city with its pricing, and rating information
-"""
+def hotel_recommendation_tool(city: str, price: str) -> str:
+    """
+    This is a hotel recommendation tool that web scrapes foursquare.com and yelp.com for hotel ratings, user comments, and hotel information.
+    Args:
+        city: A string representing a valid city (e.g., 'Dubai, Singapore')
+        price: A string representing a valid price for the hotel rooms (e.g., '$160/night, $100/day')
+
+    Returns:
+        A hotel name in a particular city with its pricing and rating information
+    """
     
-if(price!="-"):
-        response=ur.request.urlopen('https://foursquare.com/explore?mode=url&near='+city+'&price='+price+'&q=Resturant')
-else:
-        response=ur.request.urlopen('https://foursquare.com/explore?mode=url&near='+city+'&q=Resturant')
-        html=response.read()
-        soup=BeautifulSoup(html,'html.parser')
-        name=[]
-        rate=[]
-        add=[]
-        y="""<li class="card singleRecommendation hasPhoto"""
-        for link in soup.find_all('li'):
-            if (y in str(link)):
-                name.append(link.find('div',{"class":"venueName"}))
-            if(link.find('div',{"class":"venueScore positive"})!=None):
-                rate.append(link.find('div',{"class":"venueScorepositive"}))
-            elif(link.find('div',{"class":"venueScoreneutral"})!=None):
-                rate.append(link.find('div',{"class":"venueScoreneutral"}))
-            else:
-                rate.append(link.find('div',{"class":"venueScoreunknown"}))
-                                                  
-        rate1=[]
-        for x in rate:
-            if(x!=None):
-                rate1.append(x.string.encode("utf-8"))
+    if price != "-":
+        response = ur.request.urlopen('https://foursquare.com/explore?mode=url&near=' + city + '&price=' + price + '&q=Resturant')
+    else:
+        response = ur.request.urlopen('https://foursquare.com/explore?mode=url&near=' + city + '&q=Resturant')
+        
+    html = response.read()
+    soup = BeautifulSoup(html, 'html.parser')
+    name = []
+    rate = []
+    add = []
+    y = """<li class="card singleRecommendation hasPhoto\""""
+    
+    for link in soup.find_all('li'):
+        if y in str(link):
+            name.append(link.find('div', {"class":"venueName"}))
+        if link.find('div', {"class":"venueScore positive"}) != None:
+            rate.append(link.find('div', {"class":"venueScorepositive"}))
+        elif link.find('div', {"class":"venueScore neutral"}) != None:
+            rate.append(link.find('div', {"class":"venueScoreneutral"}))
+        else:
+            rate.append(link.find('div', {"class":"venueScoreunknown"}))
+    
+    rate1 = []
+    
+    for x in rate:
+        if x != None:
+            rate1.append(x.string.encode("utf-8"))
         else:
             rate1.append("?")
-
-    ##print(rate1)
-
-        name1=[]
-        for x in name:
-            name1.append(x.a.string.encode("utf-8"))
-        
-            name_rate=[]
-        for i in range(0,len(name1)):
-            name_rate.append((name1[i],rate1[i]))
     
-return name_rate
+    name1 = []
+    for x in name:
+        name1.append(x.a.string.encode("utf-8"))
+        
+    name_rate = []
+    for i in range(0, len(name1)):
+        name_rate.append((name1[i], rate1[i]))
+        
+    return name_rate
 
 
-              
-def yelp(city,price):
-    name_rate=[]
-    count=0
-    while(count<30):
-        if(price!="-"):
-            response=ur.request.urlopen('https://www.yelp.com/search?find_desc=Restaurants&find_loc='+city+'&start='+str(count)+'&ed_attrs=RestaurantsPriceRange2.'+price)
+def yelp(city, price):
+    name_rate = []
+    count = 0
+    while (count < 30):
+        if price != "-":
+            response = ur.request.urlopen('https://www.yelp.com/search?find_desc=Restaurants&find_loc=' + city + '&start=' + str(count) + '&ed_attrs=RestaurantsPriceRange2.' + price)
         else:
-            response=ur.request.urlopen('https://www.yelp.com/search?find_desc=Restaurants&find_loc='+city+'&start='+str(count))
-            
-        html=response.read()
-        count+=10
-        soup=BeautifulSoup(html,'html.parser')
+            response = ur.request.urlopen('https://www.yelp.com/search?find_desc=Restaurants&find_loc=' + city + '&start=' + str(count))
+        
+        html = response.read()
+        count += 10
+        soup = BeautifulSoup(html, 'html.parser')
+        
+        y = """<li class="regular-search-result">"""
+        name = []
+        rate = []
+        name1 = []
+        rate1 = []
         
-        y="""<li class="regular-search-result">"""
-        name=[]
-        rate=[]
-        name1=[]
-        rate1=[]
         for link in soup.find_all('li'):
             if (y in str(link)):
-                name.append(link.find('span',{"class":"indexed-bizname"}))
-                rate.append(link.find('img',{"class":"offscreen"}))
+                name.append(link.find('span', {"class":"indexed-bizname"}))
+                rate.append(link.find('img', {"class":"offscreen"}))
+                
         for i in name:
             name1.append(i.span.string.encode("utf-8"))
             
         for i in rate:
             try:
-                s=str(i['alt'])
+                s = str(i['alt'])
                 rate1.append(float(s[0:3]))
             except:
                 rate1.append("?")
-        for i in range(0,len(name1)):
-            name_rate.append((name1[i],rate1[i]))
         
-    return name_rate
+        for i in range(0, len(name1)):
+            name_rate.append((name1[i], rate1[i]))
     
+    return name_rate
 
 
 root=tk.Tk()
 root.geometry("645x660")
 root.configure(bg='#211C5F')
 root.title("Scraping Recommender")
+
 def locate():
     send_url = 'http://freegeoip.net/json'
     r = requests.get(send_url)
@@ -217,7 +218,6 @@ def locate():
     e1.insert(tk.END,"Current Location")
     return city
 
-
 def get():
     city=e1.get()
     if(city=="Current Location"):
@@ -233,16 +233,14 @@ def get():
 
 @tool
 def SearchTool(DuckDuckGoSearchTool): query
-    
-"""
+    """
     name = "web_search"
     description = "Performs a web search for a query and returns a list of the top search results formatted as markdown with page titles and urls."
     inputs = {"query": {"type": "string", "description": "The search query to perform."}}
     output_type = "string"
-"""
-
-search_tool = DuckDuckGoSearchTool()
-agent = CodeAgent(tools=[search_tool], model=model, add_base_tools=False, verbosity_level=2)
-
+    """
+    
+    search_tool = DuckDuckGoSearchTool()
+    agent = CodeAgent(tools=[search_tool], model=model, add_base_tools=False, verbosity_level=2)
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()