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()