Update app.py

app.py

@@ -190,29 +190,33 @@ if not st.session_state.response_ready:
             st.session_state.response_ready = True
 
 # Show follow-up questions & continue button **only if HAL has responded**
-if st.session_state.response_ready:
-    follow_up_options = [follow_up, "Explain differently", "Give me an example"]
-    selected_option = st.radio("What would you like to do next?", follow_up_options)
+if st.button("Send"):
+    if user_input:
+        response, follow_up, st.session_state.chat_history, image_url = get_response(
+            system_message="You are a helpful AI assistant.",
+            user_text=user_input,
+            chat_history=st.session_state.chat_history
+        )
+
+        # Display response
+        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
+
+        # Display NASA image if available
+        if image_url:
+            st.image(image_url, caption="NASA Image of the Day")
+
+        # Follow-up question suggestions
+        follow_up_options = [follow_up, "Explain differently", "Give me an example"]
+        selected_option = st.radio("What would you like to do next?", follow_up_options)
+
+        if st.button("Continue"):
+            if selected_option:
+                response, _, st.session_state.chat_history, _ = get_response(
+                    system_message="You are a helpful AI assistant.",
+                    user_text=selected_option,
+                    chat_history=st.session_state.chat_history
+                )
+                st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
-    # Placeholder for the Continue button so it can be refreshed dynamically
-    continue_button_placeholder = st.empty()
-
-    if continue_button_placeholder.button("Continue"):
-        if selected_option:
-            response, _, st.session_state.chat_history, _ = get_response(
-                system_message="You are a helpful AI assistant.",
-                user_text=selected_option,
-                chat_history=st.session_state.chat_history
-            )
-            st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
-
-            # Reset response tracking so "Send" button appears for new input
-            st.session_state.response_ready = False
-
-# Reset chat and buttons
-if st.sidebar.button("Reset Chat"):
-    st.session_state.chat_history = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
-    st.session_state.response_ready = False  # Reset so the Send button reappears
-    st.experimental_rerun()