Update app.py

app.py

@@ -146,30 +146,42 @@ for message in st.session_state.chat_history:
         st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {message['content']}</div>", unsafe_allow_html=True)
 st.markdown("</div>", unsafe_allow_html=True)
 
-# --- Single Input Box for Both Initial and Follow-Up Messages ---
-user_input = st.chat_input("Type your message here...")  # Only ONE chat_input()
-
-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
-    )
+# --- Input & Button Handling ---
+user_input = st.text_area("Type your message:", height=100)
+
+send_button_placeholder = st.empty()
+
+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)
+            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")
+            if image_url:
+                st.image(image_url, caption="NASA Image of the Day")
 
-    # Store follow-up question in session state
-    st.session_state.follow_up = follow_up
-    st.session_state.response_ready = True  # Enables follow-up response cycle
+            # Store follow-up question
+            st.session_state.follow_up = follow_up
+            st.session_state.response_ready = True  # Hide Send button after response
 
-# Display follow-up question inside chat if available
+# Conversational Follow-up
 if st.session_state.response_ready and st.session_state.follow_up:
     st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {st.session_state.follow_up}</div>", unsafe_allow_html=True)
 
-    # Reset response state so user can type next input
-    st.session_state.response_ready = False
+    next_input = st.text_input("HAL is waiting for your response...")
+
+    if next_input:
+        response, _, st.session_state.chat_history, _ = get_response(
+            system_message="You are a helpful AI assistant.",
+            user_text=next_input,
+            chat_history=st.session_state.chat_history
+        )
+        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
+
+        st.session_state.response_ready = False
+        st.session_state.follow_up = ""