Update app.py

app.py

@@ -334,27 +334,25 @@ def main():
     
     s1, s2, s3 = st.columns([1, 1, 1])
     
-    user_input = None
+    user_suggested_input = None
     
     with s1:
         if st.button(choice1, use_container_width=True):
-            user_input = choice1
+            user_suggested_input = choice1
     
     with s2:
         if st.button(choice2, use_container_width=True):
-            user_input = choice2
+            user_suggested_input = choice2
     
     with s3:
         if st.button(choice3, use_container_width=True):
-            user_input = choice3
+            user_suggested_input = choice3
     
-    # Also allow user input through text box
-    if not user_input:
-        user_input = st.chat_input("Ask me anything")
+    user_input = st.chat_input("Ask me anything")
         
-    if user_input:
-        with st.spinner('Generating Response...'):
-            response = handle_chat(user_prompt, user_input, teapot_ai)
+
+    with st.spinner('Generating Response...'):
+        response = handle_chat(user_prompt, user_suggested_input or user_input, teapot_ai)
        
 
 if __name__ == "__main__":