From 2d3d096fd7bc78472e3a851b7c10b5712d145200 Mon Sep 17 00:00:00 2001 From: GinHa <15201596918@163.com> Date: Fri, 17 Jul 2026 14:58:05 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8DLLM=20base64=E8=B5=84?= =?UTF-8?q?=E6=BA=90=E5=AE=9A=E4=BD=8DBUG?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- video-gen-api/app/utils/media.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/video-gen-api/app/utils/media.py b/video-gen-api/app/utils/media.py index 77e7cb87..29e966a7 100644 --- a/video-gen-api/app/utils/media.py +++ b/video-gen-api/app/utils/media.py @@ -46,8 +46,14 @@ async def media_to_base64(url: str, fallback_mime: str = "image/png", max_mb: in return f"data:{mime};base64,{b64}" # 本地路径 - relative_path = url.replace("/uploads/", "", 1).lstrip("/") - file_path = os.path.join(settings.UPLOAD_LOCAL_PATH, relative_path) + if url.startswith("/generate/"): + type_pre = '/generate/videos/' if url.startswith("/generate/videos/") else '/generate/images/' + type_path = settings.STORAGE_LOCAL_PATH if url.startswith("/generate/videos/") else settings.STORAGE_IMAGE_LOCAL_PATH + relative_path = url.replace(type_pre, "", 1).lstrip("/") + file_path = os.path.join(type_path, relative_path) + else: + relative_path = url.replace("/uploads/", "", 1).lstrip("/") + file_path = os.path.join(settings.UPLOAD_LOCAL_PATH, relative_path) mime = mimetypes.guess_type(file_path)[0] or fallback_mime with open(file_path, "rb") as f: b64 = base64.b64encode(f.read()).decode()