Update app.py

app.py

@@ -149,6 +149,9 @@ 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...")  # Single input box
 
+# --- Single Input Box for Both Initial and Follow-Up Messages ---
+user_input = st.chat_input("Type your message here...")  # Single input box
+
 if user_input:
     response, follow_up, st.session_state.chat_history, image_url = get_response(
         system_message="You are a helpful AI assistant.",
@@ -173,6 +176,14 @@ if st.session_state.response_ready and st.session_state.follow_up:
     st.session_state.response_ready = False
 
 
+# Display follow-up question inside chat if available
+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
+
+
 # Conversational Follow-up with "Enter to Send"
 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)