bot.py

import math
from words import *

words_list = WORDS
init_words_list = words_list.copy()

freq = { chr(letter+65): [0]*5 for letter in range(26) }
for word in words_list:
    for i in range(5):
        freq[word[i]][i] += 1
freq1 = freq.copy()

#functiile pentru joc

def pattern(pickedWord, guessWord):
    res = ['0'] * 5
    for i in range(5):
        if guessWord[i] == pickedWord[i]:
            res[i] = '2'
            #pickedWord = pickedWord[:i] + "#" + pickedWord[i+1:]
            #guessWord = guessWord[:i] + "@" + guessWord[i+1:]
        else:
            res[i] = '!'
    for i in range(5):
        if res[i] == '2':
            continue
        ind = pickedWord.find(guessWord[i])
        if ind != -1:
            res[i] = '1'
            #pickedWord = pickedWord[:ind] + "#" + pickedWord[ind+1:]
        else:
            res[i] = '0'
    return "".join(res)


def filter_words_pattern(guessWord, givenPattern):
    aux = []
    aux.clear()
    for word in words_list:
        word1 = (word+'.')[:-1]
        guessWord1 = (guessWord+'.')[:-1]
        ok = 0
        for i in range(5):
            if givenPattern[i] == "2":
                if word1[i] != guessWord1[i]:
                    ok = 1
                    break
                #word1 = word1[:i] + "#" + word1[i+1:]
                #word1 = word1[:i] + "@" + word1[i + 1:]
        if ok == 1:
            for i in range(5):
                freq[word[i]][i] -= 1
            continue
        for i in range(5):
            if givenPattern[i] == "1":
                if word1.find(guessWord1[i]) == -1 or word1[i] == guessWord1[i]:
                    ok = 1
                    break
                #else:
                    #ind = word1.find(guessWord1[i])
                    #word1 = word1[:ind] + "#" + word1[ind + 1:]

            if givenPattern[i] == "0":
                if word1.find(guessWord1[i]) != -1:
                    ok = 1
                    break
                #else:
                    #ind = word1.find(guessWord1[i])
                    #word1 = word1[:ind] + "#" + word1[ind + 1:]

        if ok == 0:
            aux.append(word)
        else:
            for i in range(5):
                freq[word[i]][i] -= 1
    words_list.clear()
    for i in range(len(aux)):
        words_list.append(aux[i])



#acum trb actualizat freq la fiecare pas
def entropy(guessWord):
    res = 0
    for i in range(5):
        letter = guessWord[i]
        green = freq[letter][i]
        yellow = 0
        for j in range(5):
            if i != j:
                yellow += freq[letter][j]
        grey = len(words_list) - green - yellow
        if green == 0 and grey == 0 and yellow == 0:
            print (guessWord)
        event_green = green / (green + yellow + grey)
        event_yellow = yellow / (green + yellow + grey)
        event_grey = grey / (green + yellow + grey)
        if green > 0:
            res -= event_green * math.log2(event_green)
        if yellow > 0:
            res -= event_yellow * math.log2(event_yellow)
        if grey > 0:
            res -= event_grey * math.log2(event_grey)
    return res

def chooseWord(guessQueue, patternQueue):
#picked_word = "VATRA"#words_list[random.randint(0, dim-1)]
#print (picked_word)
    words_list = WORDS
    init_words_list = words_list.copy()
    freq = { chr(letter+65): [0]*5 for letter in range(26) }
    for word in words_list:
        for i in range(5):
            freq[word[i]][i] += 1

    freq1 = freq.copy()

    dim = 11454

    patterns = []
    v = [0, 0, 0, 0, 0, 0]
    while v[0] == 0:
        var = ""
        for i in range(1, 6):
            var += str(v[i])
        patterns.append(var)
        i = 5
        while i > 0 and v[i] == 2:
            v[i] = 0
            i -= 1
        v[i] += 1
    guessQueue.put("TAREI")
    dict1 = {word: 0 for word in init_words_list}
    dict1["TAREI"] = 1
    used = [0 for letter in range(26)]
    sol = "TAREI"
    all_pattern = "00000"
    while True:
        if patternQueue.empty() == 0 :
            maxim = -1
            nr = 0
            current_pattern = patternQueue.get()
            filter_words_pattern(sol, current_pattern)
            guesses = 0
            if guesses < 6 and len(words_list) > 2:
                for i in range(5):
                    if current_pattern[i] == "2":
                        all_pattern = all_pattern[:i] + "2" + all_pattern[i + 1:]
                for i in range(5):
                    if all_pattern[i] == "2":
                        nr += 1
                        used[ord(sol[i]) - 65] = 1
                if nr < 3:
                    for word in init_words_list:
                        if dict1[word] == 0:
                            val = entropy(word)
                            if val > maxim:
                                maxim = val
                                sol = word
                else:
                    for word in init_words_list:
                        ok = 0
                        for i in range(5):
                            if used[ord(word[i]) - 65]:
                                ok += 1
                        if ok >= 3:
                            continue
                        if dict1[word] == 0:
                            val = entropy(word)
                            if val > maxim:
                                maxim = val
                                sol = word
            else:
                for word in words_list:
                    ok = 0
                    if dict1[word] == 0:
                        val = entropy(word)
                        if val > maxim:
                            maxim = val
                            sol = word
            if len(words_list) == 0:
                break
            if len(words_list) == 1:
                break
            dict1[sol] = 1
            guessQueue.put(sol)
            guesses = guesses + 1
    if len(words_list) == 1:
        guessQueue.put(words_list[0])


#picked_word = words_list[random.randint(0, dim-1)]
#picked_word = "SUTAI"
#picked_word = "NAZUL"
#picked_word = "SOSIT"
#picked_word = "IUBIM"
#piked_word = "VOTUL"

'''
LODBA
TAREI
CULCA
GLOSA
LOBDA
LODBA
'''
#print (picked_word)

#print (entropy("TARIE"))
#print (entropy("TAREI"))
#acum da tarei cu entropie maxima

'''
sol_word = ""
maxim = -10
for word in words_list:
    okc = 0
    for i in range(5):
        for j in range(5):
            if i != j and word[i] == word[j]:
                okc = 1
    if okc:
        continue
    val = entropy(word)
    if val > maxim:
        maxim = val
        sol_word = word
print (sol_word)
#tarei!! pe cuv cu litere dif
'''



#pattern1 = pattern(picked_word, "TAREI")
#filter_words_pattern("TAREI", pattern1)
#print ("TAREI", end="\n")
#print (sum(freq["E"]))
#print(freq["E"])
#print (len(words_list))
#print(pattern1)

'''sol = "@@@@@"

dict1 = {word: 0 for word in init_words_list}
dict1["TAREI"] = 1

used = [0 for letter in range(26)]
while pattern(picked_word, sol) != "22222" and len(words_list) > 1:
    maxim = -1
    if len(words_list) > 5:
        for word in words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    else:
        freq = freq1.copy()
        for word in init_words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    #pattern1 = pattern(picked_word, sol)
    sol = "@@@@@"
    #nr = 0
    #for i in range(5):
        #if pattern1[i] == "2":
            #nr += 1
    if nr < 3:
        for word in words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    else:
        freq = freq1
        for word in init_words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    for word in words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    #lista trb lasata nemodif pentru a gasi cel mai bun cuv
    #pentru lista ramasa de cuv
 for word in init_words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word


 prev_pattern = pattern(picked_word, sol)
    nr = 0

    for i in range(5):
        if prev_pattern[i] == "2":
            nr += 1
            used[ord(sol[i])-65] = 1
    if nr < 3:
        for word in init_words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
    else:
        for word in init_words_list:
            ok = 0
            for i in range(5):
                if used[ord(word[i])-65]:
                    ok += 1
            if ok >= 3:
                continue
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word

    if len(words_list) <= 1:
        break
    #print(sol)
    dict1[sol] = 1
    current_pattern = pattern(picked_word, sol)
    filter_words_pattern(sol, current_pattern)

#if len(words_list) == 1 and sol != words_list[0]:
    #print (words_list[0])'''

'''
VOTUL
TAREI
CUCUL
OOGON
SMASH
BIBIL
FIFUL
PAPAL
AVIVA
VOTUL
nemodif



'''

'''
SUTAI
TAREI
OUATI
CUTAI
MUTAI
SUTAI
cu lista modif

SUTAI
TAREI
OUTUL
ACCES
SUTAI
cu lista nemodif

SUTAI
TAREI
OUATI
CUTAI
MAMOS
SUTAI
hibrid 2
'''

'''
NAZUL
TAREI
SULUL
CACIC
NINGA
BAZZZ
NAZUL
cu lista nemodif

NAZUL
TAREI
SALUL
NANUL
NAPUL
NAZUL
cu lista modif

NAZUL
TAREI
SALUL
NANUL
BAZZZ
NAZUL
hibrid
'''

'''
SOSIT
TAREI
OOCIT
BOBIT
DODIT
MOMIT
GOGIT
POPIT
SOLIT
SOSIT
lista modif

SOSIT
TAREI
COCUL
MOMIM
PODIS
SOSIT
lista nemodif

SOSIT
TAREI
OOCIT
SPLIN
SOSIT
hibrid

'''

'''
IUBIM
TAREI
LINUL
SUDIM
GOGIM
JUPIM
IUBIM
hibrid 2

IUBIM
TAREI
MOLUL
SUDIC
INFIG
IUBIM
lista nemodif

IUBIM
TAREI
LINUL
SUDIM
FUGIM
JUPIM
IUBIM
lista modif


'''


'''
avg = 0

for word in init_words_list:
    freq = freq1.copy()
    words_list = init_words_list.copy()
    picked_word = word
    pattern1 = pattern(picked_word, "TAREI")
    filter_words_pattern("TAREI", pattern1)
    #print("TAREI", end="\n")

    sol = "@@@@@"
    nr = 1
    dict1 = {word: 0 for word in init_words_list}
    dict1["TAREI"] = 1

    while pattern(picked_word, sol) != "22222" and len(words_list) > 1:
        maxim = -1
        sol = "@@@@@"
        for word in words_list:
            if dict1[word] == 0:
                val = entropy(word)
                if val > maxim:
                    maxim = val
                    sol = word
        if len(words_list) <= 1:
            break
        #print(sol)
        nr += 1
        dict1[sol] = 1
        current_pattern = pattern(picked_word, sol)
        filter_words_pattern(sol, current_pattern)
        if nr > 9:
            print("bucla inf")
            break
    if len(words_list) == 1 and sol != words_list[0]:
        nr += 1
        #print(words_list[0])
    avg += nr
avg /= len(init_words_list)
print ("average ul este:")
print(avg)

'''


#snaps, hobai, colai, angli - dureaza mult !!update snaps ok, hobai 6 try-uri, colai merge prost probabil mai trebuie calibrat programul pentru trecerea la entropia totala, angli 5 incercari
#votul, totul - dureaza mult, sunt multe cuvinte articulate si trebuie gasite literele prin metoda cealalta
#nopti - dureaza mult pe varianta cu entropie toatala !!update rezolvat dupa calibrare
#brahe - multe incercari !!update: 5 incercari dupa implementare hibrida
#vidai - 7 try-uri vs 8 in varianta initiala
#pare ca atunci cand se duce la pasul 2 pe CALII devine naspa