Merge branch 'main' of https://gitee.com/wg123/video-gen

video-gen-app/src/api/index.ts

@@ -165,6 +165,43 @@ export async function uploadVideo(file: File, durationSeconds?: number): Promise
   return await res.json();
 }
 
+function privatePortraitUploadEndpoint(path: string, durationSeconds?: number): string {
+  const base = `${import.meta.env.VITE_API_BASE || 'http://localhost:8000'}/api${path}`;
+  const query = typeof durationSeconds === 'number' && durationSeconds > 0
+    ? `?duration_seconds=${encodeURIComponent(String(durationSeconds))}`
+    : '';
+  return `${base}${query}`;
+}
+
+async function uploadPrivatePortraitFile(path: string, file: File, durationSeconds?: number, errorMessage = '素材上传失败'): Promise<UploadResourceResult> {
+  const form = new FormData();
+  form.append('file', file);
+  const token = localStorage.getItem('auth_token');
+  const res = await fetch(privatePortraitUploadEndpoint(path, durationSeconds), {
+    method: 'POST',
+    headers: token ? { Authorization: `Bearer ${token}` } : {},
+    body: form,
+  });
+  if (!res.ok) throw new Error(errorMessage);
+  return await res.json();
+}
+
+export async function uploadPrivatePortraitImage(file: File): Promise<UploadResourceResult> {
+  return uploadPrivatePortraitFile('/private-portrait/uploads/image', file, undefined, '真人图片素材上传失败');
+}
+
+export async function uploadPrivatePortraitVideo(file: File, durationSeconds?: number): Promise<UploadResourceResult> {
+  return uploadPrivatePortraitFile('/private-portrait/uploads/video', file, durationSeconds, '真人视频素材上传失败');
+}
+
+export async function uploadPrivatePortraitVirtualImage(file: File): Promise<UploadResourceResult> {
+  return uploadPrivatePortraitFile('/private-portrait/virtual/uploads/image', file, undefined, '虚拟图片素材上传失败');
+}
+
+export async function uploadPrivatePortraitVirtualVideo(file: File, durationSeconds?: number): Promise<UploadResourceResult> {
+  return uploadPrivatePortraitFile('/private-portrait/virtual/uploads/video', file, durationSeconds, '虚拟视频素材上传失败');
+}
+
 export async function uploadHotOpeningImage(file: File): Promise<UploadResourceResult> {
   const form = new FormData();
   form.append('file', file);
@@ -962,7 +999,7 @@ export async function getPrivatePortraitValidateSession(sessionId: string): Prom
   return api.get<PrivatePortraitValidateSession>(`/private-portrait/validate-sessions/${sessionId}`);
 }
 
-export async function createPrivatePortraitAsset(projectId: string, payload: { url: string; assetType?: string; name?: string | null; videoDuration?: number | null; videoCoverUrl?: string | null; fileSize?: number | null; mimeType?: string | null }): Promise<PrivatePortraitAsset> {
+export async function createPrivatePortraitAsset(projectId: string, payload: { url: string; assetType?: string; name?: string | null; videoDuration?: number | null; videoCoverUrl?: string | null; fileSize?: number | null; mimeType?: string | null; uploadResourceId?: string | null }): Promise<PrivatePortraitAsset> {
   return api.post<PrivatePortraitAsset>(`/private-portrait/projects/${projectId}/assets`, {
     url: payload.url,
     asset_type: payload.assetType || 'Image',
@@ -971,6 +1008,7 @@ export async function createPrivatePortraitAsset(projectId: string, payload: { u
     video_cover_url: payload.videoCoverUrl || null,
     file_size: payload.fileSize ?? null,
     mime_type: payload.mimeType || null,
+    upload_resource_id: payload.uploadResourceId || null,
   });
 }
 
@@ -1032,7 +1070,7 @@ export async function deletePrivatePortraitVirtualProject(projectId: string): Pr
   await api.delete(`/private-portrait/virtual-projects/${projectId}`);
 }
 
-export async function createPrivatePortraitVirtualAsset(projectId: string, payload: { url: string; assetType?: string; name?: string | null; videoDuration?: number | null; videoCoverUrl?: string | null; fileSize?: number | null; mimeType?: string | null }): Promise<PrivatePortraitAsset> {
+export async function createPrivatePortraitVirtualAsset(projectId: string, payload: { url: string; assetType?: string; name?: string | null; videoDuration?: number | null; videoCoverUrl?: string | null; fileSize?: number | null; mimeType?: string | null; uploadResourceId?: string | null }): Promise<PrivatePortraitAsset> {
   return api.post<PrivatePortraitAsset>(`/private-portrait/virtual-projects/${projectId}/assets`, {
     url: payload.url,
     asset_type: payload.assetType || 'Image',
@@ -1041,6 +1079,7 @@ export async function createPrivatePortraitVirtualAsset(projectId: string, paylo
     video_cover_url: payload.videoCoverUrl || null,
     file_size: payload.fileSize ?? null,
     mime_type: payload.mimeType || null,
+    upload_resource_id: payload.uploadResourceId || null,
   });
 }