ソースを参照

修改prompt,调试

zhangyong 7 ヶ月 前
コミット
c3709f5c88
2 ファイル変更25 行追加4 行削除
  1. 2 1
      common/cover_gpt4o.py
  2. 23 3
      video_cover_method/cover_method.py

+ 2 - 1
common/cover_gpt4o.py

@@ -17,7 +17,8 @@ class CoverAnalyzer:
             url = "http://aigc-api.cybertogether.net//aigc/dev/test/gpt"
             payload = json.dumps({
                 "imageList": [cover],
-                "prompt": "请帮我完成图片分析工作",
+                # "prompt": "请帮我完成图片分析工作",
+                "prompt": "请帮我完成图片分析工作:你只需要理解图片正中心600 × 480的区域画面,根据该区域画面和特征,给出相应的回答",
                 "responseFormat": {
                     "type": "json_schema",
                     "json_schema": {

+ 23 - 3
video_cover_method/cover_method.py

@@ -50,7 +50,7 @@ class CoverMethod:
             execution_time = end_time - start_time
             print(f"代码运行时间: {execution_time:.4f} 秒")
             highest_cover_url, highest_cover_sum = CoverMethod.cover_data_dispose(cover_all_data)  # 获取分数最高的视频封面
-            if highest_cover_url == None:
+            if not highest_cover_url:
                 AliyunLogger.logging( video_id, "视频封面分数都为0,不做封面更改", "2002", cover_all_data)
                 text = (
                     f"**通知类型**: 视频封面分数都为0,不做封面更改\n"
@@ -61,11 +61,15 @@ class CoverMethod:
                                    "【 封面修改通知 】" )
                 return
             cover_id, old_cover_url = PQ.get_pq_cover_id(video_id)  # 获取原封面 和 封面ID
-
             if cover_id == None:
                 AliyunLogger.logging( video_id, "获取视频原封面和封面ID失败", "3002", cover_all_data, old_cover_url, highest_cover_url)
                 insert_pq_data( [video_id] )
                 return
+            old_cover_data = CoverAnalyzer.cover_list_data([old_cover_url])
+            if not old_cover_data:
+                old_cover_sum = 0
+            else:
+                old_cover_url, old_cover_sum = CoverMethod.cover_data_dispose( old_cover_data )
             if highest_cover_url == old_cover_url:
                 AliyunLogger.logging( video_id, "原封面与新封面一致,不做封面修改", "2003", cover_all_data, old_cover_url,
                                       highest_cover_url )
@@ -73,6 +77,20 @@ class CoverMethod:
                     f"**通知类型**: 原封面与新封面一致,不做封面修改\n"
                     f"**站内视频ID链接**: https://admin.piaoquantv.com/cms/post-detail/{video_id}/detail\n"
                     f"**原视频封面链接**: {old_cover_url}\n"
+                    f"**原视频封面分数**: {old_cover_sum}\n"
+                    f"**新视频封面链接**: {highest_cover_url}\n"
+                    f"**新视频封面分数**: {highest_cover_sum}\n"
+                )
+                Feishu.finish_bot( text,
+                                   "https://open.feishu.cn/open-apis/bot/v2/hook/63745308-c53e-4030-90b9-77383b057252",
+                                   "【 封面修改通知 】" )
+                return
+            if int(highest_cover_sum) < int(old_cover_sum):
+                text = (
+                    f"**通知类型**: 原封面分数比新封面分数高,不做封面修改\n"
+                    f"**站内视频ID链接**: https://admin.piaoquantv.com/cms/post-detail/{video_id}/detail\n"
+                    f"**原视频封面链接**: {old_cover_url}\n"
+                    f"**原视频封面分数**: {old_cover_sum}\n"
                     f"**新视频封面链接**: {highest_cover_url}\n"
                     f"**新视频封面分数**: {highest_cover_sum}\n"
                 )
@@ -87,6 +105,7 @@ class CoverMethod:
                     f"**通知类型**: 视频封面修改成功\n"
                     f"**站内视频ID链接**: https://admin.piaoquantv.com/cms/post-detail/{video_id}/detail\n"
                     f"**原视频封面链接**: {old_cover_url}\n"
+                    f"**原视频封面分数**: {old_cover_sum}\n"
                     f"**新视频封面链接**: {highest_cover_url}\n"
                     f"**新视频封面分数**: {highest_cover_sum}\n"
                 )
@@ -100,6 +119,7 @@ class CoverMethod:
                     f"**通知类型**: 视频封面修改失败\n"
                     f"**站内视频ID链接**: https://admin.piaoquantv.com/cms/post-detail/{video_id}/detail\n"
                     f"**原视频封面链接**: {old_cover_url}\n"
+                    f"**原视频封面分数**: {old_cover_sum}\n"
                     f"**新视频封面链接**: {highest_cover_url}\n"
                     f"**新视频封面分数**: {highest_cover_sum}\n"
                 )
@@ -122,4 +142,4 @@ class CoverMethod:
 
 
 if __name__ == '__main__':
-    CoverMethod.cover_method("")
+    CoverMethod.cover_method("30479284")