Update app.py

app.py

@@ -30,26 +30,29 @@
 #         st.write(f"Chatbot: {resp}")
 
 
-
 import os
 import streamlit as st
 from langchain.chains import ConversationChain
 from langchain_openai import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
 from streamlit_extras.add_vertical_space import add_vertical_space
-from streamlit_extras.chat_elements import message
 
-# Set OpenAI API Key (Ensure you add it in Streamlit Secrets)
+# Set up OpenAI API Key securely
 if "openai_api_key" in st.secrets:
     os.environ["OPENAI_API_KEY"] = st.secrets["openai_api_key"]
 else:
     st.error("🔑 OpenAI API Key is missing! Please add it to Streamlit secrets.")
     st.stop()
 
-# Initialize chatbot
+# Initialize chatbot with memory
 def init_chatbot(model="gpt-4o-mini"):
-    memory = ConversationBufferMemory()
-    return ConversationChain(llm=ChatOpenAI(model=model), memory=memory, verbose=False)
+    try:
+        memory = ConversationBufferMemory()
+        chatbot = ConversationChain(llm=ChatOpenAI(model=model), memory=memory, verbose=False)
+        return chatbot
+    except Exception as e:
+        st.error(f"⚠️ Error initializing chatbot: {e}")
+        return None
 
 if "chatbot" not in st.session_state:
     st.session_state.chatbot = init_chatbot()
@@ -60,16 +63,22 @@ st.markdown("""
         body {
             background-color: #f5f5f5;
         }
-        .chat-container {
-            background-color: #ffffff;
-            padding: 20px;
+        .stChatMessage {
+            padding: 10px;
             border-radius: 10px;
-            box-shadow: 0px 4px 6px rgba(0,0,0,0.1);
+            margin: 5px 0;
+        }
+        .user-message {
+            background-color: #dcf8c6;
+            text-align: right;
+        }
+        .bot-message {
+            background-color: #ffffff;
         }
     </style>
 """, unsafe_allow_html=True)
 
-# Sidebar - Settings
+# Sidebar - Model Selection
 st.sidebar.title("⚙️ Settings")
 model_choice = st.sidebar.radio("Select Model", ("gpt-4o-mini", "gpt-4", "gpt-3.5-turbo"))
 
@@ -79,43 +88,42 @@ if model_choice != st.session_state.chatbot.llm.model_name:
 
 # Title and Description
 st.title("💬 LangChain AI Chatbot")
-st.write("### Hi, I'm a chatbot built with LangChain powered by GPT. How can I assist you today?")
+st.write("### Hi, I'm a chatbot powered by GPT. How can I assist you today?")
 
-# Chat history
+# Chat history storage
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
-# User Input
-user_input = st.text_input("You:", placeholder="Ask me anything...")
+# User Input via Chat Input (better UX)
+user_input = st.chat_input("Type your message here...")
 
 # Process input
 if user_input:
     with st.spinner("Thinking..."):
-        response = st.session_state.chatbot.run(user_input)
-        if response:
-            st.session_state.chat_history.append(("user", user_input))
-            st.session_state.chat_history.append(("bot", response))
+        try:
+            response = st.session_state.chatbot.run(user_input)
+            if response:
+                st.session_state.chat_history.append(("user", user_input))
+                st.session_state.chat_history.append(("bot", response))
+        except Exception as e:
+            st.error(f"⚠️ Error generating response: {e}")
 
 # Display chat history
 st.write("### 🗨️ Conversation")
-chat_container = st.container()
-
-with chat_container:
-    for role, text in st.session_state.chat_history:
-        if role == "user":
-            message(text, is_user=True, avatar_style="thumbs")
-        else:
-            message(text, is_user=False, avatar_style="bottts")
-
-# Add some spacing
-add_vertical_space(2)
+for role, text in st.session_state.chat_history:
+    with st.chat_message(role):
+        st.markdown(f"**{role.capitalize()}**: {text}")
 
 # Collapsible Chat History
-with st.expander("📜 Chat History"):
+with st.expander("📜 View Full Chat History"):
     for role, text in st.session_state.chat_history:
         st.write(f"**{role.capitalize()}**: {text}")
 
+# Add spacing
+add_vertical_space(2)
+
 # Footer
 st.markdown("---")
 st.markdown("Developed with ❤️ using Streamlit & LangChain")
 
+