Update app.py

app.py

@@ -2,11 +2,11 @@ import streamlit as st
 from dotenv import load_dotenv
 import os
 from langchain_openai import ChatOpenAI
-from langchain.schema import SystemMessage, HumanMessage, AIMessage
+from langchain.schema import SystemMessage, HumanMessage
 from langchain.memory import ConversationBufferMemory
-from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain.chains import ConversationChain
 
-# ✅ Load environment variables
+# ✅ Load API keys
 load_dotenv("apiroute.env")
 api_key = os.getenv("OPENAI_API_KEY")
 api_base = os.getenv("OPENAI_API_BASE")
@@ -21,50 +21,58 @@ llm = ChatOpenAI(
 if "memory" not in st.session_state:
     st.session_state.memory = ConversationBufferMemory(return_messages=True)
 
-# ✅ Setup prompt with system message only once
-prompt = ChatPromptTemplate.from_messages([
-    SystemMessage(
-        content=(
-            "You are Gremmy, a helpful, friendly AI chatbot created by Frederick using Google's Gemma model. "
-            "You are smart, polite, and informative.\n"
-            "- Introduce yourself as Gremmy when needed.\n"
-            "- Speak in a concise and helpful tone.\n"
-            "- For more information about your creator, visit: https://www.linkedin.com/in/j-frederick-paul-35801a179/"
-        )
-    ),
-    MessagesPlaceholder(variable_name="chat_history"),
-    HumanMessage(content="{input}")
-])
+# ✅ Setup flag to inject system prompt only once
+if "intro_done" not in st.session_state:
+    st.session_state.intro_done = False
+
+# ✅ Create system message describing the assistant
+system_message = SystemMessage(
+    content=(
+        "You are Gremmy, a helpful, friendly AI chatbot created by Frederick using Google's Gemma model. "
+        "You are smart, polite, and informative.\n"
+        "- Introduce yourself as Gremmy when needed.\n"
+        "- Speak in a concise and helpful tone.\n"
+        "- For more information about your creator, visit: https://www.linkedin.com/in/j-frederick-paul-35801a179/"
+    )
+)
+
+# ✅ Setup ConversationChain (adds memory)
+chat = ConversationChain(
+    llm=llm,
+    memory=st.session_state.memory,
+    verbose=False
+)
 
 # ✅ Streamlit UI
 st.set_page_config(page_title="Chatbot", layout="centered")
 st.title("💬 Chat with Gremmy")
 
-# ✅ Initialize chat history
+# ✅ Initialize history
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# ✅ Display chat history
+# ✅ Display history
 for sender, msg in st.session_state.history:
     st.markdown(f"**{sender}:** {msg}")
 
-# ✅ Input form
+# ✅ Input box
 with st.form(key="chat_form", clear_on_submit=True):
     user_input = st.text_input("Talk to me", key="user_message")
     submitted = st.form_submit_button("Send")
 
-# ✅ Generate response
+# ✅ Handle message
 if submitted and user_input:
-    # Add to history
+    # Inject system message only once
+    if not st.session_state.intro_done:
+        st.session_state.memory.chat_memory.add_message(system_message)
+        st.session_state.intro_done = True
+
+    # Add user message
     st.session_state.history.append(("You", user_input))
-    st.session_state.memory.chat_memory.add_message(HumanMessage(content=user_input))
 
-    # Format and run the prompt
-    full_prompt = prompt.format_messages(
-        chat_history=st.session_state.memory.chat_memory.messages,
-        input=user_input
-    )
-    response = llm(full_prompt)
-    st.session_state.history.append(("Gremmy", response.content))
-    st.session_state.memory.chat_memory.add_message(AIMessage(content=response.content))
+    # Get response from Gremmy
+    response = chat.run(user_input)
+
+    # Add Gremmy response
+    st.session_state.history.append(("Gremmy", response.strip()))
     st.rerun()