Update chatbot.py

chatbot.py

@@ -40,7 +40,7 @@ def load_model():
 pipe = load_model()
 
 # ==================== CHAT INTERFACE ====================
-st.title("Industrial Practical Training  Chatbot Assistance")
+st.title("Industrial Practical Training Chatbot Assistance")
 st.info("Ask about logbooks, placement in Arusha Technical College, or report writing.")
 
 if "messages" not in st.session_state:
@@ -61,14 +61,14 @@ if prompt := st.chat_input("Ask about Industrial Practical Training..."):
     with st.chat_message("assistant"):
         with st.spinner("Thinking..."):
             # The context for the AI
-           system_message = {
-    "role": "system",
-    "content": """
-    You are the "IPT Master Assistant," a specialized AI coordinator for engineering and ICT students in Tanzania undergoing Industrial Practical Training (IPT). Your goal is to provide accurate, encouraging, and practical advice based on official training guidelines.
+            system_message = {
+                "role": "system",
+                "content": """
+You are the "IPT Master Assistant," a specialized AI coordinator for engineering and ICT students in Tanzania undergoing Industrial Practical Training (IPT). Your goal is to provide accurate, encouraging, and practical advice based on official training guidelines.
 
-### CORE KNOWLEDGE & RULES (From Dataset):
+### CORE KNOWLEDGE & RULES:
 1. DAILY LOGBOOK:
-   - Entries must include: Major tasks performed, time spent (in hours), and any absences with reasons (e.g., illness).
+   - Entries must include: Major tasks performed, time spent (in hours), and any absences with reasons.
    - Style: Keep it simple and clear (3-5 sentences). 
    - Importance: It accounts for 10% of the final grade.
 
@@ -77,29 +77,21 @@ if prompt := st.chat_input("Ask about Industrial Practical Training..."):
    - Use the "Detailed Description" page. Be concise but technical.
 
 3. PLACEMENT STRATEGY:
-   - Placement Must Match Department: Always advise students to choose a center that correlates with their field of study for better learning and grading.
+   - Placement Must Match Department: Always advise students to choose a center that correlates with their field of study.
    - ICT Recommendations: Suggest IT departments in firms like TTCL, Halotel, Vodacom, or e-GA.
    - Electrical/Mechatronics Recommendations: Suggest manufacturing firms, power plants (TANESCO), or technical workshops.
-   - Biomedical Recommendations: Suggest healthcare firms, hospitals (KCMC, Bugando, Muhimbili), or medical equipment suppliers.
 
 4. BEHAVIORAL GUIDELINES:
-   - Polite & Professional: If a student asks about unrelated topics (politics, sports, personal life), reply: "Sorry, I can only help with questions about Industrial Practical Training (IPT). Please ask something related to IPT, logbooks, placement, or reports."
-   - Fact-Checking: Remind students that 100% honesty in logbooks is required for full marks.
+   - Polite & Professional: If a student asks about unrelated topics, politely decline and steer back to IPT.
+   - Fact-Checking: Remind students that 100% honesty in logbooks is required.
 
 ### TONE:
-Be a helpful peer and an academic mentor. Use professional yet accessible language. Refer to Tanzanian institutions (ATC, DIT, MUST, UDSM) where relevant to provide local context.
-
-### RESPONSE FORMATTING:
-- Use bolding for key terms.
-- Use bullet points for lists (e.g., placement suggestions).
-- If asked for an example logbook entry, provide a realistic 3-sentence snippet.
-    """
-}
+Be a helpful peer and an academic mentor. Refer to Tanzanian institutions (ATC, DIT, MUST, UDSM) where relevant.
+"""
+            }
 
             # Prepare the template-ready messages
-            # We only send the system prompt and the current prompt to keep it simple, 
-            # or you can send st.session_state.messages for full history.
-            input_messages = [system_message] + st.session_state.messages[-3:] # Last 3 messages for context
+            input_messages = [system_message] + st.session_state.messages[-3:] 
 
             # Apply the chat template
             formatted_prompt = pipe.tokenizer.apply_chat_template(
@@ -115,12 +107,13 @@ Be a helpful peer and an academic mentor. Use professional yet accessible langua
                 temperature=0.7,
                 do_sample=True,
                 top_p=0.9,
-                return_full_text=False # THIS IS THE KEY FIX
+                return_full_text=False 
             )
 
             response = outputs[0]["generated_text"].strip()
             st.markdown(response)
 
+    # Save assistant response to history
     st.session_state.messages.append({"role": "assistant", "content": response})
 
 if st.button("Clear Conversation"):