app.py

import gradio as gr
from transformers import pipeline

classifier = pipeline("text-classification", model="WJL110/emotion-classifier")

label_map = {
    "LABEL_0": "快乐",
    "LABEL_1": "愤怒",
    "LABEL_2": "悲伤"
}

def classify_text(text):
    result = classifier(text)[0]
    emotion = label_map.get(result['label'], result['label'])
    # 返回一个字典，Gradio 会自动将其渲染为分类条形图
    return {emotion: result['score']}

demo = gr.Interface(
    fn=classify_text,
    inputs=gr.Textbox(label="输入文本"),
    outputs=gr.Label(num_top_classes=1, label="情感预测"),
    examples = [
    ["终于拿到心仪公司的Offer了，太激动了！"],          # 快乐
    ["刚才被无缘无故骂了一顿，真是莫名其妙！"],          # 愤怒
    ["看着以前的照片，眼泪止不住地流下来。"],            # 悲伤
    ["这家餐厅的味道简直太惊艳了，强烈推荐！"],          # 快乐
    ["买的东西全是坏的，商家还不退款，气炸了。"],        # 愤怒
    ["听到他生病住院的消息，我心里很难受。"],            # 悲伤
    ["这就是你所谓的负责任吗？太让人失望了。"],          # 愤怒
    ["努力了这么久还是失败了，感觉很无力。"],            # 悲伤
    ["今天天气真好，适合出去野餐，心情美美哒。"],        # 快乐
    ["排队排了两小时，结果告诉我卖完了，浪费时间！"],      # 愤怒
    ["和相恋五年的恋人分手了，曾经的海誓山盟都成了泡影，深夜里忍不住哭了！"],  # 悲伤
    ["有人故意插队还理直气壮，说了他两句还反过来骂我，太气人了！"], # 愤怒
    ["收到闺蜜送的惊喜礼物，拆开的瞬间嘴角就没下来过！"], # 快乐
    ["网购到假货，申请退货商家还拒绝，找平台投诉也没人管，太离谱了！"], # 愤怒
    ["自己种的多肉终于开花了，小小的花苞太可爱了，成就感满满！"], # 快乐
    ["爷爷走了，再也听不到他喊我的小名，再也吃不到他做的红烧肉了，好难过！"], # 悲伤
    ["外卖超时一个小时，餐品都凉透了，联系商家还态度敷衍，气炸了！"], # 愤怒
    ["工作中被同事抢功劳，明明是我做的方案，他却说是自己的，真气人！"], # 愤怒
    ["终于抢到了演唱会的门票，能去见喜欢的歌手了，激动到睡不着！"], # 快乐
    ["宠物生病了，每天都要去医院打针，看着它痛苦的样子，我心里也跟着疼！"], # 悲伤
    ["好好停在车位里的车被人刮了，肇事车主还逃逸了，太没素质了！"], # 愤怒
    ["晨跑时看到了绝美日出，朝霞染红天空，太治愈了！"], # 快乐
    ["和家人一起去游乐园玩，过山车、旋转木马，重温童年快乐！"], # 快乐
    ["有人在公共场合大声外放视频，提醒了好几次都不听，太没公德心了！"], # 愤怒
    ["朋友借了钱一直不还，催了好几次都找借口推脱，太不讲信用了！"], # 愤怒
    ["买到了超划算的超市打折商品，省了不少钱，感觉赚翻了！"], # 快乐
    ["去看了一场超搞笑的喜剧电影，全场笑个不停，心情瞬间变好！"], # 快乐
    ["因为工作调动要离开生活了十年的城市，舍不得这里的朋友和熟悉的一切！"], # 悲伤
    ["孩子第一次主动说“妈妈我爱你”，瞬间被暖化，所有辛苦都值得了！"], # 快乐
    ["培训机构卷钱跑路，交了几万块学费只上了几节课，维权无门，太坑人了！"], # 愤怒
    ["在网上发表正常观点，却被人恶意攻击、人身辱骂，太没素质了！"], # 愤怒
    ["精心养的花草因为忘记浇水都枯萎了，看着它们蔫掉的样子，心里好难过！"], # 悲伤
    ["工作得到了客户的高度认可，还收到了感谢信，太有动力了！"], # 快乐
    ["商家虚假宣传，说的是正品结果是高仿，联系客服还拒不承认，太过分了！"], # 愤怒
    ["手机抽中了免单大奖，简直不敢相信自己的运气，太惊喜了！"], # 快乐
    ["邻居大半夜还在装修，噪音巨大，敲门提醒还不配合，根本没法休息！"], # 愤怒
    ["曾经那么要好的朋友，慢慢变成了陌生人，没有争吵，只是渐行渐远，好遗憾！"], # 悲伤
    ["发现一家隐藏的宝藏咖啡店，咖啡香浓、环境安静，太喜欢了！"], # 快乐
    ["信用卡被盗刷，联系银行处理还被要求提供各种繁琐证明，处理效率极低，太气人了！"], # 愤怒
    ["和朋友打游戏赢了好几局，配合超默契，太过瘾了！"], # 快乐
    ["考试失利，辜负了父母的期望，看到他们失望的眼神，我心里充满了愧疚和难过！"], # 悲伤
    ["买到的食品里面有异物，联系商家还拒不负责，说我是故意找茬，太气人了！"], # 愤怒
    ["加班到深夜终于完成项目，提交成功的那一刻太有成就感了！"], # 快乐
    ["租房时遇到黑中介，不退押金还编造各种理由扣款，太坑人了！"], # 愤怒
    ["去公园散步时遇到了可爱的小狗，还让我摸了摸它的头，太治愈了！"], # 快乐
    ["养了好几年的宠物突然离世，看着它的照片，眼泪止不住地掉，心都碎了！"], # 悲伤
    ["有人占用消防通道停车，提醒他还不以为然，说“停一会儿怎么了”，太没安全意识了！"], # 愤怒
    ["周末去郊外露营，呼吸着新鲜空气，看着星空，简直太惬意了！"], # 快乐
    ["同事给我带了家乡的特产，味道独特又美味，太贴心了！"], # 快乐
    ["朋友背叛了我，把我的秘密告诉别人，还倒打一耙说我错了，真气人！"], # 愤怒
    ["自己动手组装的书架成功了，看着整齐的书籍，太有成就感了！"], # 快乐
    ["工作压力太大，每天都很焦虑，感觉自己快要撑不下去了，好无助！"], # 悲伤
    ["快递被人冒领了，查监控找到当事人还不归还，态度嚣张，太气人了！"], # 愤怒
    ["去银行办事，工作人员效率极低，排队两小时就办了五分钟的事，太浪费时间了！"], # 愤怒
    ["终于把拖延了好久的报告写完了，卸下重担的感觉真轻松！"], # 快乐
    ["因为身体原因不能做自己喜欢的工作，只能放弃梦想，心里充满了遗憾！"], # 悲伤
    ["外卖小哥提前送达了我着急用的东西，还贴心提醒签收，太靠谱了！"], # 快乐
    ["餐厅服务员态度极差，点单时不耐烦，催单还被怼，花钱买气受！"], # 愤怒
    ["童年时最喜欢的玩具丢了，那是爸爸送我的生日礼物，再也找不回来了！"], # 悲伤
    ["心仪的球鞋终于降价了，果断下单，坐等收货的心情太美妙了！"], # 快乐
    ["上班路上被闯红灯的电动车撞到，对方不仅不道歉还恶语相向，真气人！"], # 愤怒
    ["手机被偷了，里面有很多重要资料，查监控看到小偷还抓不到，太郁闷了！"], # 愤怒
    ["通过努力涨工资了，以后可以给家人更好的生活，太开心了！"], # 快乐
    ["亲人得了绝症，医生说没有治愈的希望了，只能眼睁睁看着他离开，太痛苦了！"], # 悲伤
    ["买到了限量版的周边产品，收藏价值拉满，太满足了！"], # 快乐
    ["商家恶意抬价，平时几十块的东西过节卖几百，还强制捆绑消费，太过分了！"], # 愤怒
    ["地铁上遇到好心人让座，还帮我提了沉重的行李箱，世界真温暖～"], # 快乐
    ["写了很久的文章不小心被删除了，没有备份，所有的心血都白费了，好难过！"], # 悲伤
    ["有人在电梯里抽烟，提醒后还说“关你什么事”，二手烟危害健康，太过分了！"], # 愤怒
    ["网购的商品和图片严重不符，质量差到离谱，申请退款还被要求承担运费，真气人！"], # 愤怒
    ["第一次尝试做蛋糕居然成功了，口感蓬松又香甜，太惊喜了！"], # 快乐
    ["因为误会被朋友误解，解释了他也不相信，心里太委屈了，忍不住哭了！"], # 悲伤
    ["邻居主动分享了自己做的手工饼干，味道超赞，邻里关系真好！"], # 快乐
    ["同事总是在背后说我坏话，造谣生事，影响我的工作和声誉，太恶心了！"], # 愤怒
    ["宠物走丢了，找了好几天都没找到，每天都在担心它有没有饭吃、会不会受欺负！"], # 悲伤
    ["被领导当众表扬工作做得好，还发了奖金，一整天都干劲十足！"], # 快乐
    ["去医院看病，医生态度敷衍，问几句就不耐烦，开的药还没用，太不负责任了！"], # 愤怒
    ["下雨天在家追剧、吃火锅，不用出门的日子也太幸福了吧！"], # 快乐
    ["喜欢的球队输了总决赛，努力了一整个赛季却没能夺冠，太让人失望了！"], # 悲伤
    ["有人故意损坏公共设施，健身器材被砸坏，长椅被涂鸦，太没公德心了！"], # 愤怒
    ["网购的衣服上身效果超棒，版型和面料都超出预期，太满意了！"], # 快乐
    ["孩子在学校被同学欺负，对方家长还蛮不讲理，不道歉还护短，真气人！"], # 愤怒
    ["和好久不见的朋友聚餐，聊起往事笑到肚子疼，太开心了！"], # 快乐
    ["看着父母日渐苍老的背影，感觉自己还没来得及好好孝顺他们，心里充满了愧疚！"], # 悲伤
    ["跑步打卡成功，突破了自己的最佳成绩，果然坚持就有回报！"], # 快乐
    ["工作中被领导无故指责，明明不是我的错却让我背锅，太委屈太气愤了！"], # 愤怒
    ["收到了暗恋对象的表白，原来他也喜欢我，双向奔赴的感觉真好！"], # 快乐
    ["看到流浪的老人和孩子，心里酸酸的，觉得他们太可怜了，却不知道能帮什么！"], # 悲伤
    ["网约车司机故意绕路，多收了几十块钱，揭穿后还不承认，态度恶劣！"], # 愤怒
    ["家人做了我最爱吃的家常菜，满满的都是家的味道，太幸福了！"], # 快乐
    ["公共厕所被人弄得乱七八糟，垃圾遍地，根本没法使用，太没素质了！"], # 愤怒
    ["熬夜看完了一本超好看的小说，结局圆满，太上头了！"], # 快乐
    ["喜欢的歌手去世了，再也听不到他的新歌，再也看不到他的演唱会了，好遗憾！"], # 悲伤
    ["终于学会了骑自行车，虽然摔了几次，但成功的那一刻太开心了！"], # 快乐
    ["奶奶得了重病，看着她痛苦的样子，我却什么都做不了，心里太难受了！"], # 悲伤
    ["和最好的闺蜜闹掰了，因为一场误会，现在见面都不说话，心里太难受了！"], # 悲伤
    ["家里的老屋要拆迁了，那里装满了我的童年回忆，再也回不去了，好难过！"], # 悲伤
    ["努力了很久的考试还是没通过，感觉自己很没用，付出的都白费了！"], # 悲伤
    ["最好的朋友要移民国外了，以后见面的机会很少了，送他的时候忍不住落泪！"], # 悲伤
    ["从小一起长大的发小意外去世，听到消息的那一刻，整个人都懵了，不敢相信！"], # 悲伤
    ["因为疫情，好几年没回家过年了，看着别人阖家团圆，心里充满了思念和难过！"], # 悲伤
    ["工作失误导致公司遭受损失，被领导批评，同事也有怨言，感觉自己很失败！"], # 悲伤
    ["暗恋的人要结婚了，新娘不是我，只能默默祝福，心里却像针扎一样疼！"], # 悲伤
    ["爷爷的老房子被一场大火烧毁了，里面的老物件和回忆都没了，好难过！"], # 悲伤
    ["和家人吵架了，说了很多伤人的话，现在很后悔，但又拉不下脸道歉！"]  # 悲伤  
],
    title="情感分析演示"
)

if __name__ == "__main__":
    demo.launch()