AI_CSP

from itertools import permutations
from sys import argv

str1 = 'TWO'
str2 = 'TWO'
str3 = 'FOUR'


if '-i' in argv:
    str1 = input("Enter str1: ")
    str2 = input("Enter str2: ")
    str3 = input("Enter str3: ")

values = {}

def str_to_num(numstr):
    num = 0
    for c in numstr:
        num = num*10 + values[c]
    return num

for c in str1+str2+str3:
    values[c] = 0

symbols = set(values.keys())
k = len(symbols)

if k > 10:
    exit('Too many symbols')

for value_ins in permutations(range(10), r=k):
    for symbol, value in zip(symbols, value_ins):
        values[symbol] = value
    num1 = str_to_num(str1)
    num2 = str_to_num(str2)
    num3 = str_to_num(str3)
    if num3 == num1 + num2:
        print(f"{num1=}, {num2=}, {num3=}")
        if '-a' not in argv:
            break