Update app.py

app.py

@@ -106,15 +106,15 @@ def generate_follow_up(user_text):
 
 # ✅ Main Response Function
 def get_response(system_message, chat_history, user_text, max_new_tokens=512):
-    action = predict_action(user_text)  # 🔥 Fix: Define 'action'
+    action = predict_action(user_text)
 
     # ✅ Handle NASA-Specific Queries
     if action == "nasa_info":
         nasa_url, nasa_title, nasa_explanation = get_nasa_apod()
         response = f"**{nasa_title}**\n\n{nasa_explanation}"
+        follow_up = generate_follow_up(user_text)
         chat_history.append({'role': 'user', 'content': user_text})
         chat_history.append({'role': 'assistant', 'content': response})
-        follow_up = generate_follow_up(user_text)
         chat_history.append({'role': 'assistant', 'content': follow_up})
         return response, follow_up, chat_history, nasa_url
 
@@ -146,10 +146,9 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     if not response:
         response = "I'm sorry, but I couldn't generate a response. Can you rephrase your question?"
 
+    follow_up = generate_follow_up(user_text)
     chat_history.append({'role': 'user', 'content': user_text})
     chat_history.append({'role': 'assistant', 'content': response})
-
-    follow_up = generate_follow_up(user_text)
     chat_history.append({'role': 'assistant', 'content': follow_up})
 
     return response, follow_up, chat_history, None
@@ -210,18 +209,18 @@ if st.sidebar.button("Reset Chat"):
 
 user_input = st.chat_input("Type your message here...")  # Make sure this is executed first
 
+
 # ✅ Chat UI
 if user_input:
-    # ✅ Save user input to chat history
-    st.session_state.chat_history.append({'role': 'user', 'content': user_input})
 
-    # ✅ Invoke the AI model properly
+    st.session_state.chat_history.append({'role': 'user', 'content': 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
     )
 
+
     # ✅ Save AI response to chat history
     st.session_state.chat_history.append({'role': 'assistant', 'content': response})