优化首帧功能,如果没传入首帧,则提取首帧。
This commit is contained in:
@@ -15,7 +15,7 @@ from pymediainfo import MediaInfo
|
||||
import pysilk
|
||||
from pydub import AudioSegment
|
||||
|
||||
from utils.video_utils import _get_first_frame
|
||||
from utils.video_utils import _get_first_frame, _get_first_frame_bytes
|
||||
from wechat_ipad import UserLoggedOut
|
||||
from wechat_ipad.client.base import WechatAPIClientBase
|
||||
|
||||
@@ -231,6 +231,11 @@ class MessageMixin(WechatAPIClientBase):
|
||||
vid_base64 = base64.b64encode(video).decode()
|
||||
file_len = len(video)
|
||||
media_info = MediaInfo.parse(BytesIO(video))
|
||||
# 如果没有传入首帧,则自己提取一次
|
||||
if not has_image:
|
||||
first_frame = _get_first_frame_bytes(video, f"frame_{int(time.time())}.jpg")
|
||||
if first_frame:
|
||||
image = first_frame
|
||||
elif isinstance(video, os.PathLike):
|
||||
video_path = Path(video)
|
||||
if not video_path.exists():
|
||||
@@ -242,7 +247,7 @@ class MessageMixin(WechatAPIClientBase):
|
||||
media_info = MediaInfo.parse(video_path)
|
||||
# 如果没有传入首帧,则自己提取一次
|
||||
if not has_image:
|
||||
first_frame = _get_first_frame(video_path, f"dify_frame_{int(time.time())}.jpg")
|
||||
first_frame = _get_first_frame(video_path, f"frame_{int(time.time())}.jpg")
|
||||
if first_frame:
|
||||
image = first_frame
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user