WordleTry.py

import numpy as np

# Get the solution to the puzzle
def get_solution():
    with open("solutions.txt", "r") as file:
        answers = file.read()
        words = answers.split()
    return np.random.choice(words)

# ask user for input
def ask_for_guess():
    guess = input('Enter your guess:')
    with open("valid-wordle-words.txt", 'r') as file:
        valid_list = file.read()
        valid_words = valid_list.split()
    if guess not in valid_words:
        return False
    return guess

def compare_input_to_solution(guess, solution):
    guess_letters = ([*guess])
    solution_letters = ([*solution])
    response = [0, 0, 0, 0, 0]
    for i in range(len(guess_letters)):
        if guess_letters[i] == solution_letters[i]:
            response[i] = 1
        #TODO: Handle double letters
        elif guess_letters[i] in solution_letters:
            response[i] = 0.5
    return response

def check_for_win(guess, solution):
    if guess == solution:
        return True
    else:
        return False


def main_game():
    solution = get_solution()
    Win = False
    guess_number = 0
    while guess_number < 6:
        guess = ask_for_guess()
        if guess == False:
            print("Invalid Guess")
            continue
        guess_number = guess_number + 1
        response = compare_input_to_solution(guess, solution)
        print(response)
        #TODO: print remaining letters
        if check_for_win(guess, solution):
            Win = True
            print("You Win!")
            break
    if Win is False:
        print("You Lose. The solution was " + solution)
    print("The Game has ended")

main_game()