zhangyong 4 months ago
parent
commit
cff84ac3d3
4 changed files with 3 additions and 6 deletions
  1. 0 1
      carry_data_handle.py
  2. 1 2
      carry_video/carry_video.py
  3. 1 2
      common/download_video.py
  4. 1 1
      common/gpt4o_mini_help.py

+ 0 - 1
carry_data_handle.py

@@ -8,7 +8,6 @@ from loguru import logger
 
 from carry_video.carry_video import CarryViode
 from common import Feishu, AliyunLogger
-from common.ffmpeg import FFmpeg
 from common.redis import get_carry_data, in_carry_video_data
 
 ENV = os.getenv('ENV', 'dev')

+ 1 - 2
carry_video/carry_video.py

@@ -12,7 +12,6 @@ from loguru import logger
 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
@@ -309,7 +308,7 @@ class CarryViode:
         logger.info(f"[+] {url}开始视频下载成功")
         logger.info(f"[+] {url}开始处理标题")
         if data["title_category"] == "AI标题" or data["trailer_share"] == "AI标题":
-            title = GPT4o.get_ai_title(
+            title = GPT4oMini.get_ai_mini_title(
                 original_title if data["title_category"] == "原标题" else data["title_category"])
         else:
             title = original_title if data["title_category"] == "原标题" else data["title_category"]

+ 1 - 2
common/download_video.py

@@ -1,4 +1,3 @@
-import html
 import os
 from urllib.parse import urlparse
 
@@ -26,7 +25,7 @@ class DownLoad:
             for i in range(3):
                 try:
                     response = requests.request("GET", video_url, headers=headers, data=payload, timeout=240)
-                    if response.status_code == 200:
+                    if response.status_code == 206:
                         # 以二进制写入模式打开文件
 
                         with open(f"{video}", "wb") as file:

+ 1 - 1
common/gpt4o_mini_help.py

@@ -151,7 +151,7 @@ class GPT4oMini:
                 return new_title
             except Exception as e:
                 retry_count += 1
-                logger.error(f"尝试第 {retry_count} 次失败,错误: {e}")
+                # logger.error(f"尝试第 {retry_count} 次失败,错误: {e}")
                 time.sleep(1)  # 延迟1秒后重试
         return "这个视频,分享给我的老友,祝愿您能幸福安康"