zhangyong 4 ヶ月 前
コミット
7204920e6f
3 ファイル変更38 行追加20 行削除
  1. 5 3
      carry_video/carry_video.py
  2. 14 0
      common/feishu_form.py
  3. 19 17
      common/gpt4o_mini_help.py

+ 5 - 3
carry_video/carry_video.py

@@ -9,7 +9,7 @@ import requests
 from datetime import datetime
 from urllib.parse import urlparse, parse_qs
 from loguru import logger
-from common import Oss, Feishu, AliyunLogger
+from common import Oss, Feishu, AliyunLogger, Material
 from common.download_video import DownLoad
 from common.ffmpeg import FFmpeg
 from common.gpt4o_mini_help import GPT4oMini
@@ -381,8 +381,10 @@ class CarryViode:
                     video_path = FFmpeg.video_ggduration(video_path, file_path, data["video_clipping_time"])
 
                 if data['trailer_share']:
-                    if data['trailer_share'] ==  "AI片尾引导":
-                        pw_srt_text = GPT4oMini.get_ai_mini_pw(title)
+                    if data['trailer_share'] ==  "AI片尾引导" or data['trailer_share'] ==  "AI片尾引导1" :
+                        prompt = Material.get_propmt_data(data['trailer_share'])
+
+                        pw_srt_text = GPT4oMini.get_ai_mini_pw(title, prompt)
                     else:
                         pw_srt_text = str(data['trailer_share'])
                     voice = data['trailer_share_audio']

+ 14 - 0
common/feishu_form.py

@@ -63,3 +63,17 @@ class Material():
         except:
             return processed_list
 
+    @classmethod
+    def get_propmt_data(cls, trailer_share):
+        try:
+            data = Feishu.get_values_batch( "Wj0TsRKc0hZrHQtmtg4cZZIwn0c", "Yxg7EK" )
+            for row in data[1:]:
+                name = row[0]
+                if trailer_share == name:
+                    print(row[1])
+                    return row[1]
+        except:
+            return None
+
+
+

+ 19 - 17
common/gpt4o_mini_help.py

@@ -3,31 +3,33 @@ import time
 
 import requests
 
+from common import Material
+
 
 class GPT4oMini:
 
     @classmethod
-    def get_ai_mini_pw(cls, title):
+    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": (
-                "请针对微信平台视频类小程序场景,面向人群是中国中老年人,在单聊、群聊场景,"
-                "根据以下示例生成结尾引导分享的脚本,引导用户发生更多的分享行为。\n"
-                "示例case1:\n"
-                "标题: 几十年前的老歌,现在很少听到了!\n"
-                "返回分享引导脚本: 这个视频真是难得,太美太好听了,看完真是回忆满满,"
-                "让人怀念那个时代,多么淳朴多么美好,今天大家都在群里转发这个视频,"
-                "看了都说,那时候虽然手里不宽裕,但心里都满怀希望,那股子拼劲,真是让人热血沸腾,"
-                "老朋友你还记得吗?如果你也跟我一样深深的怀念,赶紧把这个视频转发到你所有群里去吧,"
-                "让大家伙都看看吧,多多转发,感恩有你!\n\n"
-                "示例case2:\n"
-                "标题: 🔺令人尊敬的老人❗❗这才是我们应该追的明星❗\n"
-                "返回分享引导脚本: 这个视频发出来5天了,转发的人却寥寥无几,"
-                "试想一次明星出轨就全民皆知,这样中国真正的正能量却无人关注,真是太讽刺了!"
-                "如果你看到了这个视频,请分享到你加入的每个群,向杨老这种无私奉献的大爱致敬!"
-                "千万别让那些低调做好事的人心寒!转发传递下去,让世界都看到,中华民族的正能量!\n\n"
+                prompt+
                 f"请分析该标题,标题为:{title},返回结尾分享引导脚本。"
             ),
             "responseFormat": {
@@ -156,4 +158,4 @@ class GPT4oMini:
         return "这个视频,分享给我的老友,祝愿您能幸福安康"
 
 if __name__ == '__main__':
-    GPT4oMini.get_ai_mini_title("这段话说出了多少人的心声 #老百姓的心声 #老百姓关心的话题 #农民的心声 #老百姓不容易")
+    GPT4oMini.get_ai_mini_pw("这段话说出了多少人的心声 #老百姓的心声 #老百姓关心的话题 #农民的心声 #老百姓不容易","AI片尾引导1")