diff --git a/components/pages/work-flow/use-workflow-data.tsx b/components/pages/work-flow/use-workflow-data.tsx index 13fa349..4258ea4 100644 --- a/components/pages/work-flow/use-workflow-data.tsx +++ b/components/pages/work-flow/use-workflow-data.tsx @@ -250,7 +250,6 @@ export function useWorkflowData({}: UseWorkflowDataProps = {}) { } taskCurrent.scenes.total_count = task.task_result.total_count; if (task.task_status !== 'COMPLETED' || taskCurrent.scenes.total_count !== realSketchResultData.length) { - console.log('task.task_status', task.task_status, 'taskCurrent.scenes.total_count', taskCurrent.scenes.total_count, 'realSketchResultData.length', realSketchResultData.length); if (task.task_status === 'FAILED' && taskCurrent.scenes.total_count === realSketchResultData.length) { continue; } @@ -457,7 +456,7 @@ export function useWorkflowData({}: UseWorkflowDataProps = {}) { characterList.push({ name: character.character_name, url: character.image_path, - status: character.image_path ? 1 : (data.character.task_status === 'COMPLETED' ? 2 : 0), + status: character.image_path ? 1 : ((data.character.task_status === 'COMPLETED' || data.character.task_status === 'FAILED') ? 2 : 0), type: 'role' }); } @@ -477,7 +476,7 @@ export function useWorkflowData({}: UseWorkflowDataProps = {}) { sketchList.push({ url: sketch.image_path, script: sketch.sketch_name, - status: sketch.image_path ? 1 : (data.sketch.task_status === 'COMPLETED' ? 2 : 0), + status: sketch.image_path ? 1 : ((data.sketch.task_status === 'COMPLETED' || data.sketch.task_status === 'FAILED') ? 2 : 0), type: 'scene' }); }