Update pages/3_Earnings_Semantic_Search_🔎_.py

pages/3_Earnings_Semantic_Search_🔎_.py

@@ -14,8 +14,8 @@ def gen_annotated_text(df):
                 
                 tag_list=[]
                 for row in df.itertuples():
-                    label = row[3]
-                    text = row[2]
+                    label = row[2]
+                    text = row[1]
                     if label == 'Positive':
                         tag_list.append((text,label,'#8fce00'))
                     elif label == 'Negative':
@@ -57,7 +57,6 @@ try:
                                     bi_enc_dict[sbert_model_name],
                                    emb_tokenizer,chain_type=chain_type)
 
-                print(result)
 
             references = [doc.page_content for doc in result['input_documents']]
 
@@ -67,7 +66,7 @@ try:
                 
             ##### Sematic Search #####
             
-            df = pd.DataFrame([(num,res,lab) for num, res, lab in zip(range(1),answer,sentiment_label)],columns=['Index','Text','Sentiment'])
+            df = pd.DataFrame.from_dict({'Text':[answer],'Sentiment':[sentiment_label]})
               
             
             text_annotations = gen_annotated_text(df)[0]