Update app.py

app.py

@@ -1,22 +1,20 @@
 import gradio as gr
 import pandas as pd
 
-dictionary = pd.read_csv("corncob_lowercase.txt",
+english_dict = pd.read_csv("corncob_lowercase.txt",
                          header = None, 
                          sep = ' ', 
                         names = ['word'])
-print(dictionary)
-dictionary = dictionary.dropna()
+english_dict = english_dict.reset_index(drop = True)
+english_dict = english_dict.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)]
-
+    full_set = set(no_centre + centre)
+    spell_bee_solver = english_dict[english_dict['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:
+        if len(words_set - full_set) == 0:
             final_words.append(words)
 
     final_word_df = pd.DataFrame(final_words)