Update app.py

app.py

@@ -211,34 +211,31 @@ user_input = st.chat_input("Type your message here...")  # Make sure this is exe
 
 
 # ✅ Chat UI
-if user_input:
+user_input = st.chat_input("Type your message here...")
 
-    st.session_state.chat_history.append({'role': 'user', 'content': user_input})
+if user_input:
+    # ✅ Ensure get_response() returns a response
     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
     )
 
+    # ✅ Set default value if `response` is missing
+    if not response:
+        response = "I'm sorry, but I couldn't generate a response."
 
-    # ✅ Save AI response to chat history
-    st.session_state.chat_history.append({'role': 'assistant', 'content': response})
-
-    # ✅ Ensure response is displayed
-    if response:
-        st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
+    # ✅ Display chatbot response
+    st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
-    # ✅ Save and display follow-up question separately
+    # ✅ Handle follow-up question
     if follow_up:
-        st.session_state.chat_history.append({'role': 'assistant', 'content': follow_up})
         st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {follow_up}</div>", unsafe_allow_html=True)
 
     # ✅ Display NASA image if available
     if image_url:
         st.image(image_url, caption="NASA Image of the Day")
 
-    st.session_state.response_ready = True
-
 
 # ✅ Check before displaying follow-up message
 if st.session_state.response_ready and st.session_state.follow_up: