import json

from edu_assistant.learning_tasks import CodingProblemAnalysis

# 创建一个CodingProblem实例
problem = CodingProblemAnalysis.build_coding_problem(
    title="求解最大值",
    question="请编写一个函数，该函数接收一个整数列表，并返回该列表的最大值。",
    standard_answer="def find_max(lst):\n\treturn max(lst)",
    analysis="在这个问题中，我们需要使用Python的内置函数max来找到列表的最大值。",
    language="python",
)

# 创建一个CodingAnswer实例
answer = CodingProblemAnalysis.build_coding_answer(
    answer="def find_max(lst):\n\treturn lst[0]",
)

analysis = CodingProblemAnalysis(lang="python")
result = analysis.start_analysis(problem, answer)
print(json.dumps(result, ensure_ascii=False, indent=4))

next_question = "如果改成返回最小值呢？"
result = analysis.ask(next_question, result["session_id"])
print(json.dumps(result, ensure_ascii=False, indent=4))