Tried adding columns to buttons results/cue

app.py

@@ -211,6 +211,7 @@ if prompt:
     st.markdown(prompt)
   #add to history
   st.session_state.messages.append({'role': 'user', 'content': prompt})
+  #TODO: replace it with zero-shot classifier
   yes = ['yes', 'again', 'Yes', 'sure', 'new word', 'yes!', 'yep', 'yeah']
   if prompt in yes:
     write_bot("Please describe your word!")
@@ -228,8 +229,11 @@ if prompt:
     st.session_state.actions.append("result")
 
 if st.session_state.actions[-1] == "result":
-  a1 = st.button('Results', key=10)
-  a2 = st.button('Cue', key=11)
+  col1, col2 = st.columns(2)
+  with col1:
+    a1 = st.button('Results', key=10)
+  with col2:
+    a2 = st.button('Cue', key=11)
   if a1:
     write_bot("Here are my guesses about your word:")
     st.write(st.session_state.results['results_print'])