import gradio as gr
from transformers import pipeline

# 加载作文评分模型（替换为你的模型名）
chef = pipeline("text-classification", model="hl250/roberta-base")


def score_essay(essay):
    """评分函数"""
    try:
        result = chef(essay)
        score = result[0]['score']

        # 根据分数给出评价
        if score >= 20:
            feedback = "🎉 优秀！内容充实，表达清晰。"
        elif score >= 15:
            feedback = "👍 良好。可以进一步丰富内容。"
        elif score >= 10:
            feedback = "💪 及格。需要改进结构和表达。"
        else:
            feedback = "📝 需要大幅提升写作水平。"

        return f"📝 作文得分: {score:.1f}/30\n\n💡 评语: {feedback}"

    except Exception as e:
        return f"❌ 评分出错: {str(e)}"


# 创建界面
demo = gr.Interface(
    fn=score_essay,
    inputs=gr.Textbox(
        lines=10,
        placeholder="请输入作文内容...",
        label="作文内容"
    ),
    outputs=gr.Textbox(label="评分结果"),
    title="🤖 AI作文评分系统",
    description="输入作文内容，获取AI自动评分和评语"
)

# 启动应用
demo.launch()