zhangyong 8 月之前
父节点
当前提交
4d782edad2
共有 1 个文件被更改,包括 11 次插入7 次删除
  1. 11 7
      common/gpt4o_help.py

+ 11 - 7
common/gpt4o_help.py

@@ -115,11 +115,15 @@ class GPT4o():
                     if content:
                         if content == 'None':
                             return "这个视频,分享给我的老友,祝愿您能幸福安康"
-                        # data = json.loads(content)
-                        # title = data["title"]
-                        # if title:
-                        #     return title
-                        # else:
+                        if content in 'title':
+                            data = json.loads(content)
+                            title = data["title"]
+                            if title:
+                                return title
+                        elif content in '新标题':
+                            data = json.loads(content)
+                            title = data['新标题']
+                            return title
                         return content
                     elif attempt == max_retries - 1:
                         return "这个视频,分享给我的老友,祝愿您能幸福安康"
@@ -385,5 +389,5 @@ class GPT4o():
 
 
 if __name__ == '__main__':
-    title = '@晟唐影视(O3xu74hiuqz32ggi) 的精彩视频'
-    GPT4o.get_title(title)
+    title = '感谢您的分享,感谢支持。'
+    GPT4o.get_ai_title(title)