Update app.py

app.py

@@ -304,7 +304,10 @@ def handle_chat(prompt, user_input, teapot_ai):
     
     return response
 
-
+def suggestion_button(suggestion_text):
+    if st.button(suggestion_text):
+        return suggestion_text
+        
 def main():
     st.set_page_config(page_title="TeapotAI Chat", page_icon=":robot_face:", layout="wide")
     
@@ -325,11 +328,11 @@ def main():
     
     s1, s2, s3 = st.columns([1, 2, 3])
     with s1:
-        user_input = "Tell me about teapotllm"
+        user_input = suggestion_button("Tell me about teapotllm")
     with s2:
-        user_input = "Who invented quantum mechanics?"
+        user_input = suggestion_button("Who invented quantum mechanics?")
     with s3:
-        user_input = "Extract the year Google was founded"
+        user_input = suggestion_button("Extract the year Google was founded")
     
     if user_input:
         with st.spinner('Generating Response...'):