Update app.py

app.py

@@ -194,30 +194,36 @@ if st.sidebar.button("Reset Chat"):
     st.session_state.follow_up = ""
 
 # ✅ Chat UI
-user_input = st.chat_input("Type your message here...")
-
 if user_input:
-    # Save user message
+    # ✅ Save user input to chat history
     st.session_state.chat_history.append({'role': 'user', 'content': user_input})
 
-    # Get AI response (replace with actual function)
-    response = "This is HAL's response."  # Example response
-    follow_up = "Would you like to explore a related topic?"  # Follow-up suggestion
+    # ✅ Invoke the AI model properly
+    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
+    )
 
-    # Save AI response & follow-up
+    # ✅ Save AI response to chat history
     st.session_state.chat_history.append({'role': 'assistant', 'content': response})
-    st.session_state.chat_history.append({'role': 'assistant', 'content': follow_up})
-
-    # Store follow-up question separately if needed
-    st.session_state.follow_up = follow_up
 
-    # ✅ Ensure response is not empty before calling st.markdown()
+    # ✅ Ensure response is displayed
     if response:
         st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
-    st.session_state.follow_up = follow_up
+    # ✅ Save and display follow-up question separately
+    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:
     st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {st.session_state.follow_up}</div>", unsafe_allow_html=True)