Add more tools

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 
+# === TOOLS ===
 def letter_counter(word, letter):
     """
     Count the number of occurrences of a letter in a word or text.
@@ -16,12 +17,47 @@ def letter_counter(word, letter):
     count = word.count(letter)
     return count
 
-demo = gr.Interface(
-    fn=letter_counter,
-    inputs=[gr.Textbox("strawberry"), gr.Textbox("r")],
-    outputs=[gr.Number()],
-    title="Letter Counter",
-    description="Enter text and a letter to count how many times the letter appears in the text."
+def prime_factors(n):
+    """
+    Compute the prime factorization of a positive integer.
+    Args:
+        n (int): The integer to factorize. Must be greater than 1.
+    Returns:
+        List[int]: A list of prime factors in ascending order.
+    Raises:
+        ValueError: If n is not greater than 1.
+    """
+    n = int(n)
+    if n <= 1:
+        raise ValueError("Input must be an integer greater than 1.")
+
+    factors = []
+    while n % 2 == 0:
+        factors.append(2)
+        n //= 2
+
+    divisor = 3
+    while divisor * divisor <= n:
+        while n % divisor == 0:
+            factors.append(divisor)
+            n //= divisor
+        divisor += 2
+
+    if n > 1:
+        factors.append(n)
+
+    return factors
+
+# === GRADIO INTERFACE ===
+demo = gr.TabbedInterface(
+    [
+        gr.Interface(letter_counter, [gr.Textbox(), gr.Textbox()], gr.Textbox(), api_name="letter_counter"),
+        gr.Interface(prime_factors, gr.Number(), gr.JSON(), api_name="prime_factors"),
+    ],
+    [
+        "Letter Counter",
+        "Prime Factors"
+    ]
 )
 
 demo.launch(mcp_server=True)