add reader

appStore/reader.py

@@ -37,9 +37,9 @@ def app():
             else:
                 warning_msg = ""
             
-            reader_highlight(haystack_doc=df,
+            df = reader_highlight(haystack_doc=df,
                                         threshold= params['threshold'])
-            # st.session_state.key1 = df
+            st.session_state.key1 = df
 
   
                

utils/reader_qa.py

@@ -82,10 +82,17 @@ def reader_highlight(haystack_doc:pd.DataFrame,
         for text in list(df1.text):
             predictions.append(reader_model(question = ques_, context = text))
 
-    st.write(predictions)
     # # getting the sector label and scores
-    # list_ = []
-    # for i in range(len(predictions)):
+    list_ = []
+    for i in range(len(predictions)):
+      list_.append(predictions['answer'])
+    df1['Extracted Text'] = list_
+
+    df = pd.concat([df,df1])
+    df = df.reset_index(drop =True)
+    df.index += 1
+
+    return df
 
     #   temp = predictions[i]
     #   placeholder = {}