Update chatbot.py

chatbot.py

@@ -41,7 +41,7 @@ pipe = load_model()
 # ==================== CHAT INTERFACE ====================
 st.title("IPTCHATBOT ASSISTANCE")
 
-st.info("Small fine-tuned model (1.7B). Answers may vary — verify facts.")
+st.info(" Answers may vary verify facts.")
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -52,7 +52,7 @@ for message in st.session_state.messages:
 
 # ... (model loading code above remains unchanged)
 
-if prompt := st.chat_input("Ask about IPT, ICT, or anything..."):
+if prompt := st.chat_input("Ask about Industrial Practical Training..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
@@ -60,11 +60,12 @@ if prompt := st.chat_input("Ask about IPT, ICT, or anything..."):
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
             # 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."
+            system_prompt = system_prompt = """
+            You are a helpful assistant ONLY for questions about Industrial Practical Training (IPT) 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.
+            If the question is not about IPT, logbooks, reports, placement, or related topics, reply exactly:
+            "Sorry, I can only answer questions about Industrial Practical Training (IPT). Please ask something related."
             """
 
             # Build messages with system prompt first