visually improved app.py

app.py

@@ -55,14 +55,17 @@ def get_random_example():
 if 'random_example' not in st.session_state:
     st.session_state.random_example = None
 
-if st.button("RANDOM EXAMPLE"):
+cols = st.columns(3)
+with cols[1]:
+    random_example_button = st.button("RANDOM EXAMPLE", use_container_width=True)
+if random_example_button:
     st.session_state.random_example = get_random_example()
     st.rerun()
 
 random_example = st.session_state.random_example
 
-def create_input(i):
-    row = input_meta_info.iloc[i]
+def create_input(df, i):
+    row = df.iloc[i]
     name = row['Name']
     description = row['Description']
 
@@ -118,20 +121,30 @@ def create_input(i):
 
 # create form
 with st.form("dynamic_form"):
-    # random_example_button = st.form_submit_button("RANDOM EXAMPLE")
+    sections = input_meta_info['Section'].unique()
+    for section in sections:
+
+        with st.container():
+            st.markdown(f"##### {section}")
+            sub_df = input_meta_info[input_meta_info['Section'] == section]
+
+            cols = st.columns(3)
+            with cols[0]:
+                for i in range(0, len(sub_df), 3):
+                    create_input(sub_df, i)
+            with cols[1]:
+                for i in range(1, len(sub_df), 3):
+                    create_input(sub_df, i)
+            with cols[2]:
+                for i in range(2, len(sub_df), 3):
+                    create_input(sub_df, i)    
+
+        # seperate line
+        st.markdown("---")
 
     cols = st.columns(3)
-    with cols[0]:
-        for i in range(0, len(input_meta_info), 3):
-            create_input(i)
     with cols[1]:
-        for i in range(1, len(input_meta_info), 3):
-            create_input(i)
-    with cols[2]:
-        for i in range(2, len(input_meta_info), 3):
-            create_input(i)    
-
-    predict_button = st.form_submit_button("PREDICT")
+        predict_button = st.form_submit_button("PREDICT", use_container_width=True, type='primary')
 
 # load mapping
 with open('./data/nacc_variable_mappings.pkl', 'rb') as file: