feat: add qr code generator

app.py

@@ -1,6 +1,6 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
-import requests
+import qrcode
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
@@ -10,18 +10,27 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def word_count(text: str) -> str:
+def get_word_count(text: str) -> str:
     """A tool that analyzes text and provide word, character, and sentence counts.
     Args:
         text: The text to analyze
     """
     words = len(re.findall(r"\b\w+\b", text))
-    chars = len(text)
-    chars_no_spaces = len(text.replace(" ", ""))
-    sentences = len(re.split(r"[.!?]+", text)) - 1
-    print(
-        f"Words: {words}, Characters: {chars}, Characters (no spaces): {chars_no_spaces}, Sentences: {sentences}"
-    )
+    return words
+
+
+@tool
+def get_qr_code(url: str) -> str:
+    """A tool that generates a QR code for a given URL.
+    Args:
+        url: The URL to encode in the QR code.
+    """
+    qr = qrcode.QRCode(version=1, box_size=10, border=5)
+    qr.add_data(url)
+    qr.make(fit=True)
+    img = qr.make_image(fill_color="black", back_color="white")
+
+    return img.get_image()
 
 
 @tool
@@ -64,7 +73,8 @@ agent = CodeAgent(
     tools=[
         final_answer,
         get_current_time_in_timezone,
-        word_count,
+        get_word_count,
+        get_qr_code,
         DuckDuckGoSearchTool(),
     ],
     max_steps=6,