from MaMaGUI import MaMaGUI
from MaMaMa import MaMaMa  # ← use macro-capable class

if __name__ == "__main__":
    prog = {
        0: 'outer',
        1: 'stop'
    }

    # use MaMaMa instead of MaMa
    machine = MaMaMa(prog, [4, 6])

    inner_prog = [
        'push(1)',
        'push(2)',
        'add',
        'stop'
    ]

    outer_prog = [
        'push(10)',
        'inner',
        'mult',
        'stop'
    ]

    machine.add_macro("inner", inner_prog, [])
    machine.add_macro("outer", outer_prog, [])

    gui = MaMaGUI(machine)
    gui.display()