Update app.py

app.py

@@ -304,9 +304,9 @@ def handle_chat(prompt, user_input, teapot_ai):
     
     return response
 
-def suggestion_button(suggestion_text, teapot_ai):
+def suggestion_button(user_prompt, suggestion_text, teapot_ai):
     if st.button(suggestion_text):
-        handle_chat(suggestion_text, teapot_ai)
+        handle_chat(user_prompt, user_input, teapot_ai)
 
 
 def main():
@@ -329,11 +329,11 @@ def main():
     
     s1, s2, s3 = st.columns([1, 2, 3])
     with s1:
-        suggestion_button("Tell me about teapotllm", teapot_ai)
+        suggestion_button(user_prompt, "Tell me about teapotllm", teapot_ai)
     with s2:
-        suggestion_button("Who invented quantum mechanics?", teapot_ai)
+        suggestion_button(user_prompt, "Who invented quantum mechanics?", teapot_ai)
     with s3:
-        suggestion_button("Extract the year Google was founded", teapot_ai)
+        suggestion_button(user_prompt, "Extract the year Google was founded", teapot_ai)
     
     if user_input:
         with st.spinner('Generating Response...'):