Add array calculation interfaces with visualization to arithmetic module

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
 from utils.text_utils_interface import letter_counter_interface
-from maths.elementary.arithmetic_interface import add_interface, subtract_interface, multiply_interface, divide_interface
+from maths.elementary.arithmetic_interface import (
+    add_interface, subtract_interface, multiply_interface, divide_interface,
+    array_calc_interface, array_calc_vis_interface
+)
 from maths.middleschool.algebra_interface import solve_linear_equation_interface, evaluate_expression_interface
 from maths.highschool.trigonometry_interface import trig_interface
 from maths.university.calculus_interface import derivative_interface, integral_interface
 
 # Group interfaces by education level
 elementary_tab = gr.TabbedInterface(
-    [add_interface, subtract_interface, multiply_interface, divide_interface],
-    ["Addition", "Subtraction", "Multiplication", "Division"],
+    [add_interface, subtract_interface, multiply_interface, divide_interface, array_calc_interface, array_calc_vis_interface],
+    ["Addition", "Subtraction", "Multiplication", "Division", "Array Calculation", "Array Calculation with Visualization"],
     title="Elementary School Math"
 )
 

maths/elementary/arithmetic.py

@@ -129,3 +129,63 @@ def create_multiplication_table(number: int, max_multiplier: int = 12) -> plt.Fi
     ax.set_xticks(multipliers)
     
     return fig
+
+def calculate_array(numbers: list, operations: list) -> float:
+    """
+    Perform a sequence of arithmetic operations on an array of numbers.
+    Example: numbers=[2, 3, 4], operations=['add', 'multiply'] means (2 + 3) * 4
+    Supported operations: 'add', 'subtract', 'multiply', 'divide'
+    """
+    if not numbers or not operations:
+        raise ValueError("Both numbers and operations must be provided and non-empty.")
+    if len(operations) != len(numbers) - 1:
+        raise ValueError("Number of operations must be one less than number of numbers.")
+    
+    result = float(numbers[0])
+    for i, op in enumerate(operations):
+        num = float(numbers[i+1])
+        if op == 'add':
+            result = add(result, num)
+        elif op == 'subtract':
+            result = subtract(result, num)
+        elif op == 'multiply':
+            result = multiply(result, num)
+        elif op == 'divide':
+            result = divide(result, num)
+            if isinstance(result, str):  # Error case
+                return result
+        else:
+            raise ValueError(f"Unsupported operation: {op}")
+    return result
+
+
+def calculate_array_with_visualization(numbers: list, operations: list) -> tuple:
+    """
+    Perform a sequence of arithmetic operations on an array of numbers and return result with visualization.
+    Visualization shows the step-by-step calculation on a number line.
+    """
+    if not numbers or not operations:
+        return "Error: Both numbers and operations must be provided and non-empty.", None
+    if len(operations) != len(numbers) - 1:
+        return "Error: Number of operations must be one less than number of numbers.", None
+    
+    step_results = [float(numbers[0])]
+    result = float(numbers[0])
+    for i, op in enumerate(operations):
+        num = float(numbers[i+1])
+        if op == 'add':
+            result = add(result, num)
+        elif op == 'subtract':
+            result = subtract(result, num)
+        elif op == 'multiply':
+            result = multiply(result, num)
+        elif op == 'divide':
+            result = divide(result, num)
+            if isinstance(result, str):  # Error case
+                return result, None
+        else:
+            return f"Error: Unsupported operation: {op}", None
+        step_results.append(result)
+    # Visualization: show all intermediate results on the number line
+    fig = create_number_line_visualization(numbers, ' -> '.join(operations), result)
+    return result, fig

maths/elementary/arithmetic_interface.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from maths.elementary.arithmetic import add, subtract, multiply, divide
+from maths.elementary.arithmetic import add, subtract, multiply, divide, calculate_array, calculate_array_with_visualization
 
 # Elementary Math Tab
 add_interface = gr.Interface(
@@ -33,3 +33,41 @@ divide_interface = gr.Interface(
     title="Division",
     description="Divide two numbers"
 )
+
+# Advanced Array Calculation Interface
+
+def parse_number_list(s):
+    """Helper to parse comma-separated numbers from string input."""
+    try:
+        return [float(x.strip()) for x in s.split(',') if x.strip() != '']
+    except Exception:
+        return []
+
+def parse_operation_list(s):
+    """Helper to parse comma-separated operations from string input."""
+    ops = [x.strip().lower() for x in s.split(',') if x.strip() != '']
+    valid_ops = {'add', 'subtract', 'multiply', 'divide'}
+    return [op for op in ops if op in valid_ops]
+
+array_calc_interface = gr.Interface(
+    fn=lambda numbers, operations: calculate_array(numbers, operations),
+    inputs=[
+        gr.Textbox(label="Numbers (comma-separated, e.g. 2, 3, 4)"),
+        gr.Textbox(label="Operations (comma-separated, e.g. add, multiply)")
+    ],
+    outputs="text",
+    title="Array Calculation",
+    description="Calculate a sequence of numbers with specified operations. Example: Numbers: 2, 3, 4; Operations: add, multiply → (2 + 3) * 4"
+)
+
+array_calc_vis_interface = gr.Interface(
+    fn=lambda numbers, operations: calculate_array_with_visualization(
+        parse_number_list(numbers), parse_operation_list(operations)),
+    inputs=[
+        gr.Textbox(label="Numbers (comma-separated, e.g. 2, 3, 4)"),
+        gr.Textbox(label="Operations (comma-separated, e.g. add, multiply)")
+    ],
+    outputs=[gr.Textbox(label="Result"), gr.Plot(label="Visualization")],
+    title="Array Calculation with Visualization",
+    description="Calculate a sequence of numbers with specified operations and see a number line visualization."
+)