Update app.py

app.py

@@ -59,13 +59,21 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
     
     #@st.cache
     def Query2id(voc, query):
-        return [voc.index[voc['word'] == word].values[0] for word in query.split()]    
+        res= [] 
+        for word in query.split():
+            try:
+                res.append(voc.index[voc['word'] == word].values[0])
+            except KeyError:
+                st.write('Token Unfound')
+                continue
+        return res
+        #return [voc.index[voc['word'] == word].values[0] for word in query.split()]
+        #return [voc.index[voc['word'] == word].values[0] for word in query.split()]    
     
     id_list = Query2id(df3, sent)
     #@st.cache
     def setQueriesVoc(df, id_list):
         res = []
-        
         for e in id_list:
             try:
                 res.extend(list(df.loc[e]['ID2'].values)) 
@@ -91,9 +99,6 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
         return tmt
     tmt = compute_score(L, id_list)    
     exp_terms = []
-    for e in sent:
-        if (e in tmt):
-            del tmt[e]
     t_li = tmt.values()
     tmexp = sorted(tmt.items(), key=lambda x: x[1], reverse=True)
     i = 0