Update app.py

app.py

@@ -292,42 +292,53 @@ def daily_log():
         st.line_chart(slp_df)
     st.markdown("""---""")    
     check_box_prb = st.checkbox('I have completed my 7 days of tracking and I am ready to know your recomendation')        
-    #st.dataframe(slp_df) 
+    #st.dataframe(slp_df)
+    if 'cb_key' not in st.session_state:
+        st.session_state['cb_key'] = False    
     if check_box_prb:
-        slp_df = slp_df.copy(deep=False)
-        #if slp_df.shape[0] == 7:
-        #st.write(slp_df.shape[0])
-        st.write("That's Awesome! Keep the momentum going!")
-        st.markdown("""**Why don't you share what you are going through? This would help us to give a better recommendation**""")
-        classifier = pipeline("zero-shot-classification", model='cross-encoder/nli-distilroberta-base')
-        sent = st.text_area('Brief your problem, we will suggest activities that could help you overcome stress',value="""As I am in college I am up all night all day getting only 5 hours of sleep daily. Eventually after some days I started facing some issues like mood swings and feeling lazy all the time , body ache, puffy eyes and ended up eating at night (one becomes snacky resulting at night) in weight gain.""")
-        prb_button = st.button(label='Submit',key='probsubmit')
+        st.session_state['cb_key'] = True
+    if  st.session_state['cb_key']: 
+        if check_box_prb:
         #slp_df = slp_df.copy(deep=False)
-        #st.dataframe(slp_df) 
-        if prb_button:
-            #st.dataframe(slp_df) 
-            #slp_df = slp_df.copy(deep=False)
-            candidate_labels = ["Sleep-disorder", "Work-stress", "Peer pressure"]
-            res = classifier(sent, candidate_labels)
-            problem = res['labels'][0]
-            st.write('We feel working on ',problem ,' could improve your overall well-being')
-            st.markdown("""###Not sure where to start?###"""+ """ \U0001f914 """+""" ###Here are our recommendations!!### \U0001f917""")
-            if problem == 'Sleep-disorder':
-                st.write("Try this [Mediatation for Sleep on Headspace](https://www.headspace.com/meditation/sleep)")
-                st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
-            elif problem == 'Work-stress':
-                st.write("Try this [Mediatation for Anxiety on Headspace](https://www.headspace.com/meditation/anxiety)")
-                st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
-            elif problem == 'Peer pressure':
-                st.write("Try this [Calm Down Meditation](https://www.headspace.com/articles/how-to-calm-down)")
-                st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
+            if slp_df.shape[0] == 7:
+        #st.write(slp_df.shape[0])
+                st.write("That's Awesome! Keep the momentum going!")
+                st.markdown("""**Why don't you share what you are going through? This would help us to give a better recommendation**""")
+                classifier = pipeline("zero-shot-classification", model='cross-encoder/nli-distilroberta-base')
+                sent = st.text_area('Brief your problem, we will suggest activities that could help you overcome stress',value="""As I am in college I am up all night all day getting only 5 hours of sleep daily. Eventually after some days I started facing some issues like mood swings and feeling lazy all the time , body ache, puffy eyes and ended up eating at night (one becomes snacky resulting at night) in weight gain.""")
+                prb_button = st.button(label='Submit',key='probsubmit')
+                #slp_df = slp_df.copy(deep=False)
+                #st.dataframe(slp_df) 
+                if 'prbsol_key' not in st.session_state:
+                    st.session_state['prbsol_key'] = False    
+                if prb_button:
+                    st.session_state['prbsol_key'] = True
+                if  st.session_state['prbsol_key']:
+                
+                    #st.dataframe(slp_df) 
+                    #slp_df = slp_df.copy(deep=False)
+                    candidate_labels = ["Sleep-disorder", "Work-stress", "Peer pressure"]
+                    res = classifier(sent, candidate_labels)
+                    problem = res['labels'][0]
+                    st.write('We feel working on ',problem ,' could improve your overall well-being')
+                    st.markdown("""###Not sure where to start?###"""+ """ \U0001f914 """+""" ###Here are our recommendations!!### \U0001f917""")
+                    if problem == 'Sleep-disorder':
+                        st.write("Try this [Mediatation for Sleep on Headspace](https://www.headspace.com/meditation/sleep)")
+                        st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
+                    elif problem == 'Work-stress':
+                        st.write("Try this [Mediatation for Anxiety on Headspace](https://www.headspace.com/meditation/anxiety)")
+                        st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
+                    elif problem == 'Peer pressure':
+                        st.write("Try this [Calm Down Meditation](https://www.headspace.com/articles/how-to-calm-down)")
+                        st.write("Try this [Yammer community for Mindfulness](https://web.yammer.com/main/org/optum.com/groups/eyJfdHlwZSI6Ikdyb3VwIiwiaWQiOiI1NzMxOTMxNzUwNCJ9/new)")
                 
                 
             #st.write(type(res['labels']))
-        #else:   
-        #    warning = '<p style="font-family:sans-serif; color:Red; font-size: 42px;">Looks like you have not tracked your habits for a week. Instant solutions may not always work. Keep track for atleast 7 days</p>'
-            #st.markdown(warning, unsafe_allow_html=True)
-            #st.write('Looks like you have not tracked your habits for a week. Instant solutions may not always work. Keep track for atleast 7 days')
+                    
+            else:   
+                warning = '<p style="font-family:sans-serif; color:Red; font-size: 42px;">Looks like you have not tracked your habits for a week. Instant solutions may not always work. Keep track for atleast 7 days</p>'
+                st.markdown(warning, unsafe_allow_html=True)
+                st.write('Looks like you have not tracked your habits for a week. Instant solutions may not always work. Keep track for atleast 7 days')
 
 page_names_to_funcs = {
     "Home": home_page,