Update app.py

app.py

@@ -147,27 +147,40 @@ for message in st.session_state.chat_history:
 st.markdown("</div>", unsafe_allow_html=True)
 
 # --- Input & Button Handling ---
-user_input = st.text_area("Type your message:", height=100)
+# --- Input Handling with "Enter to Send" ---
+user_input = st.chat_input("Type your message here...")  # Removes "Send" button
+
+if user_input and not st.session_state.response_ready:
+    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
+    )
+
+    st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
-send_button_placeholder = st.empty()
+    if image_url:
+        st.image(image_url, caption="NASA Image of the Day")
 
-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
-            )
+    st.session_state.follow_up = follow_up
+    st.session_state.response_ready = True  # Ensures response cycle continues
 
-            st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
+# 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)
 
-            if image_url:
-                st.image(image_url, caption="NASA Image of the Day")
+    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)
 
-            # Store follow-up question
-            st.session_state.follow_up = follow_up
-            st.session_state.response_ready = True  # Hide Send button after response
+        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: