From fd92d523393ec940711255be9a009bb6f3faa8c9 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 6 Jan 2024 20:32:38 -0600 Subject: [PATCH] send empty progress with initial job --- api/onnx_web/chain/edit_metadata.py | 4 +++- api/onnx_web/server/api.py | 5 +++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/chain/edit_metadata.py b/api/onnx_web/chain/edit_metadata.py index 61aae34c..4ddebe17 100644 --- a/api/onnx_web/chain/edit_metadata.py +++ b/api/onnx_web/chain/edit_metadata.py @@ -1,6 +1,6 @@ from typing import Any, Optional -from ..params import HighresParams, ImageParams, Size, StageParams, UpscaleParams +from ..params import HighresParams, ImageParams, Size, SizeChart, StageParams, UpscaleParams from ..server import ServerContext from ..worker import WorkerContext from .base import BaseStage @@ -8,6 +8,8 @@ from .result import StageResult class EditMetadataStage(BaseStage): + max_tile = SizeChart.max + def run( self, _worker: WorkerContext, diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 4bdf4ef8..40fb6c10 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -97,6 +97,11 @@ def job_reply(name: str): return jsonify( { "name": name, + "queue": Progress(0, 0).tojson(), # TODO: use real queue position + "status": JobStatus.PENDING, + "stages": Progress(0, 0).tojson(), + "steps": Progress(0, 0).tojson(), + "tiles": Progress(0, 0).tojson(), } )