Update pages/3_Earnings_Semantic_Search_🔎_.py

pages/3_Earnings_Semantic_Search_🔎_.py

@@ -52,10 +52,10 @@ if search_input is not None:
         
         print(sen_df)
         
-        def gen_annotated_text(para):
+        def gen_annotated_text(para,df=sen_df):
             tag_list = []
             for i in sent_tokenize(para):
-                label = sen_df.loc[sen_df['text']==i, 'label'].values[0]
+                label = df.loc[df['text']==i, 'label'].values[0]
                 if label == 'Negative':
                     tag_list.append((i,label,'#faa'))
                 elif label == 'Positive':
@@ -64,7 +64,7 @@ if search_input is not None:
                     tag_list.append((i,label,'#fea'))
             return tag_list  
         
-        text_to_annotate = [gen_annotated_text(para) for para in df.Text.tolist()]
+        text_to_annotate = [gen_annotated_text(para,sen_df) for para in df.Text.tolist()]
         
         first,second = text_to_annotate[0],text_to_annotate[-1]