Update app.py

app.py

@@ -2,77 +2,65 @@ import streamlit as st
 from dotenv import load_dotenv
 import os
 from langchain_openai import ChatOpenAI
-from langchain.schema import SystemMessage, HumanMessage
+from langchain.schema import SystemMessage
 from langchain.memory import ConversationBufferMemory
 from langchain.chains import ConversationChain
 
-# ✅ Load API keys
+# ✅ Load environment variables
 load_dotenv("apiroute.env")
 api_key = os.getenv("OPENAI_API_KEY")
 api_base = os.getenv("OPENAI_API_BASE")
 
-# ✅ Initialize LLM
+# ✅ Setup LLM
 llm = ChatOpenAI(
     model_name="google/gemma-3n-e2b-it:free",
-    temperature=0.7
+    temperature=0.7,
 )
 
-# ✅ Setup memory
+# ✅ Memory setup
 if "memory" not in st.session_state:
     st.session_state.memory = ConversationBufferMemory(return_messages=True)
 
-# ✅ 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/"
-    )
+# ✅ System prompt
+system_prompt = (
+    "You are Gremmy, a helpful, friendly AI chatbot created by Frederick using Google's Gemma model.\n"
+    "- Introduce yourself **only once**, unless asked.\n"
+    "- Be concise, polite, and informative.\n"
+    "- Avoid repeating your background repeatedly.\n"
+    "- For more about your creator: https://www.linkedin.com/in/j-frederick-paul-35801a179/"
 )
 
-# ✅ Setup ConversationChain (adds memory)
+# ✅ Inject system message only once
+if "system_added" not in st.session_state:
+    st.session_state.memory.chat_memory.add_message(SystemMessage(content=system_prompt))
+    st.session_state.system_added = True
+
+# ✅ Setup conversation chain
 chat = ConversationChain(
     llm=llm,
     memory=st.session_state.memory,
     verbose=False
 )
 
-# ✅ Streamlit UI
+# ✅ Page config and title
 st.set_page_config(page_title="Chatbot", layout="centered")
 st.title("💬 Chat with Gremmy")
 
-# ✅ Initialize history
+# ✅ Track and display history
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# ✅ Display history
 for sender, msg in st.session_state.history:
     st.markdown(f"**{sender}:** {msg}")
 
-# ✅ Input box
+# ✅ Input UI
 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")
 
-# ✅ Handle message
+# ✅ Handle submission
 if submitted and user_input:
-    # 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))
-
-    # Get response from Gremmy
     response = chat.run(user_input)
-
-    # Add Gremmy response
     st.session_state.history.append(("Gremmy", response.strip()))
     st.rerun()