mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2026-01-09 06:50:17 -06:00
Code Issues Packages Projects Releases Wiki Activity

changed bytes to base64

Browse Source
This commit is contained in:
Dillon DuPont
2025-05-15 15:24:13 -04:00
parent 667be22ee5
commit 56438440e6
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/computer-server/computer_server/diorama/diorama.py
View File

@@ -149,17 +149,20 @@ class Diorama:
logger.warning(f"Command was cancelled: {action}")
return None
async def screenshot(self, as_bytes: bool = True) -> Union[bytes, Image]:
async def screenshot(self, as_bytes: bool = True) -> Union[str, Image.Image]:
import base64
result, img = await self._send_cmd("screenshot")
self._scene_hitboxes = result.get("hitboxes", [])
self._scene_size = img.size
if as_bytes:
# PIL Image to bytes
# PIL Image to bytes, then base64 encode for JSON
import io
img_byte_arr = io.BytesIO()
img.save(img_byte_arr, format="PNG")
img_byte_arr = img_byte_arr.getvalue()
return img_byte_arr
img_bytes = img_byte_arr.getvalue()
img_b64 = base64.b64encode(img_bytes).decode("ascii")
return img_b64
else:
return img