Update app.py

app.py

@@ -57,11 +57,10 @@ def generate_follow_up(user_text):
     Generates a concise and conversational follow-up question related to the user's input.
     """
     prompt_text = (
-        f"Given the user's question: '{user_text}', generate a single friendly follow-up question. "
-        "Make it short, conversational, and natural—like a human would ask. "
-        "Example: If the user asks 'What is a quark?', respond with something like "
-        "'Would you like to learn about the six types of quarks?' "
-        "Do NOT include phrases like 'A natural follow-up question could be'."
+        f"Given the user's question: '{user_text}', generate a SHORT and SIMPLE follow-up question. "
+        "Make it conversational and friendly. Example: "
+        "'Would you like to learn more about the six types of quarks?' "
+        "Do NOT provide long explanations—just ask a friendly follow-up question."
     )
 
     hf = get_llm_hf_inference(max_new_tokens=32, temperature=0.7)
@@ -105,6 +104,9 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256):
     chat_history.append({'role': 'user', 'content': user_text})
     chat_history.append({'role': 'assistant', 'content': response})
 
+    if sentiment == "NEGATIVE":
+        response = "I'm here to help. Let me know what I can do for you. 😊"
+
     follow_up = generate_follow_up(user_text)
     chat_history.append({'role': 'assistant', 'content': follow_up})
 
@@ -119,8 +121,7 @@ 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.session_state.last_topic = ""
-    st.rerun()  # 
+    st.experimental_rerun()
 
 # Custom Chat Styling
 st.markdown("""
@@ -154,33 +155,25 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# --- Chat History Display (Ensures All Messages Are Visible) ---
+# Chat History Display
 st.markdown("<div class='container'>", unsafe_allow_html=True)
-
 for message in st.session_state.chat_history:
     if message["role"] == "user":
         st.markdown(f"<div class='user-msg'><strong>You:</strong> {message['content']}</div>", unsafe_allow_html=True)
     else:
         st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {message['content']}</div>", unsafe_allow_html=True)
-
 st.markdown("</div>", unsafe_allow_html=True)
 
 # --- Single Input Box for Both Initial and Follow-Up Messages ---
-user_input = st.chat_input("Type your message here...")  # Uses Enter to submit
+user_input = st.chat_input("Type your message here...")  # Only ONE chat_input()
 
 if user_input:
-    # Save user message in chat history
-    st.session_state.chat_history.append({'role': 'user', 'content': user_input})  
-
-    # Generate HAL's 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
     )
 
-    st.session_state.chat_history.append({'role': 'assistant', 'content': response})
-
     st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
 
     if image_url:
@@ -189,7 +182,9 @@ if user_input:
     st.session_state.follow_up = follow_up
     st.session_state.response_ready = True  # Enables follow-up response cycle
 
+# Display follow-up question inside chat if available
 if st.session_state.response_ready and st.session_state.follow_up:
-    st.session_state.chat_history.append({'role': 'assistant', 'content': 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)
+
+    # Reset response state so user can type next input
     st.session_state.response_ready = False