import gradio as gr
import pandas as pd

dictionary = pd.read_csv("corncob_lowercase.txt",
                         header = None, 
                        names = ['word'])
print(dictionary)
def spell_bee_solver(no_centre, centre):
    no_centre_set = set(no_centre)
    # spell_bee_solver = dictionary[dictionary['word'].str.contains(str(centre), regex = False)]
    spell_bee_solver = dictionary
    final_words = list()
    for i in range(0, spell_bee_solver.shape[0]):
        words = spell_bee_solver['word'].iloc[i]
        words_set = set(words)
        if len(words_set - no_centre_set) == 0:
            final_words.append(words)

    final_word_df = pd.DataFrame(final_words)
    return(final_word_df)

with gr.Blocks() as app:
    with gr.Row():
        no_centre = gr.Textbox(label = 'Letters Outside of Centre')
        centre = gr.Textbox(label = 'Centre Letter')
    with gr.Row():
        solve_button = gr.Button(value = 'Solve')
    with gr.Row():
        output_df = gr.DataFrame()
    solve_button.click(spell_bee_solver, inputs = [no_centre, centre], outputs = [output_df])

app.launch(debug = True, share = False)