Update app.py

app.py

@@ -103,7 +103,6 @@ def generate_follow_up(user_text):
 
     return cleaned_output
 
-# ✅ Main Response Function
 # ✅ Main Response Function
 def get_response(system_message, user_text, max_new_tokens=800):
     """
@@ -207,6 +206,16 @@ if st.sidebar.button("Reset Chat"):
     st.session_state.response_ready = False
     st.session_state.follow_up = ""
 
+# ✅ Display Chat History
+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)
+
+
 # ✅ Chat UI
 user_input = st.chat_input("Type your message here...")