Update app.py

app.py

@@ -105,10 +105,6 @@ 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})
 
-    # ✅ Removes unnecessary sentiment apology
-    if sentiment == "NEGATIVE":
-        response += ""  # Keeps response normal instead of adding "I'm sorry to hear that"
-
     follow_up = generate_follow_up(user_text)
     chat_history.append({'role': 'assistant', 'content': follow_up})
 
@@ -116,77 +112,4 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256):
 
 # --- Chat UI ---
 st.title("🚀 HAL - Your NASA AI Assistant")
-st.markdown("🌌 *Ask me about space, NASA, and beyond!*")
-
-# Sidebar: Reset Chat
-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
-st.markdown("""
-    <style>
-    .user-msg {
-        background-color: #0078D7;
-        color: white;
-        padding: 10px;
-        border-radius: 10px;
-        margin-bottom: 5px;
-        width: fit-content;
-        max-width: 80%;
-    }
-    .assistant-msg {
-        background-color: #333333;
-        color: white;
-        padding: 10px;
-        border-radius: 10px;
-        margin-bottom: 5px;
-        width: fit-content;
-        max-width: 80%;
-    }
-    .container {
-        display: flex;
-        flex-direction: column;
-        align-items: flex-start;
-    }
-    @media (max-width: 600px) {
-        .user-msg, .assistant-msg { font-size: 16px; max-width: 100%; }
-    }
-    </style>
-""", unsafe_allow_html=True)
-
-# 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...")  # Only ONE chat_input()
-
-if 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
-    )
-
-    st.markdown(f"<div class='assistant-msg'><strong>HAL:</strong> {response}</div>", unsafe_allow_html=True)
-
-    if image_url:
-        st.image(image_url, caption="NASA Image of the Day")
-
-    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.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
+st