zhangyong 8 months ago
parent
commit
4d782edad2
1 changed files with 11 additions and 7 deletions
  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)