#!/usr/bin/env python3

import pandas as pd
import os

def automat_fun(recursive_df: pd.DataFrame):
    first_row = recursive_df.iloc[0]
    recursive_df = recursive_df.iloc[1:,:]
    return lambda char, state: first_row[1] if state == first_row[0] and char == first_row[2] else automat_fun(recursive_df)(char, state)

def create_automat_from_file(file_path: str):
    df = pd.read_csv(file_path, sep=' ', header=None)
    accept_state = df.tail(1).iloc[0, 0]
    df = df.iloc[:-1,:]
    return accept_state, automat_fun(df)

file_path = os.path.join(os.path.dirname(__file__), 'fsa_description.arg')
accept_state, automat_function = create_automat_from_file(file_path)

txt = input('Write binary sequence: ')

while txt != 'exit':
    state = 0
    for c in txt:
        if (c == '\n'):
            break
        state = automat_function(int(c), state)
    print('YES' if state == accept_state else "NO")
    txt = input('Write binary sequence: ')