Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,7 @@ import os
 QUESTION = "Compute the integral of f(x) = x^2."
 MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 hf_token = os.getenv("HF_TOKEN")
+SUBJECT = "Calculus BC"
 
 # Check if HF token is set
 if not hf_token:
@@ -28,6 +29,31 @@ def get_llm(model_id=MODEL, max_new_tokens=130, temperature=0.7):
 # create llm
 llm = get_llm()
 
+#  prompts
+prompt = f"""
+    You are an AI assistant designed to support high school students in the subject of {SUBJECT}. 
+    Your role is to offer friendly, helpful, and in-depth guidance, just like a supportive teacher would.
+    
+    Please follow these guidelines:
+    
+    1. Maintain a polite, respectful, and professional tone at all times.
+    2. Adhere to ethical principles — do not promote cheating, harmful behavior, or misinformation.
+    3. Interact in a warm, encouraging, and student-centered style — use clear explanations, positive reinforcement, and examples when needed.
+"""
+
+p_explanation = """
+    4. Focus on thoroughly explaining the question by breaking down its components. Clarify the key concepts and definitions involved, ensuring that the explanation helps the reader fully understand what the question is asking. Avoid jumping to answers or examples; instead, concentrate on making the meaning and scope of the question clear.
+    5. Do not include specific examples or real-world applications in your response.
+"""
+p_example = """
+    4. Focus on providing three distinct examples that illustrate different aspects or variations of the question. Each example should highlight a unique approach or scenario related to the topic, helping to clarify the concept from multiple perspectives.
+    5. Do not include any explanation or real-world applications in your response.
+"""
+p_application = """
+    4. Provide two clear and relevant real-world applications related to the question or topic. Explain how each application connects to the concepts being discussed, demonstrating practical uses or implications.
+    5. Do not include any explanation or examples in your response.
+"""
+
 # Initialize session state
 if "help_clicks" not in st.session_state:
     st.session_state.help_clicks = 0
@@ -74,22 +100,37 @@ with st.container():
 
             with col1:
                 if st.button("📝 Explain the question"):
-                    prompt = f"[INST]You are a thoughtful AI assistant.\nUser: {QUESTION} [/INST]\nAI:"
+                    prompt = (
+                        "[INST]<<SYS>>\n"
+                        f"{prompt + p_explanation}\n"
+                        "<</SYS>>\n\n"
+                        f"{QUESTION}\n"
+                        "[/INST]"
+                    )
                     st.session_state.response = llm.invoke(prompt)
-
-                    print("Model Response:", st.session_state.response)
-
                     
                     st.session_state.button_clicked = "Explain the question"
             with col2:
                 if st.button("💡 Give an example"):
-                    prompt = f"[INST]You are a thoughtful AI assistant.\nUser: {QUESTION} [/INST]\nAI:"
+                    prompt = (
+                        "[INST]<<SYS>>\n"
+                        f"{prompt + p_example}\n"
+                        "<</SYS>>\n\n"
+                        f"{QUESTION}\n"
+                        "[/INST]"
+                    )
                     st.session_state.response = llm.invoke(prompt)
                     
                     st.session_state.button_clicked = "Give an example"
             with col3:
                 if st.button("🤔 Who cares?"):
-                    prompt = f"[INST]You are a thoughtful AI assistant.\nUser: {QUESTION} [/INST]\nAI:"
+                    prompt = (
+                        "[INST]<<SYS>>\n"
+                        f"{prompt + p_application}\n"
+                        "<</SYS>>\n\n"
+                        f"{QUESTION}\n"
+                        "[/INST]"
+                    )
                     st.session_state.response = llm.invoke(prompt)
                     
                     st.session_state.button_clicked = "Who cares?"