Update app.py

app.py

@@ -17,6 +17,9 @@ if "chat_history" not in st.session_state:
 if "response_ready" not in st.session_state:
     st.session_state.response_ready = False  # Tracks whether HAL has responded
 
+if "follow_up" not in st.session_state:
+    st.session_state.follow_up = ""  # Stores follow-up question
+
 # --- Set Up Model & API Functions ---
 model_id = "mistralai/Mistral-7B-Instruct-v0.3"
 
@@ -99,6 +102,7 @@ st.markdown("🌌 *Ask me about space, NASA, and beyond!*")
 if st.sidebar.button("Reset Chat"):
     st.session_state.chat_history = [{"role": "assistant", "content": "Hello! How can I assist you today?"}]
     st.session_state.response_ready = False
+    st.session_state.follow_up = ""
     st.experimental_rerun()
 
 # Custom Chat Styling
@@ -161,11 +165,13 @@ if not st.session_state.response_ready:
             if image_url:
                 st.image(image_url, caption="NASA Image of the Day")
 
+            # Store follow-up question
+            st.session_state.follow_up = follow_up
             st.session_state.response_ready = True  # Hide Send button after response
 
 # Conversational Follow-up
-if st.session_state.response_ready:
-    st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {follow_up}</div>", unsafe_allow_html=True)
+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...")
 
@@ -176,4 +182,6 @@ if st.session_state.response_ready:
             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  # Allow new input
+
+        st.session_state.response_ready = False
+        st.session_state.follow_up = ""