Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import pickle
 import streamlit as st
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationalRetrievalChain
@@ -68,25 +67,60 @@ qa_chain = ConversationalRetrievalChain.from_llm(
 )
 
 # Streamlit app
+st.set_page_config(page_title="AI Course Recommendation Chatbot", page_icon=":book:")
 st.title("AI Course Recommendation Chatbot")
 
+# Custom CSS for styling
+st.markdown("""
+<style>
+    .chat-container {
+        max-width: 800px;
+        margin: auto;
+        padding: 20px;
+        border-radius: 10px;
+        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.1);
+        background-color: #f9f9f9;
+    }
+    .user-message {
+        background-color: #d1e7dd;
+        border-radius: 10px;
+        padding: 10px;
+        margin: 5px 0;
+        text-align: left;
+    }
+    .bot-message {
+        background-color: #e2e3e5;
+        border-radius: 10px;
+        padding: 10px;
+        margin: 5px 0;
+        text-align: left;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Initialize chat history in session state
 if 'chat_history' not in st.session_state:
     st.session_state.chat_history = []
 
+# User input for course interests
 user_query = st.text_input("What are you looking to learn?", "")
 
+# Button to get course recommendations
 if st.button("Get Recommendation"):
     if user_query:
         response = qa_chain({"question": user_query})
         recommendation = response["answer"]
         
-        # Update chat history
+        # Update chat history with user's query and recommendation
         st.session_state.chat_history.append({"user": user_query, "bot": recommendation})
 
-# Display chat history
-for chat in st.session_state.chat_history:
-    st.write(f"**You:** {chat['user']}")
-    st.write(f"**HONEY BEE:** {chat['bot']}")
+# Display chat history in a styled format
+with st.container():
+    st.markdown('<div class="chat-container">', unsafe_allow_html=True)
+    for chat in st.session_state.chat_history:
+        st.markdown(f'<div class="user-message"><strong>You:</strong> {chat["user"]}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="bot-message"><strong>AI Recommendation:</strong> {chat["bot"]}</div>', unsafe_allow_html=True)
+    st.markdown('</div>', unsafe_allow_html=True)
 
 # Optional: Add a button to clear the chat history
 if st.button("Clear Chat History"):