diff --git a/chatchat-server/chatchat/webui_pages/utils.py b/chatchat-server/chatchat/webui_pages/utils.py
index 11d9edc7..47d60176 100644
--- a/chatchat-server/chatchat/webui_pages/utils.py
+++ b/chatchat-server/chatchat/webui_pages/utils.py
@@ -20,6 +20,7 @@ import httpx
 import contextlib
 import json
 import os
+import base64
 from io import BytesIO
 from chatchat.server.utils import set_httpx_config, api_address, get_httpx_client