bradley6597's picture
Update app.py
54922b6
raw
history blame
1.44 kB
import gradio as gr
import pandas as pd
dictionary = pd.read_csv("corncob_lowercase.txt",
header = None,
sep = ' ',
names = ['word'])
print(dictionary)
dictionary = dictionary.dropna()
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)]
final_words = list()
for i in range(0, spell_bee_solver.shape[0]):
words = spell_bee_solver['word'].iloc[i]
print(words)
words_set = set(words)
if len(words_set - no_centre_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_value('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()
solve_button.click(spell_bee_solver, inputs = [no_centre, centre], outputs = [output_df])
app.launch(debug = True, share = False)