Update app.py

app.py

@@ -147,10 +147,7 @@ for message in st.session_state.chat_history:
 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
+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(
@@ -159,22 +156,16 @@ if 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")
 
     # Store follow-up question in session state
     st.session_state.follow_up = follow_up
-    st.session_state.response_ready = True  # Ensures response cycle continues
-
-# 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
-
+    st.session_state.response_ready = True  # Enables follow-up response cycle
 
 # Display follow-up question inside chat if available
 if st.session_state.response_ready and st.session_state.follow_up:
@@ -182,38 +173,3 @@ if st.session_state.response_ready and st.session_state.follow_up:
 
     # 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)
-
-    next_input = st.chat_input("HAL is waiting for your response...")  # Uses Enter to send
-
-    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 = ""
-
-# 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)
-
-    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 = ""