updated with formatting and overall flushing out

app.py

@@ -12,6 +12,15 @@ st.set_page_config(page_title='auditory skills resources', page_icon='icon-128x1
 if 'one' not in st.session_state:
     st.session_state['one'] = 'value'
 
+if 'two' not in st.session_state:
+    st.session_state['two'] = 'value'
+
+if 'three' not in st.session_state:
+    st.session_state['three'] = 'value'
+
+if 'four' not in st.session_state:
+    st.session_state['four'] = 'value'
+
 #main page content
 with st.sidebar:
     st.title('auditory skills')
@@ -24,10 +33,6 @@ with st.sidebar:
 
     st.write('Made with resources provided by the HATCH (Helping Adults Talk to Children) Lab. The HATCH lab is located at Idaho State University in Meridian, Idaho and seeks to ensure that adults have access to tools and resources to optimize the language of children through connection and engagement')
 
-#for callback
-def proc(key):
-    st.session_state['one'] = key
-    #st.write(st.session_state['one'])
 
 def main():
         #st.markdown('hello')
@@ -67,21 +72,43 @@ def main():
         step_cols = st.columns([.5,.85,.85,.85,.85,.5])
         step_cols[1].button('Does your child respond to a familiar voice?',on_click=set_form_step,args=['Jump',1])
         step_cols[2].button('Does your child listen to somebody speaking?',on_click=set_form_step,args=['Jump',2])
-        step_cols[3].button('When somebody is speaking, does your child turn his/her head towards the speaker?',on_click=set_form_step,args=['Jump',3])
-        step_cols[4].button('Is your child interested in toys producing sounds or music?',on_click=set_form_step,args=['Jump',4])
+        step_cols[3].button('Is your child interested in toys producing sounds or music?',on_click=set_form_step,args=['Jump',3])
+        step_cols[4].button('Does your child look for a speaker they cannot see?',on_click=set_form_step,args=['Jump',4])
 
     ### Replace Wizard Form Body with this ###
     def wizard_form_body():
         if st.session_state['current_step'] == 1:
-            with st.form("question-1"):
-                st.write("session state is one")
-                #st.text_input("enter text",on_change=proc,key='key',args=("Brightness",))
-                textone = st.text_input("enter text")
-                submitted = st.form_submit_button("Submit")
-                if submitted:
-                    #sequence = textone
-                    #output = classifier(sequence, sequence_labels)
-                    st.session_state['one'] = textone
+            form1 = st.form("question-1")
+            form1.write("Describe how your child reacts when a familiar voice is presented.  Responses can be anything such as looking at the person, moving, a facial expression (e.g., smile), or a sound they make.")
+            textone = form1.text_input(" ")
+            submitted1 = form1.form_submit_button("Submit")
+            if submitted1:
+                st.session_state['one'] = textone
+
+        if st.session_state['current_step'] == 2:
+            form2 = st.form("question-2")
+            form2.write("Describe how your child reacts when someone is talking to them and responds with actions (e.g., looking at the person speaking, smiling) and vocalizations (e.g., laughing, babbling back and forth with speaker).")
+            texttwo = form2.text_input(" ")
+            submitted2 = form2.form_submit_button("Submit")
+            if submitted2:
+                st.session_state['two'] = texttwo
+
+        if st.session_state['current_step'] == 3:
+            form3 = st.form("question-3")
+            form3.write("Describe what happens when your child plays with or notices toys that are musical or have sound qualities.  Your child may pick up the toy, move it, look at it, and/or mouth it.")
+            textthree = form3.text_input(" ")
+            submitted3 = form3.form_submit_button("Submit")
+            if submitted3:
+                st.session_state['three'] = textthree
+
+        if st.session_state['current_step'] == 4:
+            form4 = st.form("question-4")
+            form4.write("Describe what would happen if, while interacting with your child, another person begins talking behind you.  Would your child stop talking? Turn and look? Search for the sound? Or perhaps vocalize?")
+            textfour = form4.text_input(" ")
+            submitted4 = form4.form_submit_button("Submit")
+            if submitted4:
+                st.session_state['four'] = textfour
+
 
 
     def wizard_form_footer():
@@ -112,15 +139,17 @@ def main():
 
     render_wizard_view()
     st.subheader('Your resources')
-    st.markdown('list here...')
-    st.markdown(st.session_state['one'])
+    #st.markdown(st.session_state['one'])
+    #st.markdown(st.session_state['two'])
+    #st.markdown(st.session_state['three'])
+    #st.markdown(st.session_state['four'])
     #sequence = "My kid will look at me when the tv is on but I'm in the other room"
     #sequence_labels=["Detection", "discrimination", "Identification", "Classification"]
 
     #output = classifier(sequence, sequence_labels)
     #st.markdown(output)
     with st.form("get-results"):
-        st.write("get results form")
+        #st.write("get results form")
         submitted = st.form_submit_button("Get Results")
         if submitted:
             #load everything
@@ -129,10 +158,15 @@ def main():
             #define classifier for zero shot classification
             classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
             #define sequence
-            sequence_labels=["Detection", "discrimination", "Identification", "Classification"]
+            sequence_labels=["Detection", "Discrimination", "Identification", "Classification"]
             sequence = st.session_state['one']
             output = classifier(sequence, sequence_labels)
-            st.markdown(output)
+            #st.markdown(output)
+            df = pd.DataFrame(output)
+            #st.markdown(df)
+            #st.markdown("For your response: " + str(df.at[0,'sequence']))
+            st.markdown("Your responses indicate your child is at the " + str(df.at[0,'labels']) + " " + str(df.at[0,'scores']) + " step of auditory processing")
+            st.markdown("Here is a link to the " + str(df.at[0,'labels']) + " resource.")