"""
@author: luojunhui
"""
import json
from openai import OpenAI

from config import moon_shot

generate_program_title_prompt = """
    请将以上标题改写成适合小程序点击和传播的小程序标题，小程序标题的写作规范如下，请学习后进行小程序标题的编写。直接输出最终的小程序标题
    小程序标题写作规范：
    1.要点前置：将最重要的信息放在标题的最前面，以快速吸引读者的注意力。例如，“5月一辈子同学，三辈子亲，送给我的老同学，听哭无数人！”中的“5月”和“一辈子同学，三辈子亲”都是重要的信息点。
    2.激发情绪：使用能够触动人心的语言，激发读者的情感共鸣。如“只剩两人同学聚会，看后感动落泪。”使用“感动落泪”激发读者的同情和怀旧情绪。
    3.使用数字和特殊符号：数字可以提供具体性，而特殊符号如“🔴”、“😄”、“🔥”等可以吸引视觉注意力，增加点击率。
    4.悬念和好奇心：创建悬念或提出问题，激发读者的好奇心。例如，“太神奇了!长江水位下降，重庆出现惊奇一幕!”中的“惊奇一幕”就是一个悬念。
    5.名人效应：如果内容与知名人士相关，提及他们的名字可以增加标题的吸引力。
    6.社会价值观：触及读者的文化和社会价值观，如家庭、友情、国家荣誉等。
    7.标点符号的运用：使用感叹号、问号等标点来增强语气和情感表达。
    8.直接的语言：使用直白、口语化的语言，易于理解，如“狗屁股，笑死我了!”。
    9.热点人物或事件：提及当前的热点人物或事件，利用热点效应吸引读者。
    10.字数适中：保持标题在10-20个字之间，既不过长也不过短，确保信息的完整性和吸引力。
    11.适当的紧迫感：使用“最新”、“首次”、“紧急”等词汇，创造一种紧迫感，促使读者立即行动。
    12.情感或价值诉求：使用如“感动”、“泪目”、“经典”等词汇，直接与读者的情感或价值观产生共鸣。
    避免误导：确保标题准确反映内容，避免夸大或误导读者。
    """

get_title_safe_score_prompt = """
    请你学习一下内容规范，以下标题可能会违反了某条内容规范。请你对标题做一个内容风险评级，1-10分，等级越高内容违规风险越大。 
    请直接输出内容风险评级的分数，不要输出你的理由、分析等内容。 
    输出:
        只需要输出危险分级分数，不要输出任何其他内容。
    内容规范为： 
    4.2 色俗内容 
        4.2.1 散布淫秽、色情内容，包括但不限于招嫖、寻找一夜情、性伴侣等。 
        4.2.2 发布有色情意味的情色文字、情色视频、情色漫画等内容。 
        4.2.3 以投稿/爆料等形式描述约炮经历、性交体验、偷情、涉隐私部位偷拍等伤风败俗的话题内容。 
        4.2.4 以低俗的配图引诱用户阅读文章、关注微信公众号。包含性撩拨、性挑逗画面；疑似女性性高潮/性虐场面；偷拍的沐浴/更衣/如厕/亲热等私密画面；女性故意露出敏感部位 (纯裸露的胸、生殖器官)以及敏感部位未打码的真人写真/艺术摄影等。 
        4.2.5 文内以低俗类的动图或引导图，诱导用户点击进而跳转至另一篇图文页或关注某个公众号。 
        4.2.6 文章主要描述PUA撩妹、撩汉等相关话题，且引导用户关注公众号/加个人微信号/加群。 
    4.11 煽动、夸大、误导类内容 平台鼓励创作者提供准确、清晰、能体现文章内容主旨的标题，不允许通过标题噱头诱导用户点击或误导用户。
        包括但不限于以下情况： 
        4.11.1 标题含有危害人身安全、恐吓侮辱、惊悚、极端内容，或者以命令式语气强迫用户阅读。 
        4.11.2 标题无依据夸大事件严重程度、紧急程度、受影响面以及事件引发的情绪。 
        4.11.3 标题以浮夸的描述，反常识强调某种食物/行为对人体健康的影响，煽动人群要/不要去做某行为。 
        4.11.4 非官方通知或者公告，但标题假借官方名义煽动获取流量，或以信息来源机密、看完即删来诱导用户。 
        4.11.5 标题故意隐藏关键信息，或无中生有部分信息，给用户造成误导。 
        4.12 违反国家法律法规禁止的内容 
            （1）违反宪法确定的基本原则的； 
            （2）危害国家安全，泄露国家秘密，颠覆国家政权，破坏国家统一的； 
            （3）损害国家荣誉和利益的； 
            （4）煽动民族仇恨、民族歧视，破坏民族团结的； 
            （5）破坏国家宗教政策，宣扬邪教和封建迷信的； 
            （6）散布不实信息，扰乱社会秩序，破坏社会稳定的； 
            （7）散布淫秽、色情、赌博、暴力、恐怖或者教唆犯罪的； 
            （8）侮辱或者诽谤他人，侵害他人合法权益的； 
            （9）煽动非法集会、结社、游行、示威、聚众扰乱社会秩序； 
            （10）以非法民间组织名义活动的； 
            （11）不符合《即时通信工具公众信息服务发展管理暂行规定》及遵守法律法规、社会主义制度、国家利益、公民合法利益、公共秩序、社会道德风尚和信息真实性等“七条底线”要求的； 
            （12）含有法律、行政法规禁止的其他内容的。
    输入的标题是： 
    """

make_title_safe_prompt = """
    以下每行为一个文章的标题，请用尽量平实的语言对以上标题进行改写，保持在10～15字左右，请注意：
    1. 不要虚构或改变标题的含义。
    2. 不要用笃定的语气描述存疑的可能性，不要将表述可能性的问句改为肯定句。
    直接输出改写后的标题列表。
    在改写完成后，再输出一次，在改写的标题前增加和标题情感、语气匹配的特殊符号，如：“🔴”、“😄”、“🔥”、“😨”等等
    输出：
        输出结果是Dict, 格式为: 
        {
        "title_v1": 请填写第一次输出的标题,
        "title_v2": 请填写第二次输出的标题
        }
    输入的标题是: 
        """


def fetch_moon_shot_response(task, input_text, output_type="text"):
    """
    调用kimi的api获取结果
    """
    # generate prompt
    match task:
        case "generate_kimi_title":
            prompt = input_text + '\n' + generate_program_title_prompt
        case "get_title_safe_score":
            prompt = get_title_safe_score_prompt + input_text
        case "make_title_safe":
            prompt = make_title_safe_prompt + input_text
        case _:
            prompt = input_text

    # init client
    client = OpenAI(
        api_key=moon_shot['api_key'],
        base_url=moon_shot['base_url']
    )

    # get response
    if output_type == "json":
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": prompt,
                }
            ],
            model=moon_shot['model'],
            response_format={"type": "json_object"}
        )
        response = chat_completion.choices[0].message.content
        response_json = json.loads(response)
        return response_json

    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": prompt,
            }
        ],
        model=moon_shot['model']
    )
    response = chat_completion.choices[0].message.content
    return response