carry_video_rewriting/common/gpt4o_mini_help.py

import json
import time

import requests

from common import Material


class GPT4oMini:

    @classmethod
    def get_ai_mini_pw(cls, title,  prompt):
        if not prompt:
            prompt = """"请针对微信平台视频类小程序场景，面向人群是中国中老年人，在单聊、群聊场景，"
                    "根据以下示例生成结尾引导分享的脚本，引导老年用户分享给好友和群，必须有6个带引导分享的句子，以老年人祝福的语气结尾。\n"
                    "示例case1：\n"
                    "标题: 几十年前的老歌，现在很少听到了！\n"
                    "返回分享引导脚本: 这个视频真是难得，太美太好听了，看完真是回忆满满，" "让人怀念那个时代，多么淳朴多么美好，今天大家都在群里转发这个视频，"  "老朋友你还记得吗？如果你也跟我一样深深的怀念，赶紧把这个视频转发到你所有群里去吧，" "有多少群就转发多少群，不要私藏" "‌每一次分享都是一次奉献，一份真诚的情感‌‌" "转发给每一位都怀念当年岁月的老友，他们都会感谢你的" "多多转发，情谊多多！,"
                    "祝每位老友健健康康，平平安安\n\n" "示例case2：\n"
                    "标题: 🔺令人尊敬的老人❗❗这才是我们应该追的明星❗\n"
                    "返回分享引导脚本: 这个视频发出来5天了，转发的人却寥寥无几，"
                    "试想一次明星出轨就全民皆知，这样中国真正的正能量却无人关注，真是太讽刺了！"
                    "如果你看到了这个视频，请分享到你加入的每个群，向杨老这种无私奉献的大爱致敬！" "发给5个群和5个好友，"
                    "千万别让那些低调做好事的人心寒！转发传递下去，让世界都看到，中华民族的正能量！"
                    "我已经转发了，也希望你能帮助传递"
                     "赠人玫瑰，手留余香，分享好视频，收获好福气\n\n" """
        url = "http://aigc-api.cybertogether.net//aigc/dev/test/gpt"
        payload = json.dumps({
            "imageList": [],
            "model": "gpt-4o-mini-2024-07-18",
            "prompt": (
                prompt+
                f"请分析该标题，标题为：{title}，返回结尾分享引导脚本。"
            ),
            "responseFormat": {
                "type": "json_schema",
                "json_schema": {
                    "strict": True,
                    "name": "share_script_result",
                    "schema": {
                        "type": "object",
                        "properties": {
                            "分享引导脚本": {
                                "type": "string",
                                "description": "生成的结尾分享引导脚本"
                            }
                        },
                        "required": ["分享引导脚本"],
                        "additionalProperties": False
                    }
                }
            }
        })
        headers = {'Content-Type': 'application/json'}
        try:
            response = requests.post(url, headers=headers, data=payload, timeout=30)
            response_data = response.json()

            data = json.loads(response_data.get('data', '{}'))
            pw  = data["分享引导脚本"]
            return pw
        except Exception as e:
            pw_srt_text = '老友们，这个视频说的太好了,请你在看完之后,不要私藏,分享给身边的亲人朋友,转发传递这份美好,这不仅是分享一个好视频,更是在传递一份深厚的祝福和好运,愿每个看到这消息的人,都能收获健康长寿和平安,希望您在看完后能够回传给我,并且把这条视频再传给3个群和3个好友,祝愿你我,健健康康,平平安安,因为3个数字代表着健康和平安,这么好的视频,千万不要在你的手里终止了,分享给群友们,他们看到以后一定会感谢你的,感谢分享,感谢传递'
            return pw_srt_text

    @classmethod
    def get_ai_mini_title(cls, title):
        url = "http://aigc-api.cybertogether.net//aigc/dev/test/gpt"
        payload = json.dumps({
            "imageList": [],
            "model": "gpt-4o-mini-2024-07-18",
            "prompt": (
                        "CONTEXT"
                        "你是一名短视频标题优化专家，任务是为短视频生成吸引力高且符合规范的标题。"
                        "OBJECTIVE"
                        "基于优秀标题要求及示例，生成新的、吸引用户注意的标题。"
                        "TIME"
                        "在标题生成请求提出后的1分钟内生成。"
                        "ACTOR"
                        "短视频标题优化专家"
                        "RESOURCES"
                        "- 原标题列表"
                        "- 优秀标题示例"
                        "- 标题生成规范和约束条件"
                        "RESPONSE"
                        "为每个原标题生成符合规范的新标题。"
                        "生成示例标题案例及要求"
                        "1. 标题开头包含醒目emoji🔴，整体字符长度必须控制在10-25个字"
                        "2. 识别需要修改的标题中可能感兴趣的人群，并在标题中表现出来。人群示例如：群友们、退休人员、50~70后等"
                        "3. 标题中可增加一些对内容的观点/态度，用人格化的方式表达，示例：太香了、老外至今难以相信"
                        "4. 标题结尾可以根据标题内容增加一些引导语，格式参考示例但不必局限于示例。示例：你们见过吗、你听对不对、说的太好了、请听、太神奇了"
                        "5. 对于包含#话题和@人的标题，则去除@人元素后，利用剩下的全部信息生成标题。"
                        "6. 不能编造、新增原标题没有的信息，如原标题没有养老金，不能在生成的标题中出现养老金。"
                        "7. 标题内不能使用强引导分享点击的词句，如：快来看看、大家都听一听、值得一看、都看看吧、你也来看看吧、大家注意、都听听等。"
                        "8. 不能使用无实质信息和强烈诱导点击、紧急、夸张、震惊的描述，避免使用“震惊国人”、“速看”、“太震撼了”等类似描述。"
                        "9. 标题需要简洁、清晰，不要使用网络流行语，如：太燃了、佛系、躺平、内卷等。"
                    
                        "Goodcase示例："
                        "⭕老外至今难以相信，中国人竟能把大桥建到天上，穿入云中"
                        "🔴未来酒店体验，群友们，请看！"
                        "⭕六七十年代的《忠字舞》，你们见过吗？"
                        "🔴哈哈哈！大哥说的太好了！太真实了"
                        "🔴今天，请记住那1700个集体赴死的年轻人，平均23岁！"
                        "🔴这才叫老同学聚会，到了这个年纪，还能聚在一起真不容易！"
                        "🔴百善孝为先，心凉了捂不热了"
                        "🔴养儿不如养狗，一件真实的事！"
                        "🔴让人受益的一段话，写得真好!"
                        "🔴“处暑三劝”！发给最好的朋友劝一劝！"
                        "🔴世间公道自在人心，善恶有报，人生智慧建议收藏！"
                        "🔴坐着电梯登上山顶！这怕是只有中国人敢想敢做！"
                        "🔴人老了！就应该这样去活！"
                    
                        "badcase示例："
                        "🌸绝对不能错过，快打开看看，越快越好"
                        "所有老年人一定要看"
                        "天大的好消息，5月开始实施❗"
                        "就在刚刚，中国突然传出重磅消息，所有人都不敢相信！ 🚩"
                        "丧尽天良！为什么生病的人越来越多，原来吃的是这些 🎈"
                        "今年的端午节太特殊，一辈子难遇一次!一定要看!错过别后悔"
                        "好消息来了，千万别划走！"
                        "紧急！已爆发，错过就晚了😱"
                    f"请分析该标题，标题为：{title}，返回新的标题。"
                    ),
            "responseFormat": {
                "type": "json_schema",
                "json_schema": {
                    "strict": True,
                    "name": "share_script_result",
                    "schema": {
                        "type": "object",
                        "properties": {
                            "新标题": {
                                "type": "string",
                                "description": "生成新的标题"
                            }
                        },
                        "required": ["新标题"],
                        "additionalProperties": False
                    }
                }
            }
        })
        headers = {'Content-Type': 'application/json'}
        max_retries = 3
        retry_count = 0
        while retry_count < max_retries:
            try:
                response = requests.post(url, headers=headers, data=payload, timeout=30)
                response_data = response.json()

                data = json.loads(response_data.get('data', '{}'))
                new_title = data["新标题"]
                return new_title
            except Exception as e:
                retry_count += 1
                # logger.error(f"尝试第 {retry_count} 次失败，错误: {e}")
                time.sleep(1)  # 延迟1秒后重试
        return "这个视频，分享给我的老友，祝愿您能幸福安康"

if __name__ == '__main__':
    GPT4oMini.get_ai_mini_pw("这段话说出了多少人的心声 #老百姓的心声 #老百姓关心的话题 #农民的心声 #老百姓不容易","AI片尾引导1")