Update app.py

app.py

@@ -215,7 +215,7 @@ Dr Jason Sanders"""
 
 with st.sidebar:
     selected_menu = option_menu("Select Option", 
-    ["Upload Document", "Extract Text", "Summarize Document", "Extract Entities","Get Answers"], 
+    ["Upload Document", "Extract Text", "Summarize Document", "Extract Entities","Detected Barriers","Get Answers"], 
         menu_icon="cast", default_index=0)
     
 
@@ -259,7 +259,23 @@ elif selected_menu == "Extract Entities":
             entities_text =get_formatted_text_for_annotation(output)
             annotated_text(*entities_text)
             st.markdown("""---""")
+            
+elif selected_menu == "Detected Barriers":
+    st.subheader('Barriers Detected')
+    barriers_to_detect = {"Chronic Pain":"Is the patint experiencing chronic pain?",
+                          ,"Mental Health Issues":"Does he have any mental issues?",
+                          "Prior History":"What is prior medical history?",
+                          "Smoking":"Does he smoke?",
+                          "Drinking":"Does he drink?",
+                          "Comorbidities":"Does he have any comorbidities?"]
+                          
+    for barrier,question_text from barriers_to_detect.items():
     
+        context = get_text_from_ocr_engine()
+        if question_text:
+            result = pipeline_qa(question=question_text,  context=context)
+            st.subheader(barrier)
+            st.text(result['answer']) 
 
 elif selected_menu == "Get Answers":
     st.subheader('Question')