from django.contrib import admin
from .models import TrainingRecord, TrainingScoreDetail


@admin.register(TrainingRecord)
class TrainingRecordAdmin(admin.ModelAdmin):
    list_display = [
        'id', 'user', 'case', 'training_mode',
        'total_score', 'evaluation_level', 'status',
        'start_time', 'duration_seconds'
    ]
    list_filter = ['training_mode', 'evaluation_level', 'status']
    search_fields = ['user__real_name', 'case__title', 'feedback']
    ordering = ['-start_time']


@admin.register(TrainingScoreDetail)
class TrainingScoreDetailAdmin(admin.ModelAdmin):
    list_display = ['id', 'record', 'dimension', 'score', 'ai_confidence']
    list_filter = ['dimension']
    search_fields = ['record__user__real_name', 'dimension']