Upload 7 files

myTools/ExtractWebContentWithSelenium.py

@@ -0,0 +1,42 @@
+from smolagents import Tool
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service as ChromeService
+from webdriver_manager.chrome import ChromeDriverManager
+from markdownify import markdownify as md
+
+class ExtractWebContentWithSelenium(Tool):
+    name = "extract_web_content_selenium"
+    description = "Visit a webpage and extract the full HTML content of a web page."
+
+    inputs = {
+        "url": {
+            "type": "string",
+            "description": "URL of the page to load"
+        }
+    }
+
+    output_type = "string"
+
+    def forward(self, url: str) -> str:
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument(
+            "user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+            "AppleWebKit/537.36 (KHTML, like Gecko) "
+            "Chrome/114.0.0.0 Safari/537.36"
+        )
+
+        # Installe automatiquement ChromeDriver
+        driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()), options=chrome_options)
+
+        try:
+            driver.get(url)
+            page_content = driver.page_source
+            markdown = md(page_content, heading_style="ATX")
+        finally:
+            driver.quit()
+
+        return markdown

myTools/ExtractWikipediaSection.py

@@ -0,0 +1,42 @@
+from smolagents import Tool
+import requests
+import re
+from markdownify import markdownify as md
+
+class ExtractWikipediaSection(Tool):
+    name = "extract_wikipedia_section"
+    description = "Extracts a specific section from a Wikipedia page in Markdown format."
+
+    inputs = {
+        "url": {
+            "type": "string",
+            "description": "URL of the Wikipedia page"
+        },
+        "section": {
+            "type": "string",
+            "description": "Title of the section to extract"
+        },
+    }
+
+    output_type = "string"
+
+    def forward(self, url: str, section: str) -> str:
+        headers = {
+            "User-Agent": "Mozilla/5.0 (compatible; WebScraper/1.0; +https://example.com/bot)"
+        }
+
+        try:
+            response = requests.get(url, headers=headers, timeout=10)
+            response.raise_for_status()
+        except Exception as e:
+            raise RuntimeError(f"Failed to fetch page: {e}")
+
+        markdown = md(response.text, heading_style="ATX")
+
+        # RegEx pour détecter la section markdown
+        pattern = rf"^##+\s*{re.escape(section)}\s*$(.*?)^##+"
+        match = re.search(pattern, markdown, re.DOTALL | re.MULTILINE)
+        if match:
+            return match.group(1).strip()
+        else:
+            return f"❌ Section '{section}' not found on page."

myTools/GetLogo.py

@@ -0,0 +1,52 @@
+from smolagents import Tool
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service as ChromeService
+from webdriver_manager.chrome import ChromeDriverManager
+from markdownify import markdownify as md
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+import random
+
+
+class GetLogo(Tool):
+    name = "get_logo"
+    description = "Lookup and search for a logo and generate the SVG code"
+
+    inputs = {
+
+    }
+
+    output_type = "string"
+
+    def forward(self) -> str:
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument(
+            "user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+            "AppleWebKit/537.36 (KHTML, like Gecko) "
+            "Chrome/114.0.0.0 Safari/537.36"
+        )
+
+        driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()), options=chrome_options)
+
+        try:
+            url = f"https://logoipsum.com/"
+            driver.get(url)
+            wait = WebDriverWait(driver, 10)
+            buttons = wait.until(
+                EC.presence_of_all_elements_located((By.CSS_SELECTOR, "div.artwork"))
+            )
+            if not buttons:
+                return "Aucun résultat trouvé."
+
+            random_logo = random.choice(buttons)
+            svg_element = random_logo.find_element(By.TAG_NAME, "svg")
+            svg_code = svg_element.get_attribute("outerHTML")
+
+            return svg_code
+        finally:
+            driver.quit()

myTools/GetPlaceholderImageTool.py

@@ -0,0 +1,18 @@
+from smolagents import Tool
+import os
+import requests
+
+
+class GetPlaceholderImageTool(Tool):
+    name = "get_placeholder_image_tool"
+    description = "Get placeholder image for website with specific size"
+
+    inputs = {
+        "height": {"type": "integer", "description": "The height of the placeholder"},
+        "width": {"type": "integer", "description": "The width of the placeholder"},
+    }
+
+    output_type = "string"
+
+    def forward(self, height: int, width: int) -> str:
+        return f"https://placehold.co/{width}x{height}"

myTools/GetSVG.py

@@ -0,0 +1,53 @@
+from smolagents import Tool
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service as ChromeService
+from webdriver_manager.chrome import ChromeDriverManager
+from markdownify import markdownify as md
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+
+class GetSVG(Tool):
+    name = "get_svg"
+    description = "Lookup and search for an icon and generate the SVG code"
+
+    inputs = {
+        "request": {
+            "type": "string",
+            "description": "The description of the svg from GetSVGList Tool"
+        }
+    }
+
+    output_type = "string"
+
+    def forward(self, request: str) -> str:
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument(
+            "user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+            "AppleWebKit/537.36 (KHTML, like Gecko) "
+            "Chrome/114.0.0.0 Safari/537.36"
+        )
+
+        driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()), options=chrome_options)
+
+        try:
+            url = f"https://phosphoricons.com/?q={request}"
+            driver.get(url)
+            wait = WebDriverWait(driver, 10)
+            buttons = wait.until(
+                EC.presence_of_all_elements_located((By.CSS_SELECTOR, "button.grid-item"))
+            )
+            if not buttons:
+                return "Aucun résultat trouvé."
+
+            first_icon = buttons[0]
+            svg_element = first_icon.find_element(By.TAG_NAME, "svg")
+            svg_code = svg_element.get_attribute("outerHTML")
+
+            return svg_code
+        finally:
+            driver.quit()

myTools/GetSVGList.py

@@ -0,0 +1,50 @@
+from smolagents import Tool
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service as ChromeService
+from webdriver_manager.chrome import ChromeDriverManager
+from markdownify import markdownify as md
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+
+class GetSVGList(Tool):
+    name = "get_svg_list"
+    description = "Get list of svg available to generate it with GetSVG Tool"
+
+    inputs = {
+
+    }
+
+    output_type = "array"
+
+    def forward(self) -> str:
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument(
+            "user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+            "AppleWebKit/537.36 (KHTML, like Gecko) "
+            "Chrome/114.0.0.0 Safari/537.36"
+        )
+
+        driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()), options=chrome_options)
+
+        try:
+            url = "https://phosphoricons.com/"
+            driver.get(url)
+
+            wait = WebDriverWait(driver, 10)
+            name_elements = wait.until(
+                EC.presence_of_all_elements_located((By.CSS_SELECTOR, "span.name"))
+            )
+
+            icon_names = [el.text.strip() for el in name_elements if el.text.strip()]
+            return icon_names
+            if not icon_names:
+                return "Aucun nom d'icône trouvé."
+
+            return "\n".join(icon_names)
+        finally:
+            driver.quit()