Update app.py

app.py

@@ -90,7 +90,7 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256):
 # --- Streamlit UI Setup ---
 st.set_page_config(page_title="NASA ChatBot", page_icon="🚀")
 
-st.title("🚀 HAL - A NASA AI Assistant")
+st.title("🚀 HAL - Your NASA AI Assistant")
 st.markdown("🌌 *Ask me about space, NASA, and beyond!*")
 
 # Ensure chat history is initialized
@@ -141,55 +141,13 @@ for message in st.session_state.chat_history:
     if message["role"] == "user":
         st.markdown(f"<div class='user-msg'><strong>You:</strong> {message['content']}</div>", unsafe_allow_html=True)
     else:
-        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {message['content']}</div>", unsafe_allow_html=True)
+        st.markdown(f"<div class='assistant-msg'><strong>Bot:</strong> {message['content']}</div>", unsafe_allow_html=True)
 
 st.markdown("</div>", unsafe_allow_html=True)
 
 # User Input Section
 user_input = st.text_area("Type your message:", height=100)
 
-# 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
-# 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.button("Send"):
     if user_input:
         response, follow_up, st.session_state.chat_history, image_url = get_response(
@@ -199,7 +157,7 @@ if st.button("Send"):
         )
 
         # Display 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>Bot:</strong> {response}</div>", unsafe_allow_html=True)
 
         # Display NASA image if available
         if image_url:
@@ -216,7 +174,4 @@ if st.button("Send"):
                     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)
-
-
-
+                st.markdown(f"<div class='assistant-msg'><strong>Bot:</strong> {response}</div>", unsafe_allow_html=True)