Update app.py

app.py

@@ -86,6 +86,84 @@ pipeline_qa = init_qa_pipeline()
 
 st.header("Intelligent Document Automation")
 
+def get_paragraphs_for_summaries():
+    paras =[]
+    paras.append("""Stephen is a 53 year old gentleman who does general duties police work. He is married and has an
+18 year old daughter at home. He is right hand dominant. Cigarettes nil, alcohol rare, allergies nil.
+DVTe nil. Past medical history hyperlipidemia and reflux testicular cancer in 2000 and right knee
+reconstruction in 1987. Medications include Nexium and Crestor.""")
+    paras.append("""History presenting complaint: Right knee and right elbow injuries. On 12 January 2020, while at work
+he was trying to apprehend a stolen vehicle. The deployed some road spikes onto the road. The
+stolen vehicle went over the road spikes. He was attempting to retrieve the spikes to prevent
+damage to civilian and police vehicles and while he was doing that, he was hit by a police vehicle
+coming from behind. The police vehicle was approximately doing 50 km/hr. The headlight of the
+police car stuck him on the lateral aspect of the right knee. He jumped in the air and flicked in the air
+with his right elbow also hitting the police car before he flung over a barrier. He was able to mobilise
+afterwards when the adrenaline was taking effect. After he cooled down that night he developed
+significant pain in both the elbow and the knee. The elbow seemed to have settled with time but he
+has got slight discomfort on the lateral epicondyle of the right elbow but otherwise no instability,
+clicking, locking or catching of the elbow.""")
+    paras.append("""His knee has medial sided pain. It is an annoying type of pain that is present at rear and with activity.
+It feels like it is getting slightly better but his main problem is that he cannot still fully flex his knee as
+he used to do before the accident. He has noticed some clicking ad a little bit of swelling in the knee.
+He has had no instability but he is still a little bit stiff when he first gets up. He has been treated with
+4 sessions of physiotherapy. He has had two operations on this knee before when it was
+reconstructed which was an open procedure and an anthroscopy later to shave off the cartilage. He
+never got back to playing rugby after his reconstruction but he was able to do martial arts and all his
+policing duties without any problem.""")
+    paras.append("""He is mildly overweight, normal gait and no effusion in the knee and a range of motion of 0 to 110
+degrees which cannot be improved passively. His right elbow has full range of motion and is stable.
+He has had an Xray on right elbow showing no fracture and an xray of knee showed early medial
+compartment osteoarthritis. He has got new clicking in his knee and has been advised to get an MRI
+scan.""")
+    paras.append("""As far as his right elbow is concerned, he should reduce weight lifting activity and any repetitive right
+upper limb activity that causes any discomfort. His right knee requires an MRI scan to look for any
+loose bodies and I will see him with the results of the scan. In the meantime, he can continue with
+his general policing duties which he has anyways been doing since the accident""")
+    return paras
+    
+def get_paragraphs_for_entities():
+    paras =[]
+    paras.append("""18 February2020
+Dr Christine Fowler""")
+    paras.append("""Dear Christine,
+Re: Stephen Parrot""")
+    paras.append("""Stephen is a 53 year old gentleman who does general duties police work. He is married and has an
+18 year old daughter at home. He is right hand dominant. Cigarettes nil, alcohol rare, allergies nil.
+DVTe nil. Past medical history hyperlipidemia and reflux testicular cancer in 2000 and right knee
+reconstruction in 1987. Medications include Nexium and Crestor.""")
+    paras.append("""History presenting complaint: Right knee and right elbow injuries. On 12 January 2020, while at work
+he was trying to apprehend a stolen vehicle. The deployed some road spikes onto the road. The
+stolen vehicle went over the road spikes. He was attempting to retrieve the spikes to prevent
+damage to civilian and police vehicles and while he was doing that, he was hit by a police vehicle
+coming from behind. The police vehicle was approximately doing 50 km/hr. The headlight of the
+police car stuck him on the lateral aspect of the right knee. He jumped in the air and flicked in the air
+with his right elbow also hitting the police car before he flung over a barrier. He was able to mobilise
+afterwards when the adrenaline was taking effect. After he cooled down that night he developed
+significant pain in both the elbow and the knee. The elbow seemed to have settled with time but he
+has got slight discomfort on the lateral epicondyle of the right elbow but otherwise no instability,
+clicking, locking or catching of the elbow.""")
+    paras.append("""His knee has medial sided pain. It is an annoying type of pain that is present at rear and with activity.
+It feels like it is getting slightly better but his main problem is that he cannot still fully flex his knee as
+he used to do before the accident. He has noticed some clicking ad a little bit of swelling in the knee.
+He has had no instability but he is still a little bit stiff when he first gets up. He has been treated with
+4 sessions of physiotherapy. He has had two operations on this knee before when it was
+reconstructed which was an open procedure and an anthroscopy later to shave off the cartilage. He
+never got back to playing rugby after his reconstruction but he was able to do martial arts and all his
+policing duties without any problem.""")
+    paras.append("""He is mildly overweight, normal gait and no effusion in the knee and a range of motion of 0 to 110
+degrees which cannot be improved passively. His right elbow has full range of motion and is stable.
+He has had an Xray on right elbow showing no fracture and an xray of knee showed early medial
+compartment osteoarthritis. He has got new clicking in his knee and has been advised to get an MRI
+scan.""")
+    paras.append("""As far as his right elbow is concerned, he should reduce weight lifting activity and any repetitive right
+upper limb activity that causes any discomfort. His right knee requires an MRI scan to look for any
+loose bodies and I will see him with the results of the scan. In the meantime, he can continue with
+his general policing duties which he has anyways been doing since the accident""")
+    paras.append("""Kind regards,
+Dr Jason Sanders""")
+    return paras
+    
 def get_text_from_ocr_engine():
 
     return """18 February2020
@@ -141,17 +219,21 @@ if selected_menu == "Upload Document":
 elif selected_menu == "Extract Text":
     st.write(get_text_from_ocr_engine())
 elif selected_menu == "Summarize Document":
-    text = get_text_from_ocr_engine()
+    paragraphs= get_paragraphs_for_summaries()
     with st.spinner("Summarizing Document..."):
-        summary_text = pipeline_summarization(text, max_length=130, min_length=10, do_sample=False)
-        # Show output
-        st.write(summary_text[0]['summary_text'])
+        for text in paragraphs:
+            summary_text = pipeline_summarization(text, max_length=130, min_length=10, do_sample=False)
+            # Show output
+            st.write(summary_text[0]['summary_text'])
         
 elif selected_menu == "Extract Entities":
-    text = get_text_from_ocr_engine()
-    output = pipeline_ner (text)
-    entities_text =get_formatted_text_for_annotation(output)
-    annotated_text(*entities_text)
+    paragraphs= get_paragraphs_for_entities()
+    
+    with st.spinner("Extracting Entities..."):
+        for text in paragraphs:
+            output = pipeline_ner (text)
+            entities_text =get_formatted_text_for_annotation(output)
+            annotated_text(*entities_text)
 
 elif selected_menu == "Get Answers":
     st.subheader('Question')