diff --git a/__init__.py b/__init__.py index 77187b2..df0e959 100644 --- a/__init__.py +++ b/__init__.py @@ -1,5 +1,5 @@ -from gemini.gemini_image_node import * -from gemini.gemini_image_preset_node import * +from .gemini.gemini_image_node import * +from .gemini.gemini_image_preset_node import * from typing_extensions import override class APIExtension(ComfyExtension): diff --git a/gemini/__init__.py b/gemini/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/gemini/gemini_image_node.py b/gemini/gemini_image_node.py index 22e23b8..02c171c 100644 --- a/gemini/gemini_image_node.py +++ b/gemini/gemini_image_node.py @@ -1,5 +1,6 @@ import io import os +import sys import json import base64 import requests @@ -23,7 +24,7 @@ class GeminiImage(io.ComfyNode): 如果获取不到,返回默认模型列表 """ try: - config_path = os.path.join(os.path.dirname(__file__), "config.json") + config_path = os.path.join(os.path.dirname(__file__), '..', "config.json") if not os.path.exists(config_path): return ["gemini-2.5-flash-image"] @@ -45,7 +46,7 @@ class GeminiImage(io.ComfyNode): 从config.json中加载并验证API凭据 返回 (base_url, api_key) 元组 """ - config_path = os.path.join(os.path.dirname(__file__), "config.json") + config_path = os.path.join(os.path.dirname(__file__), '..', "config.json") # 检查配置文件是否存在 if not os.path.exists(config_path): diff --git a/gemini/gemini_image_preset_node.py b/gemini/gemini_image_preset_node.py index c075822..69a655a 100644 --- a/gemini/gemini_image_preset_node.py +++ b/gemini/gemini_image_preset_node.py @@ -9,7 +9,7 @@ from PIL import Image from io import BytesIO from typing_extensions import override from comfy_api.latest import ComfyExtension, io -from .image_utils import tensor_to_base64_string +from ..utils.image_utils import tensor_to_base64_string # 公共函数,用于处理预设数据 def load_preset_data(): diff --git a/utils/__init__.py b/utils/__init__.py new file mode 100644 index 0000000..e69de29