Update app.py

app.py

@@ -15,10 +15,13 @@ def my_custom_tool(arg1: str, arg2: str) -> str:
         arg1: The input text to search.
         arg2: The letter to count (e.g., 'a').
     """
-    if len(arg2) != 1 or not arg2.isalpha():
-        return "Error: arg2 must be a single alphabet letter."
-    count = arg1.count(arg2)
-    return f"The letter '{arg2}' appears {count} time(s) in the text."
+    try:
+        if len(arg2) != 1 or not arg2.isalpha():
+            return "Error: arg2 must be a single alphabet letter."
+        count = arg1.count(arg2)
+        return f"The letter '{arg2}' appears {count} time(s) in the text."
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: