Upload bloco_de_carnaval_info_finder.py

bloco_de_carnaval_info_finder.py

@@ -0,0 +1,132 @@
+from smolagents import CodeAgent, HfApiModel, DuckDuckGoSearchTool, VisitWebpageTool
+from bs4 import BeautifulSoup
+import pandas as pd
+def bloco_de_carnaval_info_finder_agent_maker(model):
+    search_web = DuckDuckGoSearchTool()
+    visit_web = VisitWebpageTool()
+
+    return CodeAgent(
+        tools=[search_web, visit_web],
+        additional_authorized_imports=['pandas', 'bs4', 'json'],
+        model=model,
+        add_base_tools=True
+        description="""This agent receives a São Paulo neighborhood and a date (yyyy-mm-dd) 
+            and return a JSON with the blocos de carnaval that will happen 
+            in that neighborhood on that date and informations about the 
+            blocos de carnaval. The agent uses the website 
+            https://www.blocosderua.com/programacao-blocos-de-carnaval-sp 
+            for that.
+
+            The webpage works as a REST API, so, to seach the day 2025-03-02 
+            in the "Capão Redondo" neighborhood, I enter in the 
+            "https://www.blocosderua.com/programacao-blocos-de-carnaval-sp?data=20250302&bairro=Capão+Redondo". 
+            This webpage contains various informations, such as a list of blocos de carnaval
+            and the links to their info. I need that the code enters in the information webpage
+            to get the information about the bloco de carnaval.
+            
+            The webpage works as a REST API, so, to seach the day 2025-03-02 
+            in the "Capão Redondo" neighborhood, it enters in the 
+            "https://www.blocosderua.com/programacao-blocos-de-carnaval-sp?data=20250302&bairro=Capão+Redondo". 
+            This webpage contains various informations, such as a list of blocos de carnaval
+            and the links to their info. The agent enter in each link to get the informations about the bloco de carnaval
+            and return a JSON with the informations about the blocos de carnaval in the following format:
+
+                {
+                    "blocos": [
+                        {
+                        "name": "Bloco 1",
+                        "time": "10:00",
+                        "location": "Rua 1",
+                        "description": "Description 1",
+                        "link": "Link 1",
+                        "price" "free"
+                        },
+                        {
+                        "name": "Bloco 2",
+                        "time": "11:00",
+                        "location": "Rua 2",
+                        "description": "Description 2",
+                        "link": "Link 2",
+                        "price" "paid"
+                        }
+                    ]
+                    }
+
+            The list of dates is as follows: 
+            02/Mar
+            03/Mar
+            04/Mar
+            06/Mar
+            07/Mar
+            08/Mar
+            09/Mar
+            16/Mar
+            26/Apr
+                        
+            The list of available neighborhoods is:
+            Aricanduva
+            Barra Funda
+            Bela Vista
+            Brasilândia
+            Brooklin
+            Butantã
+            Campo Limpo
+            Capão Redondo
+            Casa Verde
+            Centro
+            Cidade Ademar
+            Consolação
+            Ermelino Matarazzo
+            Freguesia do Ó
+            Guaianases
+            Guaianazes
+            Guarapiranga
+            Ibirapuera
+            Interlagos
+            Ipiranga
+            Itaim Paulista
+            Itaquera
+            Jabaquara
+            Jardim Maringa
+            Jardim Monte Azul
+            Jardim Ângela
+            Lapa
+            Lauzane Paulista
+            Limão
+            M Boi Mirim
+            M'Boi Mirim
+            Moema
+            Mooca
+            Morumbi
+            Parelheiros
+            Parque Ipe
+            Penha
+            Perdizes
+            Perus
+            Pinheiros
+            Pirituba
+            Pompeia
+            Santa Cecilia
+            Santa Cecília
+            Santana
+            Santo Amaro
+            Sapopemba
+            São Mateus
+            São Miguel Paulista
+            Tatuapé
+            Tremembé
+            Tucuruvi
+            Vila Carrão
+            Vila Cisper
+            Vila Guilherme
+            Vila Guilhermina
+            Vila Leopoldina
+            Vila Madalena
+            Vila Mariana
+            Vila Olimpia
+            Vila Olímpia
+            Vila Prudente
+            Vila Regente Feijó
+            Vila São José
+            Vila Virginia"""
+    )