allow skip current image in progress api

Author: evshiron

modules/api/api.py

@@ -167,7 +167,7 @@ def pnginfoapi(self, req: PNGInfoRequest):
 
         return PNGInfoResponse(info=result[1])
 
-    def progressapi(self):
+    def progressapi(self, req: ProgressRequest = Depends()):
         # copy from check_progress_call of ui.py
 
         if shared.state.job_count == 0:
@@ -188,7 +188,7 @@ def progressapi(self):
         progress = min(progress, 1)
 
         current_image = None
-        if shared.state.current_image:
+        if shared.state.current_image and not req.skip_current_image:
             current_image = encode_pil_to_base64(shared.state.current_image)
 
         return ProgressResponse(progress=progress, eta_relative=eta_relative, state=shared.state.dict(), current_image=current_image)

modules/api/models.py

@@ -157,6 +157,9 @@ class PNGInfoRequest(BaseModel):
 class PNGInfoResponse(BaseModel):
     info: str = Field(title="Image info", description="A string with all the info the image had")
 
+class ProgressRequest(BaseModel):
+    skip_current_image: bool = Field(default=False, title="Skip current image", description="Skip current image serialization")
+
 class ProgressResponse(BaseModel):
     progress: float = Field(title="Progress", description="The progress with a range of 0 to 1")
     eta_relative: float = Field(title="ETA in secs")