Update app.py

app.py

@@ -13,14 +13,28 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def compare_numbers(a:str, b:str)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
+    """
+    Compare two numeric values literally (float comparison).
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        a: First number (string).
+        b: Second number (string).
+    Returns:
+        A string stating which one is larger, or if they are equal.
     """
-    return "What magic will you build ?"
+    try:
+        num_a = float(a)
+        num_b = float(b)
+
+        if num_a > num_b:
+            return f"{a} is larger than {b}"
+        elif num_a < num_b:
+            return f"{b} is larger than {a}"
+        else:
+            return f"{a} and {b} are equal"
+    except Exception as e:
+        return f"Error comparing values '{a}' and '{b}': {e}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -77,9 +91,15 @@ def dicebear_pixel_avatar(seed: str,
         resp = requests.get(url, timeout=20)
         resp.raise_for_status()
 
-        # Encode image in base64
-        encoded = base64.b64encode(resp.content).decode("utf-8")
-        return f"data:image/{fmt};base64,{encoded}"
+        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}"
 
@@ -107,7 +127,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, dicebear_pixel_avatar], ## 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, compare_numbers], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,