Upload tool

calculator.py

@@ -1,6 +1,40 @@
-import re
 from transformers import Tool
-from huggingface_hub import list_models
+
+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):
@@ -11,10 +45,6 @@ class SimpleCalculatorTool(Tool):
 
     inputs = ["text"]
     outputs = ["text"]
-    regex = r"(\d+(?:\.\d+)?)(\+|\-|\*|\/)(\d+(?:\.\d+)?)"
 
     def __call__(self, input_str: str):
-        match = re.match(self.regex, input_str)
-        if match is None:
-            raise ValueError("Input string is not a valid mathematical expression.")
-        return eval(f"{match.group(1)} {match.group(2)} {match.group(3)}")
+        return simple_calculator(input_str)

requirements.txt

@@ -1,3 +1 @@
 transformers
-re
-huggingface_hub