Update app.py

app.py

@@ -9,13 +9,14 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def random_polyphonic_chars(_: str, __: str) -> str:
-    """Generates 10 random Chinese polyphonic characters (多音字).
+def random_polyphonic_chars(n_str: str, _: str) -> str:
+    """随机生成n个多音字.
     Args:
-        _: Unused argument.
-        __: Unused argument.
+        n_str: Number of polyphonic characters to generate (as string).
+        _: Unused second argument.
     """
     import random
+
     polyphonic_chars = [
         "行", "重", "长", "乐", "和",
         "难", "好", "觉", "着", "藏",
@@ -23,8 +24,17 @@ def random_polyphonic_chars(_: str, __: str) -> str:
         "干", "差", "空", "折", "弹"
     ]
 
-    selected = random.sample(polyphonic_chars, 10)
-    return "随机选出的10个多音字是：" + "、".join(selected)
+    try:
+        n = int(n_str)
+        if n <= 0:
+            return "请输入大于0的数字。"
+        if n > len(polyphonic_chars):
+            return f"最多只能生成 {len(polyphonic_chars)} 个不同的多音字。"
+        
+        selected = random.sample(polyphonic_chars, n)
+        return f"随机生成的 {n} 个多音字是：" + "、".join(selected)
+    except ValueError:
+        return "请输入有效的整数作为参数。"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: