Update app.py

app.py

@@ -321,28 +321,40 @@ def main():
     choice2 = random.choice(list2)
     choice3 = random.choice(list3)
     
-    s1, s2, s3 = st.columns([1, 1, 1])
+
+    # Initialize user_suggested_input in session_state if it doesn't exist
+    if 'user_suggested_input' not in st.session_state:
+        st.session_state.user_suggested_input = None
     
-    user_suggested_input = None
+    # Define columns
+    s1, s2, s3 = st.columns([1, 1, 1])
     
+    # Button click handling
     with s1:
         if st.button(choice1, use_container_width=True):
-            user_suggested_input = choice1
+            st.session_state.user_suggested_input = choice1
     
     with s2:
         if st.button(choice2, use_container_width=True):
-            user_suggested_input = choice2
+            st.session_state.user_suggested_input = choice2
     
     with s3:
         if st.button(choice3, use_container_width=True):
-            user_suggested_input = choice3
+            st.session_state.user_suggested_input = choice3
     
+    # Input field for chat
     user_input = st.chat_input("Ask me anything")
-        
-    if user_input:
+    
+    # Handle chat input and button selection
+    if user_input or st.session_state.user_suggested_input:
         with st.spinner('Generating Response...'):
-            response = handle_chat(user_prompt, user_suggested_input or user_input, teapot_ai)
-           
+            response = handle_chat(user_prompt, st.session_state.user_suggested_input or user_input, teapot_ai)
+            st.write(response)
+    
+    # Reset the input field after generating the response if needed
+    if user_input or st.session_state.user_suggested_input:
+        st.session_state.user_suggested_input = None  # Reset suggested input after processing
+
 
 if __name__ == "__main__":
     main()