Update app.py

app.py

@@ -148,30 +148,59 @@ st.markdown("</div>", unsafe_allow_html=True)
 # User Input Section
 user_input = st.text_area("Type your message:", height=100)
 
-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)
+# Ensure session state is initialized for response tracking
+if "response_ready" not in st.session_state:
+    st.session_state.response_ready = False  # False until HAL responds
+
+# Input box for user message
+user_input = st.text_area("Type your message:", height=100)
+
+# Create a placeholder for the Send button so it can be removed dynamically
+send_button_placeholder = st.empty()
+
+# Only show the Send button if HAL has not responded
+if not st.session_state.response_ready:
+    if send_button_placeholder.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 HAL's 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")
+
+            # Set response_ready to True so the Send button disappears
+            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)
+
+    # 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()
+