Upload tool

app.py

@@ -0,0 +1,4 @@
+from transformers import launch_gradio_demo
+from calc import SimpleCalculatorTool
+
+launch_gradio_demo(SimpleCalculatorTool)

calc.py

@@ -0,0 +1,51 @@
+from transformers import Tool
+
+def simple_calculator(expression):
+    # Find the index of the operator
+    operator_index = -1
+    for i, char in enumerate(expression):
+        if char in ('+', '-', '*', '/'):
+            operator_index = i
+            break
+
+    # Ensure that an operator was found
+    if operator_index == -1:
+        return "Invalid expression"
+
+    # Split the expression into the operator and operands
+    operator = expression[operator_index]
+    operand1 = float(expression[:operator_index])
+    operand2 = float(expression[operator_index + 1:])
+
+    # Perform the calculation based on the operator
+    if operator == '+':
+        result = operand1 + operand2
+    elif operator == '-':
+        result = operand1 - operand2
+    elif operator == '*':
+        result = operand1 * operand2
+    elif operator == '/':
+        result = operand1 / operand2
+    else:
+        raise ValueError("Input string is not a valid mathematical expression.")
+
+    # Convert the result to a string and return it
+    return str(result)
+
+# Test the function
+expression = "1+3"
+result = simple_calculator(expression)
+
+
+class SimpleCalculatorTool(Tool):
+    name = "simple_calculator"
+    description = (
+        "This is a tool that returns the result of a simple mathematical expression written in a string."
+    )
+
+    inputs = ["text"]
+    outputs = ["text"]
+    output_type = "text"
+
+    def __call__(self, input_str: str):
+        return simple_calculator(input_str)

requirements.txt

@@ -0,0 +1 @@
+transformers

tool_config.json

@@ -0,0 +1,9 @@
+{
+  "description": "This is a tool that returns the result of a simple mathematical expression written in a string.",
+  "inputs": [
+    "text"
+  ],
+  "name": "simple_calculator",
+  "output_type": "text",
+  "tool_class": "calc.SimpleCalculatorTool"
+}