Update app.py

app.py

@@ -74,23 +74,24 @@ Include:
 - One short class activity
 """
     ),
-"general question": PromptTemplate(
-    input_variables=["context", "query"],
-    template="""
-You are a Virginia Geometry Standards of Learning (SOL) assistant.
+   "general question": ChatPromptTemplate.from_messages([
+        HumanMessagePromptTemplate.from_template(
+            """
+You are a Virginia Geometry SOL assistant.
 
-From the following standards content:
+From the following SOL context:
 {context}
 
-Identify the SOL standard (e.g., G.RLT.1, G.TR.3) that most accurately answers the user's question: "{query}"
+Identify the SOL standard (e.g., G.RLT.1) that best matches this query: "{query}"
 
 Respond with:
-1. The SOL code (e.g., G.RLT.3)
-2. The **exact standard description** for that code, as written in the context
+1. The exact SOL code (e.g., G.RLT.1)
+2. The exact description line from the SOL guide
 
-Do not summarize. Copy the official description exactly.
+Do not summarize. Only copy from the context.
 """
-)
+        )
+    ])