Update app.py

app.py

@@ -304,10 +304,7 @@ def handle_chat(user_prompt, user_input, teapot_ai):
     
     return response
 
-def suggestion_button(suggestion_text):
-    if st.button(suggestion_text, use_container_width=True):
-        return suggestion_text
-        
+
 def main():
     st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
@@ -339,13 +336,15 @@ def main():
     s1, s2, s3 = st.columns([1, 1, 1])
     
     with s1:
-        user_input = suggestion_button(choice1) or user_input
+        if st.button(choice1):
+            user_input = choice1
         
     with s2:
-        user_input = suggestion_button(choice2) or user_input
-        
+         if st.button(choice2):
+            user_input = choice2
     with s3:
-        user_input = suggestion_button(choice3) or user_input
+        if st.button(choice3):
+            user_input = choice3
         
     if user_input:
         with st.spinner('Generating Response...'):