Update chatbot.py

chatbot.py

@@ -50,22 +50,41 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-if prompt := st.chat_input("Ask about Industrial Practical Training..."):
+# ... (model loading code above remains unchanged)
+
+if prompt := st.chat_input("Ask about IPT, ICT, or anything..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
 
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
-            chat_messages = [{"role": "user", "content": prompt}]
-            outputs = pipe(chat_messages, max_new_tokens=300, temperature=0.7, do_sample=True, top_p=0.9)
+            # Define the system prompt here
+            system_prompt = """
+            You are a knowledgeable assistant for ICT and engineering students in Tanzania.
+            IPT means Industrial Practical Training (Industrial Attachment) — a mandatory work placement for students to gain real-world experience.
+            Always use this correct meaning unless the context clearly means something else.
+            If unsure, say: "In most educational contexts in Tanzania, IPT stands for Industrial Practical Training."
+            """
+
+            # Build messages with system prompt first
+            chat_messages = [
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": prompt}
+            ]
+
+            outputs = pipe(
+                chat_messages,
+                max_new_tokens=300,
+                temperature=0.7,
+                do_sample=True,
+                top_p=0.9
+            )
+
             response = outputs[0]["generated_text"]
             if isinstance(response, str) and response.startswith(prompt):
                 response = response[len(prompt):].strip()
-            st.markdown(response)
 
-    st.session_state.messages.append({"role": "assistant", "content": response})
+            st.markdown(response)
 
-if st.button("Clear Chat"):
-    st.session_state.messages = []
-    st.rerun()
+    st.session_state.messages.append({"role": "assistant", "content": response})