import dramatiq
import json
import time
from openai import OpenAI
from django.db import transaction
from django.utils import timezone
from utils.shortcuts import get_env, DRAMATIQ_WORKER_ARGS
from .models import FlowchartSubmission, FlowchartSubmissionStatus

@dramatiq.actor(**DRAMATIQ_WORKER_ARGS(max_retries=3))
def evaluate_flowchart_task(submission_id):
    """异步AI评分任务"""
    try:
        submission = FlowchartSubmission.objects.get(id=submission_id)
        
        # 更新状态为处理中
        submission.status = FlowchartSubmissionStatus.PROCESSING
        submission.save()
        
        start_time = time.time()
        
        # 使用固定评分标准
        system_prompt = build_evaluation_prompt(submission.problem)
        
        # 构建用户提示词，包含标准答案对比
        user_prompt = f"""
        请对以下Mermaid流程图进行评分：
        
        学生提交的流程图：
        ```mermaid
        {submission.mermaid_code}
        ```

        标准答案参考：
        ```mermaid
        {submission.problem.mermaid_code}
        ```
        """
        # 如果有流程图提示，添加到提示词中
        if submission.problem.flowchart_hint:
            user_prompt += f"""
        
        设计提示：{submission.problem.flowchart_hint}
        """
        
        user_prompt += """
        
        请按照评分标准进行详细评估，并给出0-100的分数。
        """
        
        # 调用AI进行评分
        api_key = get_env("AI_KEY")
        if not api_key:
            raise Exception("AI_KEY is not set")
            
        client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")
        
        response = client.chat.completions.create(
            model="deepseek-chat",
            messages=[
                {"role": "system", "content": system_prompt},
                {"role": "user", "content": user_prompt}
            ],
            temperature=0.3,
        )
        
        ai_response = response.choices[0].message.content
        score_data = parse_ai_evaluation_response(ai_response)
        
        processing_time = time.time() - start_time
        
        # 保存评分结果
        with transaction.atomic():
            submission.ai_score = score_data['score']
            submission.ai_grade = score_data['grade']
            submission.ai_feedback = score_data['feedback']
            submission.ai_suggestions = score_data.get('suggestions', '')
            submission.ai_criteria_details = score_data.get('criteria_details', {})
            submission.ai_provider = 'deepseek'
            submission.ai_model = 'deepseek-chat'
            submission.processing_time = processing_time
            submission.status = FlowchartSubmissionStatus.COMPLETED
            submission.evaluation_time = timezone.now()
            submission.save()
        
        # 推送评分完成通知
        from utils.websocket import push_flowchart_evaluation_update
        push_flowchart_evaluation_update(
            submission_id=str(submission.id),
            user_id=submission.user_id,
            data={
                "type": "flowchart_evaluation_completed",
                "score": score_data['score'],
                "grade": score_data['grade'],
            }
        )
        
    except Exception as e:
        # 处理失败
        submission.status = FlowchartSubmissionStatus.FAILED
        submission.save()
        
        # 推送错误通知
        from utils.websocket import push_flowchart_evaluation_update
        push_flowchart_evaluation_update(
            submission_id=str(submission.id),
            user_id=submission.user_id,
            data={
                "type": "flowchart_evaluation_failed",
                "submission_id": str(submission.id),
                "error": str(e)
            }
        )
        raise e

def build_evaluation_prompt(problem):
    """构建AI评分提示词 - 使用固定标准"""
    
    # 使用固定的评分标准
    criteria_text = """
- 逻辑正确性 (权重: 1.0, 最高分: 40): 检查流程图的逻辑是否正确，包括条件判断、循环结构等
- 完整性 (权重: 0.8, 最高分: 30): 检查流程图是否包含所有必要的步骤和分支
- 规范性 (权重: 0.6, 最高分: 20): 检查流程图符号使用是否规范，是否符合标准
- 清晰度 (权重: 0.4, 最高分: 10): 评估流程图的整体布局和连线情况（不用考虑节点ID是否复杂）
"""
    
    return f"""
你是一个专业的编程教学助手，负责评估学生提交的Mermaid流程图。

评分标准：
{criteria_text}

评分要求：
1. 仔细分析流程图的逻辑正确性、完整性和清晰度
2. 检查是否涵盖了题目的所有要求
3. 评估流程图的规范性和可读性
4. 给出0-100的分数
5. 提供详细的反馈和改进建议

评分等级：
- S级 (90-100分): 优秀，逻辑清晰，完全符合要求
- A级 (80-89分): 良好，基本符合要求，有少量改进空间
- B级 (70-79分): 及格，基本正确但存在一些问题
- C级 (0-69分): 需要改进，存在明显问题

请以JSON格式返回评分结果：
{{
    "score": 85,
    "grade": "A",
    "feedback": "详细的反馈内容",
    "suggestions": "改进建议",
    "criteria_details": {{
        "逻辑正确性": {{"score": 35, "max": 40, "comment": "逻辑基本正确"}},
        "完整性": {{"score": 25, "max": 30, "comment": "缺少部分步骤"}},
        "规范性": {{"score": 18, "max": 20, "comment": "符号使用规范"}},
        "清晰度": {{"score": 8, "max": 10, "comment": "布局清晰"}}
    }}
}}
"""

def parse_ai_evaluation_response(ai_response):
    """解析AI评分响应"""
    try:
        import re
        json_match = re.search(r'\{.*\}', ai_response, re.DOTALL)
        if json_match:
            data = json.loads(json_match.group())
        else:
            data = {
                "score": 60,
                "grade": "C",
                "feedback": "AI评分解析失败，请重新提交",
                "suggestions": "",
                "criteria_details": {}
            }
        return data
    except Exception:
        return {
            "score": 60,
            "grade": "C", 
            "feedback": "AI评分解析失败，请重新提交",
            "suggestions": "",
            "criteria_details": {}
        }