Update app.py

app.py

@@ -5,6 +5,8 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
+import os
+from urllib.parse import quote
 
 from Gradio_UI import GradioUI
 
@@ -34,6 +36,59 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def dicebear_pixel_avatar(seed: str,
+                          size: int = 256,
+                          background: str = "transparent",
+                          rounded: bool = True,
+                          fmt: str = "png") -> str:
+    """
+    Fetch a pixel-art avatar from the DiceBear API and save it to disk.
+    Args:
+        seed: String used to deterministically generate the avatar.
+        size: Image size in pixels (10–1024).
+        background: "transparent" or a hex color without the '#', e.g. "b6e3f4".
+        rounded: If True, applies radius=50 for rounded corners.
+        fmt: One of: "png", "svg", "jpg", "jpeg", "webp".
+    Returns:
+        Local file path to the saved image, or an error string.
+    """
+    try:
+        if fmt not in ("png", "svg", "jpg", "jpeg", "webp"):
+            fmt = "png"
+
+        # DiceBear v9 pixel-art endpoint
+        base = f"https://api.dicebear.com/9.x/pixel-art/{fmt}"
+        params = [f"seed={quote(seed)}", f"size={size}"]
+
+        # Background handling
+        if background and background.lower() != "transparent":
+            # solid background color (omit '#' in hex)
+            params.append(f"backgroundColor={quote(background)}")
+            params.append("backgroundType=solid")
+
+        # Rounded corners
+        if rounded:
+            params.append("radius=50")
+
+        url = base + "?" + "&".join(params)
+
+        resp = requests.get(url, timeout=20)
+        resp.raise_for_status()
+
+        out_dir = "outputs"
+        os.makedirs(out_dir, exist_ok=True)
+        safe_seed = "".join(c for c in seed if c.isalnum() or c in ("-", "_"))[:40] or "seed"
+        filename = os.path.join(out_dir, f"dicebear_pixel_{safe_seed}_{size}.{fmt}")
+
+        with open(filename, "wb") as f:
+            f.write(resp.content)
+
+        return filename
+    except Exception as e:
+        return f"Error generating avatar: {e}"
+
+
 search_web = DuckDuckGoSearchTool()
 visit_page = VisitWebpageTool()
 final_answer = FinalAnswerTool()
@@ -57,7 +112,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, search_web, visit_page], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, search_web, visit_page， dicebear_pixel_avatar], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,