Update app.py

app.py

@@ -221,18 +221,24 @@ with st.sidebar:
 if selected_menu == "Upload Document":
     uploaded_file = st.file_uploader("Choose a file")        
     if uploaded_file is not None:
-        ocr_text  = get_text_from_ocr_engine(uploaded_file)
+        ocr_text  = get_text_from_ocr_engine()
         st.write("Upload Successful")
 elif selected_menu == "Extract Text":
     st.write(get_text_from_ocr_engine())
 elif selected_menu == "Summarize Document":
     paragraphs= get_paragraphs_for_summaries()
+    
     with st.spinner("Summarizing Document..."):
+        tags_found = ["Injury Details", "Past Medical Conditions", "Injury Management Plan", "GP Correspondence"]
+        st.write("This document is about:")
+        st.write(";".join(["#" + tag for tag in tags_found]))
+        st.markdown("""---""")
         for text in paragraphs:
             summary_text = pipeline_summarization(text, max_length=130, min_length=30, do_sample=False)
             # Show output
             st.write(summary_text[0]['summary_text'])
             st.markdown("""---""")
+     
         
 elif selected_menu == "Extract Entities":
     paragraphs= get_paragraphs_for_entities()
@@ -243,6 +249,7 @@ elif selected_menu == "Extract Entities":
             entities_text =get_formatted_text_for_annotation(output)
             annotated_text(*entities_text)
             st.markdown("""---""")
+    
 
 elif selected_menu == "Get Answers":
     st.subheader('Question')