Update app.py

app.py

@@ -33,13 +33,15 @@ def next_word(text, pipe):
             res_dict['الكلمة المقترحة'].append(e['token_str'])
             res_dict['العلامة'].append(e['score'])
     return res_dict
+
 if (st.button('بحث', disabled=False)):
     text_st = sent+ ' <mask>'
     dict_next_words = next_word(text_st, pipe)
     df = pd.DataFrame.from_dict(dict_next_words)
-    df.reset_index(drop=True, inplace=True)
-    st.dataframe(df)
+    new_df = df[~df['الكلمة المقترحة'].isin(sent)]
+    st.dataframe(new_df)
 #using Graph   
+
 if (st.checkbox('الاستعانة بالرسم البياني المعرفي الاحتمالي', value=False)):
     a = time()
     VocMap = './voc.csv'
@@ -104,7 +106,9 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
         i+=1
         if (i==10):
             break
-    
+    for e in sent:
+        if (e in tmt):
+            del tmt[e]
     res_df = pd.DataFrame.from_dict(dict_res)
     res_df.index += 1
     
@@ -112,13 +116,15 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
     exec_time = (b-a)
     text_st = sent+ ' <mask>'
     dict_next_words = next_word(text_st, pipe)
+    
     df = pd.DataFrame.from_dict(dict_next_words)
-    df.index += 1
+    new_df = df[~df['الكلمة المقترحة'].isin(sent)]
+    new_df.index += 1
     str_time = str(exec_time)[:3]
     
     st.markdown("""---""")
     st.header("الكلمات المقترحة باستعمال النموذج اللغوي")
-    st.dataframe(df)
+    st.dataframe(new_df)
     st.markdown("""---""")
     st.header("الكلمات المقترحة باستعمال الرسم البياني")
     st.dataframe(res_df)