Update app.py

app.py

@@ -100,8 +100,17 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256):
 # Streamlit UI Setup
 st.set_page_config(page_title="NASA ChatBot", page_icon="🚀")
 
-st.title("🚀 NASA AI ChatBot")
-st.markdown("🌌 *Powered by Hugging Face & NASA APIs!*")
+st.title("🚀 HAL")
+# Chat Display with Updated Styling
+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>Bot:</strong> {message['content']}</div>", unsafe_allow_html=True)
+
+st.markdown("</div>", unsafe_allow_html=True)
 
 # Custom CSS for chat styling
 st.markdown("""