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

from applications.functions.chatgpt import OpenAIServer
from applications.config import moon_shot
from applications.config import deep_seek
from applications.log import logging


class KimiServer(object):
    """
    Kimi Server
    """

    @classmethod
    async def search_kimi_schedule(cls, params, safe_score):
        """
        搜索阶段 kimi 操作
        :param params:
        :param safe_score:
        :return:
        """
        title = params['article_title'].split("@@")[-1]
        contents = params['article_text']
        trace_id = params['content_id']
        try:
            kimi_title = await cls.kimi_title(title, ai_model=moon_shot)
            # 判断kimi 标题是否安全
            title_score = await cls.get_kimi_title_safe_score(kimi_title, ai_model=moon_shot)
            kimi_green_title = None
            if int(title_score) > safe_score:
                kimi_green_title = await cls.make_kimi_title_safer(title, ai_model=moon_shot)
                
            logging(
                code="1005",
                info="generate_title",
                data={
                    "ori_title": title,
                    "kimi_title": kimi_title,
                    "title_score": title_score,
                    "new_kimi_title": kimi_green_title
                },
                trace_id=trace_id
            )
            kimi_title = kimi_green_title if kimi_green_title else kimi_title

        except Exception as e:
            logging(
                code="4002",
                info="kimi 挖掘失败--{}, 采用 gpt".format(e),
                trace_id=trace_id
            )
            gpt = OpenAIServer()
            kimi_title = gpt.gpt_mining(
                text=title,
                prompt=gpt.prompt_generate(
                    prompt_type="title"
                )
            )

        kimi_title = kimi_title.replace("'", "").replace('"', "").replace("\\", "")
        try:
            kimi_info = await cls.kimi_mining(contents, ai_model=moon_shot)
        except Exception as e:
            logging(
                code="4002",
                info="kimi 挖掘失败--{}, 采用 gpt".format(e),
                trace_id=trace_id
            )
            gpt = OpenAIServer()
            kimi_info = gpt.gpt_mining(
                text=contents,
                prompt=gpt.prompt_generate(
                    prompt_type="text"
                )
            )
            kimi_info = kimi_info.replace('```json', '').replace('```', '')
            kimi_info = json.loads(kimi_info)
        kimi_info['k_title'] = kimi_title
        kimi_info['ori_title'] = title
        logging(
            code="1005",
            info="kimi_mining",
            data=kimi_info,
            trace_id=trace_id
        )
        return kimi_info

    @classmethod
    async def ask_kimi(cls, question):
        """
        Ask Kimi for information
        :param question: tiny text
        :return: "{}"
        """
        single_title_prompt = """
            我会给你一个视频标题，需要你帮我用你所学的知识来帮我分析出以下信息，信息我都写到 json 里面了
            {
                "key_words": [],  # 返回三个关键词
                "search_keys": [], # 标题可能的搜索关键词，返回 3 个
                "extra_keys": [], # 关心这个视频的用户还会关心哪些关键词， 返回 3 个
                "theme": 标题的主题， 用一个词概括
            }
            只需要返回一个 json，key 和上面的一样，
            我给你的标题是: 
            """
        client = OpenAI(
            api_key='sk-5DqYCa88kche6nwIWjLE1p4oMm8nXrR9kQMKbBolNAWERu7q',
            base_url="https://api.moonshot.cn/v1"
        )
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": single_title_prompt + question,
                }
            ],
            model="moonshot-v1-8k",
        )
        response = chat_completion.choices[0].message.content.replace('```json', '').replace('```', '')
        try:
            response = json.loads(response)
            return response
        except:
            return {}

    @classmethod
    async def ask_kimi_and_save_to_local(cls, info_tuple):
        """
        save file to local
        :return:
        """
        title, trace_id, save_path = info_tuple[0], info_tuple[1], info_tuple[2]
        if os.path.exists(save_path):
            logging(
                code="2001",
                info="该 video 信息已经挖掘完成---{}".format(title),
                function="ask_kimi_and_save_to_local",
                trace_id=trace_id,
            )
        else:
            os.makedirs(os.path.dirname(save_path), exist_ok=True)
            if not title:
                result = {}
            else:
                result = await cls.ask_kimi(title)
            logging(
                code="2001",
                info="kimi-result",
                data=result,
                trace_id=trace_id,
                function="ask_kimi_and_save_to_local"
            )
            with open(save_path, "w", encoding="utf-8") as f:
                f.write(json.dumps(result, ensure_ascii=False))

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

    @classmethod
    async def kimi_mining(cls, text, ai_model):
        """
        通过文章来挖掘出有效的信息
        :param text:
        :param ai_model:
        :return:
        """
        text_prompt = """
        请从我给你的文章中挖掘出以下信息并且返回如下结果。
        你返回的结果是一个 json, 格式如下:
        {
            "content_keys": [] # 同时提供三个与文章内容高度相关的关键词，这些关键词将用于网络上搜索相关视频内容,
            "content_title": 一个总结性的标题，该标题应简洁并能够反映文章的主要内容
        }
        你需要处理的文本是：
        """
        client = OpenAI(
            api_key=ai_model['api_key'],
            base_url=ai_model['base_url']
        )
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": text_prompt + text,
                }
            ],
            model=ai_model['model'],
            response_format={"type": "json_object"}
        )
        content = json.loads(chat_completion.choices[0].message.content)
        return content

    @classmethod
    async def get_kimi_title_safe_score(cls, kimi_title, ai_model):
        """
        获取kimi_title的安全分数
        """
        prompt = f"""
        请你学习一下内容规范，以下标题可能会违反了某条内容规范。请你对标题做一个内容风险评级，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）含有法律、行政法规禁止的其他内容的。
        输入的标题是： {kimi_title}
        """
        client = OpenAI(
            api_key=ai_model['api_key'],
            base_url=ai_model['base_url']
        )
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": prompt
                }
            ],
            model=ai_model['model']
        )
        response = chat_completion.choices[0].message.content
        return response

    @classmethod
    async def make_kimi_title_safer(cls, ori_title, ai_model):
        """
        安全化 kimi title
        """
        prompt = """
            以下每行为一个文章的标题，请用尽量平实的语言对以上标题进行改写，保持在10～15字左右，请注意：
            1. 不要虚构或改变标题的含义。
            2. 不要用笃定的语气描述存疑的可能性，不要将表述可能性的问句改为肯定句。
            直接输出改写后的标题列表。
            在改写完成后，再输出一次，在改写的标题前增加和标题情感、语气匹配的特殊符号，如：“🔴”、“😄”、“🔥”、“😨”等等
            输出：
                输出结果是Dict, 格式为: 
                {
                "title_v1": 请填写第一次输出的标题,
                "title_v2": 请填写第二次输出的标题
                }
            输入的标题是: 
        """
        client = OpenAI(
            api_key=ai_model['api_key'],
            base_url=ai_model['base_url']
        )
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": prompt + ori_title,
                }
            ],
            model=ai_model['model'],
            response_format={"type": "json_object"}
        )
        response = chat_completion.choices[0].message.content
        response_json = json.loads(response)
        safer_title = response_json["title_v2"]
        return safer_title