Update app.py

app.py

@@ -22,15 +22,56 @@ os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
 
 llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.2)
 
-# Prompt templates
+# ✅ Prompt templates
 templates = {
     "flashcard": PromptTemplate(
         input_variables=["context", "query"],
         template="""
 {context}
 
-Create 5 flashcards based on: "{query}"
-Each card should have a question and short answer.
+Create 5 flashcards based on the topic: "{query}"
+Each flashcard should include:
+- A clear question
+- A short answer
+Focus on high school geometry understanding.
+"""
+    ),
+    "lesson plan": PromptTemplate(
+        input_variables=["context", "query"],
+        template="""
+Given the following retrieved SOL text:
+{context}
+
+Generate a Geometry lesson plan based on: "{query}"
+Include:
+1. Simple explanation of the concept.
+2. Real-world example.
+3. Engaging class activity.
+Be concise and curriculum-aligned for high school.
+"""
+    ),
+    "worksheet": PromptTemplate(
+        input_variables=["context", "query"],
+        template="""
+{context}
+
+Create a student worksheet for: "{query}"
+Include:
+- Concept summary
+- A worked example
+- 3 practice problems
+"""
+    ),
+    "proofs": PromptTemplate(
+        input_variables=["context", "query"],
+        template="""
+{context}
+
+Generate a proof-focused geometry lesson plan for: "{query}"
+Include:
+- Student-friendly explanation
+- Real-world connection
+- One short class activity
 """
     )
 }