From 0f808a62df43e33cda12da73368a4d3fe8dbee46 Mon Sep 17 00:00:00 2001
From: Maruhi <38591025+maruhidd@users.noreply.github.com>
Date: Wed, 25 Dec 2024 20:58:00 +0900
Subject: [PATCH] Add import base64

---
 server.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server.py b/server.py
index 57f062ad..26c464d7 100644
--- a/server.py
+++ b/server.py
@@ -336,6 +336,7 @@ class PromptServer():
                 for key, value in json_data.items():
                     if key == "image":
                         file_name = json_data.get("filename")
+                        import base64
                         value = base64.b64decode(value)
                         import io
                         image = aiohttp.web.FileField(name=key, filename=file_name, file=io.BytesIO(value), content_type='image/png', headers=None)