import asyncio
from orchestrator import orchestrator

async def run_sanity():
    print("--- 🟢 SANITY CHECK ALGEBRA (SMADAR) ---")
    res_alg = await orchestrator.smart_solve(
        problem_text="סמדר קנתה מחברות חשבון ב-4 שקלים וחלקות ב-5 שקלים. כמה קנתה?",
        data_anchor={"function_equations": ["4x + 5y = 120", "y = x + 6"]},
        grade="ז'",
        category="ALGEBRA",
        image_data=None,
        ambiguity_warning=False
    )
    print("ALGEBRA RESPONSE SUCCESS:")
    print("Final Answer:", res_alg.get("final_answer"))
    print("Logic Error:", res_alg.get("logic_error"))

    print("\n--- 🟢 SANITY CHECK GEOMETRY (CIRCLE AREA) ---")
    res_geom = await orchestrator.smart_solve(
         problem_text="חשב את שטח המעגל שמשוואתו (x-3)**2 + (y-4)**2 = 25",
         data_anchor={"function_equations": ["(x-3)**2 + (y-4)**2 = 25"]},
         grade="י'",
         category="GEOMETRY",
         image_data=None,
         ambiguity_warning=False
    )
    print("GEOMETRY RESPONSE SUCCESS:")
    print("Final Answer:", res_geom.get("final_answer"))
    print("Logic Error:", res_geom.get("logic_error"))

if __name__ == "__main__":
    asyncio.run(run_sanity())