Update app.py

app.py

@@ -305,7 +305,7 @@ def handle_chat(user_prompt, user_input, teapot_ai):
     return response
 
 def suggestion_button(suggestion_text):
-    if st.button(suggestion_text):
+    if st.button(suggestion_text, use_column_width=True):
         return suggestion_text
         
 def main():
@@ -335,22 +335,18 @@ def main():
     choice1 = random.choice(list1)
     choice2 = random.choice(list2)
     choice3 = random.choice(list3)
+    
     s1, s2, s3 = st.columns([1, 2, 3])
     
     with s1:
-        st.markdown("<div style='text-align: center;'>", unsafe_allow_html=True)
-        user_input = suggestion_button(choice1) or None
-        st.markdown("</div>", unsafe_allow_html=True)
-    
+        user_input = suggestion_button(choice1) or user_input
+        
     with s2:
-        st.markdown("<div style='text-align: center;'>", unsafe_allow_html=True)
         user_input = suggestion_button(choice2) or user_input
-        st.markdown("</div>", unsafe_allow_html=True)
-    
+        
     with s3:
-        st.markdown("<div style='text-align: center;'>", unsafe_allow_html=True)
         user_input = suggestion_button(choice3) or user_input
-        st.markdown("</div>", unsafe_allow_html=True)
+        
     if user_input:
         with st.spinner('Generating Response...'):
             response = handle_chat(user_prompt, user_input, teapot_ai)