bradley6597's picture
Changed naming and added column names
c0ab3ed
raw
history blame
1.52 kB
import gradio as gr
import pandas as pd
english_dict = pd.read_csv("dictionary.txt",
header = None,
sep = ' ',
names = ['word'])
english_dict = english_dict.reset_index(drop = True)
english_dict = english_dict.dropna()
def spell_bee_solver(no_centre, centre):
full_set = set(no_centre.lower() + centre.lower())
spell_bee_solver = english_dict[english_dict['word'].str.contains(str(centre.lower()), regex = False)]
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 - full_set) == 0:
final_words.append(words)
final_word_df = pd.DataFrame(final_words)
final_word_df.columns = ['word']
final_word_df['word_length'] = final_word_df['word'].str.len()
final_word_df = final_word_df[final_word_df['word_length'] > 3]
final_word_df = final_word_df.sort_values('word_length', ascending = False)
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(headers = ['word', 'word_length'])
solve_button.click(spell_bee_solver, inputs = [no_centre, centre], outputs = [output_df])
app.launch(debug = True, share = False)