Update app.py

app.py

@@ -105,7 +105,7 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     """
     sentiment = analyze_sentiment(user_text)
     action = predict_action(user_text)
-    
+
     # Extract style instruction if present
     style_instruction = ""
     lower_text = user_text.lower()
@@ -141,32 +141,34 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
             "AI: Please provide a detailed explanation in depth. "
             "Ensure your response covers the topic thoroughly and is written in a friendly, conversational style, "
             "starting with a phrase like 'Certainly!', 'Of course!', or 'Great question!'." 
-            "Answer exclusively in English, and do not include extra commentary."+ style_clause +
+            "Answer exclusively in English, and do not include extra commentary." + style_clause +
             "\nHAL:"
         )
     )
     
     chat = prompt | hf.bind(skip_prompt=True) | StrOutputParser(output_key='content')
     response = chat.invoke(input=dict(system_message=system_message, user_text=user_text, chat_history=filtered_history))
-    # Remove any extra markers if present.
     response = response.split("HAL:")[-1].strip()
-    
+
     # Fallback in case the generated answer is empty
     if not response:
         response = "Certainly, here is an in-depth explanation: [Fallback explanation]."
-    
+
     chat_history.append({'role': 'user', 'content': user_text})
     chat_history.append({'role': 'assistant', 'content': response})
-    
+
+    # 🔍 FIX: Only override if strongly negative and NOT a question
     if sentiment == "NEGATIVE" and not user_text.strip().endswith("?"):
         response = "I'm sorry you're feeling this way. I'm here to help. What can I do to assist you further?"
         chat_history[-1]['content'] = response
-    
-    follow_up = generate_follow_up(user_text)
-    chat_history.append({'role': 'assistant', 'content': follow_up})
-    
+        follow_up = None  # 🚨 Don't generate follow-up if negative sentiment triggers a different message
+    else:
+        follow_up = generate_follow_up(user_text)
+        chat_history.append({'role': 'assistant', 'content': follow_up})
+
     return response, follow_up, chat_history, None
 
+
 # --- Chat UI ---
 st.title("🚀 HAL - Your NASA AI Assistant")
 st.markdown("🌌 *Ask me about space, NASA, and beyond!*")