Update pages/machine_learning.py

pages/machine_learning.py

@@ -96,11 +96,14 @@ with col1:
         submit = st.form_submit_button("Send")
 
     if submit and user_input:
-        system_prompt = (
-            f"You are a machine learning mentor with {experience_label.lower()} experience. "
-            f"Answer only machine learning questions in a friendly tone and within 150 words. "
-            f"Politely inform if the question is out of scope."
-        )
+        system_prompt = f"""Act as a Machine learning mentor with {experience_label} years of experience. Teach in a friendly, approachable manner while following these strict rules:
+        1. Only answer questions related to machine learning programming (including libraries, frameworks, and tools in the machine learning ecosystem)
+        2. For any non-machine learning query, respond with exactly: "I specialize only in Machine learning programming. This appears to be a non-machine learning topic."
+        3. Never suggest you can help with non-machine learning topics
+        4. Keep explanations clear, practical, and beginner-friendly when appropriate
+        5. Include practical examples when explaining concepts
+        6. For advanced topics, assume the student has basic machine learning knowledge"""        
+       
         messages = [SystemMessage(content=system_prompt), HumanMessage(content=user_input)]
         result = ml_mentor.invoke(messages)
         st.session_state[PAGE_KEY].append((user_input, result.content))