Update app.py

app.py

@@ -37,11 +37,14 @@ if 'ft' in st.session_state or st.button('Next step'):
         )
         st.session_state['ft'] = chat_completion.choices[0].message.content
     
+    am = st.text_area("Assumption to be tested",
+                      placeholder="""What is the assumption you want to verify
+                      through the polling""")
     st.session_state['ft'] = st.text_area(
         "Factors to be considered in questions",
         st.session_state['ft'],
         height=300)
-    st.write('Edit above factors, add or remove if needed')
+    st.info('Before next step. Edit above factors, add or remove if needed')
 
     ft = st.session_state['ft']
     rf = os.environ.get("DESIGN_PROMPT")
@@ -52,8 +55,8 @@ if 'ft' in st.session_state or st.button('Next step'):
             {
                 "role": "user",
                 "content": f"""Extract all factors from the text to create context.
-                Generate 5 question based on the text and context.
-                Write the questionnaire form in markdown\n\n
+                Generate 5 question based on the text and context.\n\n
+                assumption to be tested through the questionnaire:\n\n{am}\n\n
                 text:\n\n{ft}\n\n
                 some design principles for your reference:\n\n
                 reference:\n\n```markdown\n{rf}\n```\n\n""",
@@ -66,8 +69,9 @@ if 'ft' in st.session_state or st.button('Next step'):
         )
 
         md = chat_completion.choices[0].message.content
-        suggested_sampling = 5 * len(md.split("\n"))
-        st.markdown(f"{md}\n\n**suggested number of sampling**: {suggested_sampling}")
+        st.markdown(md)
+        st.info('Suggested number of sampling is 5 to 10 times of number of questions', icon="ℹ️")
         # dump questions into excel and allow download
 
-        del st.session_state['ft']
+        if st.button('Restart'):
+            del st.session_state['ft']