Update server.py

server.py

@@ -18,21 +18,134 @@ def get_local_resource():
 
 @mcp.tool()
 async def generate_chapter_summary(chapter_name: str) -> str:
-    """Provides source material and instructions for a math summary."""
+    """
+    Provides source material + strict instructions for a math chapter summary.
+    """
     raw_data = get_local_resource()
+
     return f"""
-    <source_material>
-    {raw_data[:4000]} 
-    </source_material>
-    SYSTEM_INSTRUCTION: You are a Mathematics Expert. 
-    1. Summarize in 3 bullets.
-    2. List formulas like $x = \\frac{{-b \\pm \\sqrt{{b^2-4ac}}}}{{2a}}$.
-    """
+You are a Mathematics Expert and Academic Tutor.
+
+TASK:
+Generate a structured summary of the chapter titled: "{chapter_name}"
+
+IMPORTANT RULES:
+- Use ONLY the provided <source_material>.
+- If something is missing in the material, explicitly write: "Not found in source".
+- Do NOT invent formulas, theorems, or examples.
+- Prefer mathematical notation and LaTeX formatting.
+- Output must be clean Markdown.
+
+OUTPUT FORMAT (STRICT):
+## Chapter: {chapter_name}
+
+### 1. 3-Line High-Level Summary (3 bullets max)
+- ...
+- ...
+- ...
+
+### 2. Key Concepts (Definitions / Theorems / Properties)
+List important definitions and results.
+Each bullet must follow:
+- **Concept Name**: Explanation (1–2 lines)
+
+### 3. Important Formulas (LaTeX only)
+Provide all important formulas mentioned in the chapter.
+Format each as:
+- **Formula Name**: $...$
+
+### 4. Step-by-Step Core Method(s)
+If the chapter contains any algorithm/method (example: solving quadratic, integration steps),
+list the procedure in numbered steps.
+
+### 5. Common Mistakes / Traps
+List 3–6 common mistakes students make.
+
+### 6. Quick Worked Example (If present in source)
+- Problem:
+- Solution (brief but clear):
+- Final Answer:
+
+If no example exists in the source, write: "Not found in source".
+
+SOURCE:
+<source_material>
+{raw_data[:10000]}
+</source_material>
+"""
+
 
 @mcp.tool()
 async def generate_quiz(chapter_name: str, difficulty: str = "medium") -> str:
+    """
+    Generates quiz questions based on chapter material.
+    """
     raw_data = get_local_resource()
-    return f"Material: {raw_data[:2000]}\nDifficulty: {difficulty}\nCreate a 3-question quiz."
+
+    return f"""
+You are a Mathematics Professor designing an exam.
+
+CHAPTER: "{chapter_name}"
+DIFFICULTY: "{difficulty}"
+
+DIFFICULTY RULES:
+- easy: direct definition/formula substitution questions
+- medium: multi-step numeric problems + conceptual reasoning
+- hard: tricky reasoning, proofs, edge cases, mixed-concept questions
+
+IMPORTANT RULES:
+- Use ONLY the provided source material.
+- Do NOT invent topics not present in the material.
+- Every question must match the difficulty level.
+- Provide answer key + short solution steps.
+- Format must be clean Markdown.
+- Use LaTeX for all formulas.
+
+OUTPUT FORMAT (STRICT):
+
+## Quiz: {chapter_name} ({difficulty})
+
+### Q1. (Conceptual / Definition Based)
+Question text...
+
+**Options (if MCQ):**
+A) ...
+B) ...
+C) ...
+D) ...
+
+### Q2. (Formula/Application Based)
+Question text...
+
+### Q3. (Multi-step Problem / Reasoning Based)
+Question text...
+
+---
+
+## Answer Key + Solutions
+
+### Q1 Solution
+- Correct Answer: ...
+- Explanation: ...
+
+### Q2 Solution
+Step 1: ...
+Step 2: ...
+Final Answer: ...
+
+### Q3 Solution
+Step 1: ...
+Step 2: ...
+Step 3: ...
+Final Answer: ...
+
+SOURCE:
+<source_material>
+{raw_data[:10000]}
+</source_material>
+"""
+
+
 
 if __name__ == "__main__":
     print("Starting FastMCP server on port 7860...")