zhangyong 4 月之前
父节点
当前提交
2ad29713b1
共有 2 个文件被更改,包括 61 次插入1 次删除
  1. 2 1
      carry_video/carry_video.py
  2. 59 0
      common/gpt4o_mini_help.py

+ 2 - 1
carry_video/carry_video.py

@@ -13,6 +13,7 @@ from common import Oss, Feishu, AliyunLogger
 from common.download_video import DownLoad
 from common.ffmpeg import FFmpeg
 from common.gpt4o_help import GPT4o
+from common.gpt4o_mini_help import GPT4oMini
 from common.redis import in_carry_video_data
 from common.sql_help import sqlCollect
 from common.tag_video import Tag
@@ -351,7 +352,7 @@ class CarryViode:
 
             if data['trailer_share']:
                 if data['trailer_share'] ==  "AI片尾引导":
-                    pw_srt_text = GPT4o.get_ai_pw(title)
+                    pw_srt_text = GPT4oMini.get_ai_mini_pw(title)
                 else:
                     pw_srt_text = str(data['trailer_share'])
                 voice = data['trailer_share_audio']

+ 59 - 0
common/gpt4o_mini_help.py

@@ -0,0 +1,59 @@
+import json
+
+import requests
+class GPT4oMini:
+
+    @classmethod
+    def get_ai_mini_pw(cls, title):
+        url = "http://aigc-api.cybertogether.net//aigc/dev/test/gpt"
+        payload = json.dumps({
+            "imageList": [],
+            "model": "gpt-4o-mini-2024-07-18",
+            "prompt": (
+                "请针对微信平台视频类小程序场景,面向人群是中国中老年人,在单聊、群聊场景,"
+                "根据以下示例生成结尾引导分享的脚本,引导用户发生更多的分享行为。\n"
+                "示例case1:\n"
+                "标题: 几十年前的老歌,现在很少听到了!\n"
+                "返回分享引导脚本: 这个视频真是难得,太美太好听了,看完真是回忆满满,"
+                "让人怀念那个时代,多么淳朴多么美好,今天大家都在群里转发这个视频,"
+                "看了都说,那时候虽然手里不宽裕,但心里都满怀希望,那股子拼劲,真是让人热血沸腾,"
+                "老朋友你还记得吗?如果你也跟我一样深深的怀念,赶紧把这个视频转发到你所有群里去吧,"
+                "让大家伙都看看吧,多多转发,感恩有你!\n\n"
+                "示例case2:\n"
+                "标题: 🔺令人尊敬的老人❗❗这才是我们应该追的明星❗\n"
+                "返回分享引导脚本: 这个视频发出来5天了,转发的人却寥寥无几,"
+                "试想一次明星出轨就全民皆知,这样中国真正的正能量却无人关注,真是太讽刺了!"
+                "如果你看到了这个视频,请分享到你加入的每个群,向杨老这种无私奉献的大爱致敬!"
+                "千万别让那些低调做好事的人心寒!转发传递下去,让世界都看到,中华民族的正能量!\n\n"
+                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)
+            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