From 5bc186badbf1afcdb7f183b7692c116453a64bfc Mon Sep 17 00:00:00 2001 From: Yazhou Cao Date: Tue, 23 Apr 2024 09:39:21 -0700 Subject: [PATCH] Fix lint errors --- vision_agent/agent/vision_agent.py | 8 +++++--- vision_agent/image_utils.py | 2 +- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index a15eddaa..81224e44 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -486,14 +486,16 @@ def log_progress(self, description: str) -> None: if self.report_progress_callback: self.report_progress_callback(description) - def _report_visualization_via_callback(self, images: List[Union[str, Path]]) -> None: + def _report_visualization_via_callback( + self, images: Sequence[Union[str, Path]] + ) -> None: """This is intended for streaming the visualization images via the callback to the client side.""" if self.report_progress_callback: - self.report_progress_callback("" ) + self.report_progress_callback("") if images: for img in images: self.report_progress_callback(f"{convert_to_b64(img)}") - self.report_progress_callback("" ) + self.report_progress_callback("") def chat_with_workflow( self, diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py index 5a28df1d..4786f84b 100644 --- a/vision_agent/image_utils.py +++ b/vision_agent/image_utils.py @@ -108,7 +108,7 @@ def convert_to_b64(data: Union[str, Path, np.ndarray, ImageType]) -> str: data = Image.open(data) if isinstance(data, Image.Image): buffer = BytesIO() - data.convert("RGB").save(buffer, format="JPEG") + data.convert("RGB").save(buffer, format="PNG") return base64.b64encode(buffer.getvalue()).decode("utf-8") else: arr_bytes = data.tobytes()