Update app.py

app.py

@@ -12,15 +12,21 @@ from spacy.matcher import PhraseMatcher
 from spacy.tokens import Span
 #import en_ner_bc5cdr_md
 import re
-
-
 from streamlit.components.v1 import html
 
-# Store the initial value of widgets in session state
-if "visibility" not in st.session_state:
-    st.session_state.visibility = "visible"
-    st.session_state.disabled = False
+if "load_state" not in st.session_state: 
+    st.session_state.load_state = False
+
+if "button_clicked" not in st.session_state: 
+    st.session_state.button_clicked = False
+    
+if "daily_button_clicked" not in st.session_state: 
+    st.session_state.daily_button_clicked = False
 
+if "past_button_clicked" not in st.session_state: 
+    st.session_state.past_button_clicked = False
+    
+       
 #nlp = en_core_web_lg.load()
 nlp = spacy.load("en_ner_bc5cdr_md")
 
@@ -135,6 +141,7 @@ dailyNoteChange.rename(columns={'STORETIME':'Time of Record',
 dailyNote = df4['Full Text'].loc[(df4['Patient_ID'] == patient) & (df4['Admission_ID']==HospitalAdmission)]
 dailyNote = dailyNote.unique()
 
+    
 ##========= Buttons to the 5 tabs ======== Temp disabled Discharge Plan and Social Notes 
 ##col1, col2, col3, col4, col5 = st.columns([1,1,1,1,1]) -- to uncomment and comment below line to include discharge plan and social notes 
 col1, col2, col5 = st.columns([1,1,1])
@@ -142,14 +149,11 @@ col6, col7 =st.columns([2,2])
 with st.container():
     with col1:
         btnAdmission = st.button("🏥 Admission")
-        if btnAdmission:
-            #nav_page('Admission')
-            inputNote = "Input Admission Note"
+        inputNote = "Input Admission Note"
+ 
     with col2:
         btnDailyNarrative = st.button('📆Daily Narrative')
-        if btnDailyNarrative:
-            inputNote = "Input Daily Narrative Note"
-#    with col3:
+#    with col3:what
 #        btnDischargePlan = st.button('🗒️Discharge Plan')
 #        if btnDischargePlan:
 #            inputNote = "Input Discharge Plan"
@@ -159,61 +163,11 @@ with st.container():
 #            inputNote = "Input Social Note"
     with col5:
         btnPastHistory = st.button('📇Past History (6 Mths)')
-           
-           
-            
-##========= on Past History Tab  =========
 
-with st.container(): 
-    if btnPastHistory:
-        with col6: 
-            st.markdown('**No. of admission past 6 months:**')
-            st.markdown(countOfAdmission)
-       
-        with col7:
-            #st.date_input('Select Admission Date') # To replace with a dropdown filter instead 
-            #st.selectbox('Past Episodes',pastHistoryEp)
-            pastHistory = st.selectbox('Select Past History Admission', pastHistoryEpDate, format_func=lambda x: 'Select an option' if x == '' else x)
-            
-if btnPastHistory:
-    
-    #st.write('Past History')
-    historyAdmission =  df3.query(
-                "Patient_ID  == @patient & CHARTDATE_HADM_ID == @pastHistory"
-                )
-    runtext = historyAdmission['hospital_course_processed'].values[0]
-
-if not(btnPastHistory) and not(btnDailyNarrative):
-    runtext =st.text_area(inputNote, str(original_text2)[1:-1], height=300)
-    
-    
-
-    
-##========= END on Past History Tab  =========
-
-## ===== Commented out as no longer in use ===== 
-# Extract words associated with each entity
-#def genEntities(ann, entity):
-#              # entity colour dict
-#              #ent_col = {'DISEASE':'#B42D1B', 'CHEMICAL':'#F06292'}
-#              ent_col = {'DISEASE':'pink', 'CHEMICAL':'orange'}
-#              # separate into the different entities
-#              entities = trans_df['Class'].unique()
-#
-#              if entity in entities:
-#                             ent = list(trans_df[trans_df['Class']==entity]['Entity'].unique())
-#                             entlist = ",".join(ent)
-#                             st.markdown(f'<p style="background-color:{ent_col[entity]};color:#080808;font-size:16px;">{entlist}</p>', #unsafe_allow_html=True)
 
 
 ##======================== Start of NER Tagging ========================
-# ====== Old NER ======
-# doc = nlp(str(original_text2))
-# colors = { "DISEASE": "pink","CHEMICAL": "orange"}
-# options = {"ents": [ "DISEASE", "CHEMICAL"],"colors": colors}
-# ent_html = displacy.render(doc, style="ent", options=options)
-# ====== End of Old NER ======
-                  
+                 
 #lemmatizing the notes to capture all forms of negation(e.g., deny: denies, denying)
 def lemmatize(note, nlp):
     doc = nlp(note)
@@ -281,21 +235,7 @@ def dedupe(items):
             yield item
             seen.add(item)
 
-lem_clinical_note= lemmatize(runtext, nlp)
-#creating a doc object using BC5CDR model
-doc = nlp(lem_clinical_note)
-options = get_entity_options()
-
-#list of negative concepts from clinical note identified by negspacy
-results0 = negation_handling(lem_clinical_note, neg_model)
-
-matcher = match(nlp, results0,"NEG_ENTITY")
-
-#doc0: new doc object with added "NEG_ENTITY label"
-doc0 = overwrite_ent_lbl(matcher,doc)
 
-#visualizing identified Named Entities in clinical input text 
-ent_html = displacy.render(doc0, style='ent', options=options)
 
 ##======================== End of NER Tagging ========================
 
@@ -329,11 +269,76 @@ def run_model(input_text):
     
     st.success(output)
 
+##========= on Past History Tab  =========  
+if btnAdmission or st.session_state["button_clicked"]:
+    st.session_state["daily_button_clicked"] =  False
+    st.session_state["past_button_clicked"] =  False
+    st.session_state["button_clicked"] =  True
+
+
+    runtext =st.text_area(inputNote, str(original_text2)[1:-1], height=300)
+    
+if btnPastHistory or st.session_state["past_button_clicked"]:
+    st.session_state["button_clicked"] =  False
+    st.session_state["daily_button_clicked"] =  False
+    st.session_state["past_button_clicked"] =  True
+
+    with st.container(): 
+        with col6: 
+
+            st.markdown('**No. of admission past 6 months:**')
+            st.markdown(countOfAdmission)
+       
+        with col7:
+            #st.date_input('Select Admission Date') # To replace with a dropdown filter instead 
+            #st.selectbox('Past Episodes',pastHistoryEp)
+            pastHistory = st.selectbox('Select Past History Admission', pastHistoryEpDate, format_func=lambda x: 'Select an option' if x == '' else x)        
+    
+    historyAdmission =  df3.query(
+                "Patient_ID  == @patient & CHARTDATE_HADM_ID == @pastHistory"
+                )
+    
+   
+    if historyAdmission.shape[0] == 0:
+        runtext = "No past episodes" 
+    else: 
+        #runtext = historyAdmission['hospital_course_processed'].values[0]
+        runtext = historyAdmission['hospital_course_processed'].values[0]
+    
+if btnDailyNarrative or st.session_state["daily_button_clicked"]:
+    st.session_state["button_clicked"] =  False
+    st.session_state["past_button_clicked"] =  False
+    st.session_state["daily_button_clicked"] =  True
+
+
+    
+
 
+lem_clinical_note= lemmatize(runtext, nlp)
+#creating a doc object using BC5CDR model
+doc = nlp(lem_clinical_note)
+options = get_entity_options()
+
+#list of negative concepts from clinical note identified by negspacy
+results0 = negation_handling(lem_clinical_note, neg_model)
+
+matcher = match(nlp, results0,"NEG_ENTITY")
+
+#doc0: new doc object with added "NEG_ENTITY label"
+doc0 = overwrite_ent_lbl(matcher,doc)
+
+#visualizing identified Named Entities in clinical input text 
+ent_html = displacy.render(doc0, style='ent', options=options)
+
+
+    
 col1, col2 = st.columns([1,1])
 
 #to not show summary and references text for Past History and Daily Narrative
-if not(btnPastHistory) and not(btnDailyNarrative):
+if btnAdmission or st.session_state["button_clicked"]:
+        st.session_state["daily_button_clicked"] =  False
+        st.session_state["past_button_clicked"] =  False
+        st.session_state["button_clicked"] =  True
         with col1:
             st.button('Summarize')
             run_model(runtext)
@@ -359,51 +364,66 @@ if not(btnPastHistory) and not(btnDailyNarrative):
             st.markdown('**NER**')
             with st.expander("See NER Details"):
                 st.markdown(ent_html, unsafe_allow_html=True)
-if btnDailyNarrative: 
+                
+                
+elif btnDailyNarrative or st.session_state["daily_button_clicked"]:
+    st.session_state["daily_button_clicked"] =  True
+    st.session_state["past_button_clicked"] =  False
+    st.session_state["button_clicked"] =  False
+   
     with st.container():
         st.markdown('Daily Progress Note (24 hour event only):') 
         st.markdown(str(dailyNote)[1:-1])
  
     
     with st.container():
-#        hide_table_row_index = """
-#                    <style>
-#                    thead tr th:first-child {display:none}
-#                    tbody th {display:none}
-#                    </style>
-#                    """
-#
-#        # Inject CSS with Markdown
-#        st.markdown(hide_table_row_index, unsafe_allow_html=True)
-#        st.table(dailyNoteChange)
         styler = dailyNoteChange.style.hide_index()
         st.write(styler.to_html(), unsafe_allow_html=True)
     st.markdown(f'<p style="color:#828080;font-size:12px;">*Current prototype displays only a single section within the daily progress note, could also potentially include all sections within each progress note and allow user to select the section changes the user wants to look at</p>', unsafe_allow_html=True)
 
 #else:
-if btnPastHistory:
+elif btnPastHistory or st.session_state["past_button_clicked"]:
+    st.session_state["past_button_clicked"] =  True
+    st.session_state["button_clicked"] =  False
+    st.session_state["daily_button_clicked"] =  False
 # ===== Adding the Disease/Chemical into a list =====    
     problem_entities = list(dedupe([t for t in doc0.ents if t.label_ == 'DISEASE']))
     medication_entities = list(dedupe([t for t in doc0.ents if t.label_ == 'CHEMICAL']))
-    st.markdown('Admission Date: ' + historyAdmission['ADMITTIME'].values[0])
-    st.markdown('Date of Discharge: ' + historyAdmission['DISCHTIME'].values[0])
-    st.markdown('Days from current admission: ' + str(historyAdmission['days_from_index'].values[0]) +' days')
+    if historyAdmission.shape[0] == 0:
+        st.markdown('Admission Date: NA')
+        st.markdown('Date of Discharge: NA') 
+        st.markdown('Days from current admission: NA')
+    else: 
+        st.markdown('Admission Date: ' + historyAdmission['ADMITTIME'].values[0])
+        st.markdown('Date of Discharge: ' + historyAdmission['DISCHTIME'].values[0])
+        st.markdown('Days from current admission: ' + str(historyAdmission['days_from_index'].values[0]) +' days')
+
     #st.markdown('Summary: ')
     st.markdown(f'<p style="color:#080808;font-size:16px;"><b>Summary: </b></p>', unsafe_allow_html=True)
     
               
     if model == "BertSummarizer":
-        st.markdown(str(historyAdmission['BertSummarizer'].values[0]))
+        if historyAdmission.shape[0] == 0:
+            st.markdown('NA')
+        else: 
+            st.markdown(str(historyAdmission['BertSummarizer'].values[0]))
     elif model == "t5seq2eq":
-        st.markdown(str(historyAdmission['t5seq2eq'].values[0]))
+        if historyAdmission.shape[0] == 0:
+            st.markdown('NA')
+        else: 
+            st.markdown(str(historyAdmission['t5seq2eq'].values[0]))     
     st.markdown(f'<p style="color:#080808;font-size:16px;"><b>Diagnosis: </b></p>', unsafe_allow_html=True)
-    st.markdown(str(historyAdmission['Diagnosis_Description'].values[0]))
-    st.markdown('**PROBLEM/ISSUE**')
-    st.markdown(f'<p style="background-color:PINK;color:#080808;font-size:16px;">{str(problem_entities)[1:-1]}</p>', unsafe_allow_html=True)
-    st.markdown('**MEDICATION**')
-    st.markdown(f'<p style="background-color:orange;color:#080808;font-size:16px;">{str(medication_entities)[1:-1]}</p>', unsafe_allow_html=True)        
-    st.markdown('Discharge Disposition: ' + str(historyAdmission['DISCHARGE_LOCATION'].values[0]))
-    with st.expander('Full Discharge Summary'):
-            #st.write("line 1  \n  line 2  \n line 3")
-            fulldischargesummary = historyAdmission['TEXT'].values[0]
-            st.write(fulldischargesummary)
+    
+    if historyAdmission.shape[0] == 0:
+        st.markdown('NA')
+    else:
+        st.markdown(str(historyAdmission['Diagnosis_Description'].values[0]))
+        st.markdown('**PROBLEM/ISSUE**')
+        st.markdown(f'<p style="background-color:PINK;color:#080808;font-size:16px;">{str(problem_entities)[1:-1]}</p>', unsafe_allow_html=True)
+        st.markdown('**MEDICATION**')
+        st.markdown(f'<p style="background-color:orange;color:#080808;font-size:16px;">{str(medication_entities)[1:-1]}</p>', unsafe_allow_html=True)        
+        st.markdown('Discharge Disposition: ' + str(historyAdmission['DISCHARGE_LOCATION'].values[0]))
+        with st.expander('Full Discharge Summary'):
+                #st.write("line 1  \n  line 2  \n line 3")
+                fulldischargesummary = historyAdmission['TEXT'].values[0]
+                st.write(fulldischargesummary)