Update app.py

app.py

@@ -17,11 +17,11 @@ from streamlit.components.v1 import html
 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 "button_clicked" not in st.session_state: 
+#    st.session_state.button_clicked = True  
+#    
+#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
@@ -272,8 +272,8 @@ def run_model(input_text):
 ##========= on Past History Tab  =========  
   
 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["button_clicked"] =  False
+    #st.session_state["daily_button_clicked"] =  False
     st.session_state["past_button_clicked"] =  True
 
     with st.container(): 
@@ -298,16 +298,16 @@ if btnPastHistory or st.session_state["past_button_clicked"]:
         #runtext = historyAdmission['hospital_course_processed'].values[0]
         runtext = historyAdmission['hospital_course_processed'].values[0]
         
-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 btnAdmission:
+    #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 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
+#if btnDailyNarrative:
+    #st.session_state["button_clicked"] =  False
+    #st.session_state["past_button_clicked"] =  False
+    #st.session_state["daily_button_clicked"] =  True
 
 
     
@@ -334,41 +334,45 @@ 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 btnAdmission or st.session_state["button_clicked"]:
-        st.session_state["daily_button_clicked"] =  False
+if btnAdmission :
+        
+        #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)
-            #sentences=runtext.split('.')
-            st.text_area('Reference text', str(reference_text), height=150)
-        with col2:
-            st.button('NER')
-            # ===== 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('**CHIEF COMPLAINT:**')
-            st.write(str(AdmissionChiefCom)[1:-1])
-            st.markdown('**ADMISSION DIAGNOSIS:**')
-            st.markdown(str(diagnosis)[1:-1].capitalize())
-            st.markdown('**PROBLEM/ISSUE**')
-            #st.markdown(problem_entities)
-            st.markdown(f'<p style="background-color:PINK;color:#080808;font-size:16px;">{str(problem_entities)[1:-1]}</p>', unsafe_allow_html=True)
-            #genEntities(trans_df, 'DISEASE')
-            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)
-            #genEntities(trans_df, 'CHEMICAL')
-            #st.table(trans_df)   
-            st.markdown('**NER**')
-            with st.expander("See NER Details"):
-                st.markdown(ent_html, unsafe_allow_html=True)
-                
+        #st.session_state["button_clicked"] =  True
+
+          
+        with st.container():
+            with col1:
+                st.button('Summarize')
+                run_model(runtext)
+                #sentences=runtext.split('.')
+                st.text_area('Reference text', str(reference_text), height=150)
+            with col2:
+                st.button('NER')
+                # ===== 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('**CHIEF COMPLAINT:**')
+                st.write(str(AdmissionChiefCom)[1:-1])
+                st.markdown('**ADMISSION DIAGNOSIS:**')
+                st.markdown(str(diagnosis)[1:-1].capitalize())
+                st.markdown('**PROBLEM/ISSUE**')
+                #st.markdown(problem_entities)
+                st.markdown(f'<p style="background-color:PINK;color:#080808;font-size:16px;">{str(problem_entities)[1:-1]}</p>', unsafe_allow_html=True)
+                #genEntities(trans_df, 'DISEASE')
+                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)
+                #genEntities(trans_df, 'CHEMICAL')
+                #st.table(trans_df)   
+                st.markdown('**NER**')
+                with st.expander("See NER Details"):
+                    st.markdown(ent_html, unsafe_allow_html=True)
+
                 
-elif btnDailyNarrative or st.session_state["daily_button_clicked"]:
-    st.session_state["daily_button_clicked"] =  True
+elif btnDailyNarrative :
+#    st.session_state["daily_button_clicked"] =  True
     st.session_state["past_button_clicked"] =  False
-    st.session_state["button_clicked"] =  False
+#    st.session_state["button_clicked"] =  False
    
     with st.container():
         st.markdown('Daily Progress Note (24 hour event only):') 
@@ -383,8 +387,8 @@ elif btnDailyNarrative or st.session_state["daily_button_clicked"]:
 #else:
 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
+#    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']))