From e30f375355505d407dfd8b93b716eaecbccd1a14 Mon Sep 17 00:00:00 2001
From: pandy <1272175942@qq.com>
Date: Fri, 20 Mar 2026 00:25:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore                      | Bin 251 -> 277 bytes
 gittools.py                     | 590 ++++++++++++++++++++++++++++++++
 py/IMGE_GeminiNode_apikey.py    |  22 +-
 py/IMGE_GeminiNode_authtoken.py |  22 +-
 py/IMGE_GeminiPro_apikey.py     |  14 +-
 py/IMGE_GeminiPro_authtoken.py  |  14 +-
 6 files changed, 646 insertions(+), 16 deletions(-)
 create mode 100644 gittools.py

diff --git a/.gitignore b/.gitignore
index d50b56d7d7e1898cd917df864dc808c129b8075b..600fab2a50e208b1cdf296d5a8c14fe49477c104 100644
GIT binary patch
delta 9
Qcmey(IF)I_?}-nk02fXKs{jB1

delta 6
NcmbQr^qX<QZvY9H10MhY

diff --git a/gittools.py b/gittools.py
new file mode 100644
index 0000000..20239c3
--- /dev/null
+++ b/gittools.py
@@ -0,0 +1,590 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Git 自动同步工具
+用法: python gitrun.py [命令]
+
+命令说明:
+  push         - 强制推送本地到远程（直接覆盖，保留历史记录）
+  pull         - 强制拉取远程到本地（直接覆盖，保留历史记录）
+  sync         - 智能同步（先拉取合并，再推送）
+  log          - 查看提交历史并支持快速回退版本
+  history      - 查看完整提交历史
+  config       - 配置远程仓库地址
+
+
+
+
+重要说明:
+  • push/pull 都会保留完整的 Git 历史记录，可以随时回退
+  • push 是纯粹的本地覆盖远程，不会先拉取远程内容
+  • pull 是纯粹的远程覆盖本地，不会保留本地未推送的修改
+  • sync 才会进行智能合并
+  • log 命令可以查看历史并直接回退版本（输入数字选择，0或回车跳过）
+  • 所有历史提交都保存在 .git 目录中，永不丢失
+"""
+
+import os
+import sys
+import subprocess
+import json
+from pathlib import Path
+from datetime import datetime
+
+CONFIG_FILE = Path.home() / '.gitrun_config.json'
+
+
+class GitSyncTool:
+    def __init__(self):
+        self.config = self.load_config()
+        self.repo_path = os.getcwd()
+        
+    def load_config(self):
+        """加载配置文件"""
+        if CONFIG_FILE.exists():
+            with open(CONFIG_FILE, 'r', encoding='utf-8') as f:
+                return json.load(f)
+        return {}
+    
+    def save_config(self):
+        """保存配置文件"""
+        with open(CONFIG_FILE, 'w', encoding='utf-8') as f:
+            json.dump(self.config, f, indent=2, ensure_ascii=False)
+    
+    def run_command(self, cmd, check=True, silent=False):
+        """执行命令并返回结果"""
+        try:
+            # Windows 上使用 GBK 编码，其他系统使用 UTF-8
+            encoding = 'gbk' if sys.platform == 'win32' else 'utf-8'
+            
+            result = subprocess.run(
+                cmd,
+                shell=True,
+                cwd=self.repo_path,
+                capture_output=True,
+                text=True,
+                encoding=encoding,
+                errors='ignore'  # 忽略无法解码的字符
+            )
+            if check and result.returncode != 0:
+                if not silent:
+                    print(f"❌ 命令执行失败: {cmd}")
+                    if result.stderr:
+                        print(f"错误信息: {result.stderr}")
+                return None
+            return result
+        except Exception as e:
+            if not silent:
+                print(f"❌ 执行命令时出错: {e}")
+            return None
+    
+    def is_git_repo(self):
+        """检查当前目录是否是 Git 仓库"""
+        result = self.run_command("git rev-parse --is-inside-work-tree", check=False, silent=True)
+        return result and result.returncode == 0
+    
+    def get_current_branch(self):
+        """获取当前分支名"""
+        result = self.run_command("git branch --show-current", silent=True)
+        return result.stdout.strip() if result else None
+    
+    def has_changes(self):
+        """检查是否有未提交的更改"""
+        result = self.run_command("git status --porcelain", silent=True)
+        return bool(result and result.stdout.strip())
+    
+    def get_remote_url(self):
+        """获取远程仓库地址"""
+        result = self.run_command("git remote get-url origin", check=False, silent=True)
+        return result.stdout.strip() if result and result.returncode == 0 else None
+    
+    def config_remote(self):
+        """配置远程仓库地址"""
+        print("🔧 配置远程仓库地址")
+        
+        # 检查是否是 Git 仓库，如果不是则初始化
+        if not self.is_git_repo():
+            print("⚠️  当前目录不是 Git 仓库")
+            init = input("是否初始化为 Git 仓库? (y/n): ").strip().lower()
+            if init == 'y':
+                result = self.run_command("git init")
+                if result:
+                    print("✅ Git 仓库初始化成功")
+                else:
+                    print("❌ Git 仓库初始化失败")
+                    return
+            else:
+                print("❌ 操作已取消")
+                return
+        
+        current_url = self.get_remote_url()
+        if current_url:
+            print(f"当前远程仓库: {current_url}")
+            change = input("是否修改? (y/n): ").strip().lower()
+            if change != 'y':
+                return
+        
+        remote_url = input("请输入远程仓库地址 (如: https://github.com/username/repo.git): ").strip()
+        
+        if not remote_url:
+            print("❌ 远程仓库地址不能为空")
+            return
+        
+        # 检查是否已有 origin
+        if current_url:
+            result = self.run_command(f"git remote set-url origin {remote_url}")
+        else:
+            result = self.run_command(f"git remote add origin {remote_url}")
+        
+        if result:
+            print(f"✅ 远程仓库配置成功: {remote_url}")
+            
+            # 保存到配置文件
+            repo_name = os.path.basename(self.repo_path)
+            if repo_name not in self.config:
+                self.config[repo_name] = {}
+            self.config[repo_name]['remote_url'] = remote_url
+            self.save_config()
+            
+            # 询问是否设置默认分支
+            print("\n💡 提示: 首次推送前需要设置默认分支")
+            set_branch = input("是否设置默认分支为 main? (y/n): ").strip().lower()
+            if set_branch == 'y':
+                # 检查当前分支
+                current_branch = self.get_current_branch()
+                if not current_branch:
+                    # 创建初始提交
+                    print("\n📝 创建初始提交...")
+                    self.run_command("git add .")
+                    self.run_command('git commit -m "Initial commit" --allow-empty')
+                    self.run_command("git branch -M main")
+                    print("✅ 默认分支已设置为 main")
+                elif current_branch != 'main':
+                    self.run_command("git branch -M main")
+                    print("✅ 分支已重命名为 main")
+                else:
+                    print("✅ 当前分支已是 main")
+    
+    def commit_changes(self, message=None):
+        """提交本地更改"""
+        if not self.has_changes():
+            print("ℹ️  没有需要提交的更改")
+            return True
+        
+        print("\n📝 检测到以下更改:")
+        self.run_command("git status --short")
+        
+        if message is None:
+            message = input("\n请输入提交信息: ").strip()
+        
+        if not message:
+            print("❌ 提交信息不能为空")
+            return False
+        
+        # 添加所有更改
+        print("\n📦 正在暂存所有更改...")
+        result = self.run_command("git add -A")
+        if not result:
+            return False
+        
+        # 提交更改
+        print(f"💾 正在提交: {message}")
+        result = self.run_command(f'git commit -m "{message}"')
+        if result:
+            print(f"✅ 提交成功")
+            return True
+        return False
+    
+    def force_push(self):
+        """纯粹的本地覆盖远程（不拉取，直接强制推送）"""
+        print("🚀 强制推送本地到远程")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        branch = self.get_current_branch()
+        if not branch:
+            print("❌ 无法获取当前分支")
+            return
+        
+        print(f"📍 当前分支: {branch}")
+        
+        # 检查远程仓库
+        remote_url = self.get_remote_url()
+        if not remote_url:
+            print("⚠️  未配置远程仓库")
+            self.config_remote()
+            remote_url = self.get_remote_url()
+            if not remote_url:
+                return
+        
+        print(f"🌐 远程仓库: {remote_url}")
+        
+        # 提交本地更改
+        print()
+        if not self.commit_changes():  # 改为交互式提交，不自动
+            print("❌ 提交失败，推送中止")
+            return
+        
+        # 强制推送（不拉取远程）
+        print(f"\n⬆️  正在强制推送到远程 ({branch})...")
+        result = self.run_command(f"git push -f origin {branch}")
+        
+        if result:
+            print(f"\n✅ 推送成功!")
+        else:
+            print("\n❌ 推送失败")
+    
+    def force_pull(self):
+        """纯粹的远程覆盖本地（不保留本地未推送的修改）"""
+        print("⬇️  强制拉取远程到本地")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        branch = self.get_current_branch()
+        if not branch:
+            print("❌ 无法获取当前分支")
+            return
+        
+        print(f"📍 当前分支: {branch}")
+        
+        # 检查远程仓库
+        remote_url = self.get_remote_url()
+        if not remote_url:
+            print("⚠️  未配置远程仓库")
+            self.config_remote()
+            return
+        
+        print(f"🌐 远程仓库: {remote_url}")
+        
+        # 检查本地更改
+        if self.has_changes():
+            print("\n⚠️  检测到未提交的本地更改，这些更改将会被丢弃")
+            self.run_command("git status --short")
+        
+        # 先获取远程最新信息
+        print(f"\n📡 正在获取远程仓库信息...")
+        self.run_command("git fetch origin")
+        
+        # 强制重置到远程版本
+        print(f"⬇️  正在强制拉取远程 ({branch})...")
+        self.run_command(f"git reset --hard origin/{branch}")
+        self.run_command("git clean -fd")
+        
+        print(f"\n✅ 拉取成功!")
+    
+    def smart_sync(self):
+        """智能同步（先拉取合并，再推送）"""
+        print("🔄 智能同步")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        branch = self.get_current_branch()
+        if not branch:
+            print("❌ 无法获取当前分支")
+            return
+        
+        print(f"📍 当前分支: {branch}")
+        
+        # 检查远程仓库
+        remote_url = self.get_remote_url()
+        if not remote_url:
+            print("⚠️  未配置远程仓库")
+            self.config_remote()
+            return
+        
+        print(f"🌐 远程仓库: {remote_url}")
+        
+        # 提交本地更改
+        print()
+        if self.has_changes():
+            if not self.commit_changes():  # 改为交互式提交
+                print("❌ 提交失败")
+                return
+        
+        # 拉取远程更改
+        print(f"\n⬇️  正在拉取远程更新 ({branch})...")
+        result = self.run_command(f"git pull origin {branch}", check=False)
+        
+        if result and result.returncode != 0:
+            # 如果有冲突
+            if "conflict" in result.stderr.lower() or "conflict" in result.stdout.lower():
+                print("\n⚠️  检测到合并冲突!")
+                print("\n冲突文件:")
+                self.run_command("git diff --name-only --diff-filter=U")
+                
+                choice = input("\n选择操作:\n1. 保留本地版本\n2. 保留远程版本\n3. 手动解决后继续\n请选择 (1/2/3): ").strip()
+                
+                if choice == '1':
+                    self.run_command("git checkout --ours .")
+                    self.run_command("git add -A")
+                    self.run_command('git commit --no-edit')
+                    print("✅ 已保留本地版本")
+                elif choice == '2':
+                    self.run_command("git checkout --theirs .")
+                    self.run_command("git add -A")
+                    self.run_command('git commit --no-edit')
+                    print("✅ 已保留远程版本")
+                else:
+                    print("ℹ️  请手动解决冲突后执行: git add . && git commit")
+                    return
+            else:
+                print("❌ 拉取失败")
+                return
+        else:
+            print("✅ 拉取成功")
+        
+        # 推送到远程
+        print(f"\n⬆️  正在推送到远程 ({branch})...")
+        push_result = self.run_command(f"git push origin {branch}")
+        
+        if push_result:
+            print(f"\n✅ 同步成功!")
+        else:
+            print("⚠️  推送失败或无新内容需要推送")
+    
+    def show_log(self, num=10):
+        """查看提交历史并支持直接回退"""
+        print(f"📋 最近 {num} 次提交历史")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        branch = self.get_current_branch()
+        print(f"📍 当前分支: {branch}\n")
+        
+        # 使用 git log 的简单格式，然后解析
+        result = self.run_command(f'git log -{num} --format="COMMIT_START%nHASH:%H%nSUBJECT:%s%nAUTHOR:%an%nDATE:%ar%nCOMMIT_END"')
+        
+        if not result:
+            return
+        
+        commits = result.stdout.strip().split('COMMIT_START')
+        commit_hashes = []
+        idx = 1
+        
+        for commit in commits:
+            if not commit.strip():
+                continue
+                
+            lines = commit.strip().split('\n')
+            commit_info = {}
+            for line in lines:
+                if ':' in line:
+                    key, value = line.split(':', 1)
+                    commit_info[key] = value.strip()
+            
+            if 'HASH' in commit_info:
+                hash_full = commit_info['HASH']
+                hash_short = hash_full[:7]
+                subject = commit_info.get('SUBJECT', '')
+                date = commit_info.get('DATE', '')
+                
+                commit_hashes.append(hash_full)
+                print(f"{idx}. [{hash_short}] {subject} -- {date}")
+                idx += 1
+        
+        # 询问是否回退
+        print()
+        choice = input("是否回退版本? (输入数字1-10回退，直接回车或输入0跳过): ").strip()
+        
+        if not choice or choice == '0':
+            print("✅ 未进行回退")
+            return
+        
+        # 判断是数字还是哈希值
+        if choice.isdigit():
+            idx = int(choice) - 1
+            if 0 <= idx < len(commit_hashes):
+                commit_hash = commit_hashes[idx][:7]
+            else:
+                print(f"❌ 无效的数字，请输入 1-{len(commit_hashes)}")
+                return
+        else:
+            commit_hash = choice
+        
+        # 验证提交是否存在
+        verify = self.run_command(f"git cat-file -t {commit_hash}", check=False, silent=True)
+        if not verify or verify.returncode != 0:
+            print(f"❌ 提交 {commit_hash} 不存在")
+            return
+        
+        # 显示目标提交信息
+        print(f"\n📍 即将回退到: {commit_hash}")
+        
+        # 选择回退方式
+        print("\n回退方式:")
+        print("1. 软回退 (保留工作区和暂存区的更改)")
+        print("2. 混合回退 (保留工作区更改，清空暂存区) [推荐]")
+        print("3. 硬回退 (完全回退，丢弃所有更改)")
+        
+        mode_choice = input("\n请选择 (1/2/3，直接回车默认选2): ").strip()
+        
+        if not mode_choice:
+            mode_choice = '2'
+        
+        mode_map = {
+            '1': ('--soft', '软回退'),
+            '2': ('--mixed', '混合回退'),
+            '3': ('--hard', '硬回退')
+        }
+        
+        if mode_choice not in mode_map:
+            print("❌ 无效选择")
+            return
+        
+        mode, mode_name = mode_map[mode_choice]
+        
+        # 执行回退
+        print(f"\n⏮️  正在执行{mode_name}...")
+        result = self.run_command(f"git reset {mode} {commit_hash}")
+        
+        if result:
+            print(f"\n✅ {mode_name}成功!")
+            print(f"📍 已回退到: {commit_hash}\n")
+            self.run_command("git status --short")
+        else:
+            print(f"\n❌ 回退失败")
+    
+    def show_full_history(self):
+        """查看完整提交历史"""
+        print("📚 完整提交历史")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        branch = self.get_current_branch()
+        print(f"📍 当前分支: {branch}\n")
+        
+        # 显示完整历史（简化格式以兼容 Windows）
+        self.run_command("git log --graph --oneline --decorate --all")
+    
+    def rollback(self):
+        """回退到指定版本"""
+        print("⏮️  版本回退")
+        
+        if not self.is_git_repo():
+            print("❌ 当前目录不是 Git 仓库")
+            return
+        
+        # 显示最近10次提交
+        print("📋 最近10次提交:\n")
+        result = self.run_command("git log -10 --pretty=format:%H|%s|%an|%ar")
+        
+        if not result:
+            return
+        
+        lines = result.stdout.strip().split('\n')
+        commits = []
+        
+        for i, line in enumerate(lines, 1):
+            if '|' in line:
+                hash_full, subject, author, date = line.split('|', 3)
+                hash_short = hash_full[:7]
+                commits.append(hash_full)
+                print(f"{i}. [{hash_short}] {subject}")
+                print(f"   作者: {author} | 时间: {date}\n")
+        
+        print()
+        
+        # 获取用户输入
+        choice = input("请输入要回退到的版本 (输入数字1-10，或输入提交哈希值，q 取消): ").strip()
+        
+        if not choice or choice.lower() == 'q':
+            print("❌ 操作已取消")
+            return
+        
+        # 判断是数字还是哈希值
+        if choice.isdigit():
+            idx = int(choice) - 1
+            if 0 <= idx < len(commits):
+                commit_hash = commits[idx][:7]
+            else:
+                print(f"❌ 无效的数字，请输入 1-{len(commits)}")
+                return
+        else:
+            commit_hash = choice
+        
+        # 验证提交是否存在
+        verify = self.run_command(f"git cat-file -t {commit_hash}", check=False, silent=True)
+        if not verify or verify.returncode != 0:
+            print(f"❌ 提交 {commit_hash} 不存在")
+            return
+        
+        # 显示目标提交信息
+        print(f"\n📍 即将回退到:")
+        self.run_command(f"git log -1 {commit_hash} --pretty=format:[%h] %s%n作者: %an | 时间: %ar")
+        
+        # 选择回退方式
+        print("\n\n回退方式:")
+        print("1. 软回退 (保留工作区和暂存区的更改)")
+        print("2. 混合回退 (保留工作区更改，清空暂存区) [推荐]")
+        print("3. 硬回退 (完全回退，丢弃所有更改)")
+        
+        mode_choice = input("\n请选择 (1/2/3，输入 q 取消): ").strip()
+        
+        if mode_choice.lower() == 'q':
+            print("❌ 操作已取消")
+            return
+        
+        mode_map = {
+            '1': ('--soft', '软回退'),
+            '2': ('--mixed', '混合回退'),
+            '3': ('--hard', '硬回退')
+        }
+        
+        if mode_choice not in mode_map:
+            print("❌ 无效选择")
+            return
+        
+        mode, mode_name = mode_map[mode_choice]
+        
+        # 执行回退
+        print(f"\n⏮️  正在执行{mode_name}...")
+        result = self.run_command(f"git reset {mode} {commit_hash}")
+        
+        if result:
+            print(f"\n✅ {mode_name}成功!")
+            print(f"📍 已回退到: {commit_hash}\n")
+            self.run_command("git status --short")
+        else:
+            print(f"\n❌ 回退失败")
+
+
+def main():
+    tool = GitSyncTool()
+    
+    if len(sys.argv) < 2:
+        print(__doc__)
+        return
+    
+    command = sys.argv[1].lower()
+    
+    if command == 'push':
+        tool.force_push()
+    elif command == 'pull':
+        tool.force_pull()
+    elif command == 'sync':
+        tool.smart_sync()
+    elif command == 'log':
+        tool.show_log(5)
+    elif command == 'history':
+        tool.show_full_history()
+    elif command == 'rollback':
+        tool.rollback()
+    elif command == 'config':
+        tool.config_remote()
+    else:
+        print(f"❌ 未知命令: {command}")
+        print(__doc__)
+
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file
diff --git a/py/IMGE_GeminiNode_apikey.py b/py/IMGE_GeminiNode_apikey.py
index b72c782..78aa482 100644
--- a/py/IMGE_GeminiNode_apikey.py
+++ b/py/IMGE_GeminiNode_apikey.py
@@ -88,9 +88,10 @@ class PDGeminiImageGenAPIKey:
                 }),
             },
             "optional": {
-                "image_ref": ("IMAGE", {
+                "images": ("IMAGE", {
                     "tooltip": "Reference image(s) for image-to-image generation"
                 }),
+                "files": ("GEMINI_INPUT_FILES", ),
                 "system_prompt": ("STRING", {
                     "multiline": True,
                     "default": GEMINI_IMAGE_SYS_PROMPT,
@@ -107,7 +108,7 @@ class PDGeminiImageGenAPIKey:
     FUNCTION = "generate_image"
     CATEGORY = "PD_Tools/Image_Generation"
 
-    def generate_image(self, api_key, prompt, model, aspect_ratio, response_modalities, seed, unique_id, image_ref=None, system_prompt=""):
+    def generate_image(self, api_key, prompt, model, aspect_ratio, response_modalities, seed, unique_id, images=None, files=None, system_prompt=""):
         """生成图片"""
         
         # 验证 API Key
@@ -129,9 +130,9 @@ class PDGeminiImageGenAPIKey:
         parts.append({"text": prompt})
         
         # 添加参考图片（如果有）
-        if image_ref is not None:
-            for i in range(image_ref.shape[0]):
-                b64_str = tensor_to_base64(image_ref[i])
+        if images is not None:
+            for i in range(images.shape[0]):
+                b64_str = tensor_to_base64(images[i])
                 parts.append({
                     "inlineData": {
                         "mimeType": "image/png",
@@ -140,6 +141,15 @@ class PDGeminiImageGenAPIKey:
                 })
                 print(f"[GEMINI_APIKEY] Added reference image {i+1}")
 
+        # Files 输入处理
+        if files:
+            for file_part in files:
+                if hasattr(file_part, "model_dump"):
+                    parts.append(file_part.model_dump(exclude_none=True))
+                else:
+                    parts.append(file_part)
+                print(f"[GEMINI_APIKEY] Added file part")
+
         # 生成配置
         generation_config = {
             "responseModalities": ["IMAGE"] if response_modalities == "IMAGE" else ["TEXT", "IMAGE"]
@@ -280,5 +290,5 @@ NODE_CLASS_MAPPINGS = {
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PDGeminiImageGenAPIKey": "PD: Gemini Image (apikey)"
+    "PDGeminiImageGenAPIKey": "PD: Gemini Image (ComfyUI Key)"
 }
diff --git a/py/IMGE_GeminiNode_authtoken.py b/py/IMGE_GeminiNode_authtoken.py
index 7a4597c..a6d157f 100644
--- a/py/IMGE_GeminiNode_authtoken.py
+++ b/py/IMGE_GeminiNode_authtoken.py
@@ -87,9 +87,10 @@ class PDGeminiImageGenAuthToken:
                 }),
             },
             "optional": {
-                "image_ref": ("IMAGE", {
+                "images": ("IMAGE", {
                     "tooltip": "Reference image(s) for image-to-image generation"
                 }),
+                "files": ("GEMINI_INPUT_FILES", ),
                 "system_prompt": ("STRING", {
                     "multiline": True,
                     "default": GEMINI_IMAGE_SYS_PROMPT,
@@ -106,7 +107,7 @@ class PDGeminiImageGenAuthToken:
     FUNCTION = "generate_image"
     CATEGORY = "PD_Tools/Image_Generation"
 
-    def generate_image(self, auth_token, prompt, model, aspect_ratio, resolution, seed, unique_id, image_ref=None, system_prompt=""):
+    def generate_image(self, auth_token, prompt, model, aspect_ratio, resolution, seed, unique_id, images=None, files=None, system_prompt=""):
         """生成图片"""
         
         print(f"[GEMINI_AUTHTOKEN] ========== 开始执行 ==========")
@@ -138,9 +139,9 @@ class PDGeminiImageGenAuthToken:
         parts.append({"text": prompt})
         
         # 添加参考图片（如果有）
-        if image_ref is not None:
-            for i in range(image_ref.shape[0]):
-                b64_str = tensor_to_base64(image_ref[i])
+        if images is not None:
+            for i in range(images.shape[0]):
+                b64_str = tensor_to_base64(images[i])
                 parts.append({
                     "inlineData": {
                         "mimeType": "image/png",
@@ -149,6 +150,15 @@ class PDGeminiImageGenAuthToken:
                 })
                 print(f"[GEMINI_AUTHTOKEN] Added reference image {i+1}")
 
+        # Files 输入处理
+        if files:
+            for file_part in files:
+                if hasattr(file_part, "model_dump"):
+                    parts.append(file_part.model_dump(exclude_none=True))
+                else:
+                    parts.append(file_part)
+                print(f"[GEMINI_AUTHTOKEN] Added file part")
+
         # 生成配置
         generation_config = {
             "responseModalities": ["IMAGE"]
@@ -277,5 +287,5 @@ NODE_CLASS_MAPPINGS = {
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PDGeminiImageGenAuthToken": "PD: Gemini Image (comfyui_AuthToken)"
+    "PDGeminiImageGenAuthToken": "PD: Gemini Image (ComfyUI AuthToken)"
 }
diff --git a/py/IMGE_GeminiPro_apikey.py b/py/IMGE_GeminiPro_apikey.py
index 0f4e955..001e786 100644
--- a/py/IMGE_GeminiPro_apikey.py
+++ b/py/IMGE_GeminiPro_apikey.py
@@ -93,6 +93,7 @@ class PDGeminiProImageGenAPIKey:
                 "images": ("IMAGE", {
                     "tooltip": "Optional reference image(s) (up to 14)"
                 }),
+                "files": ("GEMINI_INPUT_FILES", ),
                 "system_prompt": ("STRING", {
                     "multiline": True,
                     "default": GEMINI_IMAGE_SYS_PROMPT,
@@ -109,7 +110,7 @@ class PDGeminiProImageGenAPIKey:
     FUNCTION = "generate_image"
     CATEGORY = "PD_Tools/Image_Generation"
 
-    def generate_image(self, api_key, prompt, model, aspect_ratio, resolution, response_modalities, seed, unique_id, images=None, system_prompt=""):
+    def generate_image(self, api_key, prompt, model, aspect_ratio, resolution, response_modalities, seed, unique_id, images=None, files=None, system_prompt=""):
         """生成图片"""
         
         # 验证 API Key
@@ -146,6 +147,15 @@ class PDGeminiProImageGenAPIKey:
             if images.shape[0] > 14:
                 print(f"[GEMINI_PRO_APIKEY] Warning: Only first 14 images will be used (provided {images.shape[0]})")
 
+        # Files 输入处理
+        if files:
+            for file_part in files:
+                if hasattr(file_part, "model_dump"):
+                    parts.append(file_part.model_dump(exclude_none=True))
+                else:
+                    parts.append(file_part)
+                print(f"[GEMINI_PRO_APIKEY] Added file part")
+
         # 生成配置
         generation_config = {
             "responseModalities": ["IMAGE"] if response_modalities == "IMAGE" else ["TEXT", "IMAGE"]
@@ -290,5 +300,5 @@ NODE_CLASS_MAPPINGS = {
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PDGeminiProImageGenAPIKey": "PD: Gemini Pro Image (apikey)"
+    "PDGeminiProImageGenAPIKey": "PD: Gemini Pro Image (ComfyUI Key)"
 }
diff --git a/py/IMGE_GeminiPro_authtoken.py b/py/IMGE_GeminiPro_authtoken.py
index 2a606f2..26f240c 100644
--- a/py/IMGE_GeminiPro_authtoken.py
+++ b/py/IMGE_GeminiPro_authtoken.py
@@ -93,6 +93,7 @@ class PDGeminiProImageGenAuthToken:
                 "images": ("IMAGE", {
                     "tooltip": "Optional reference image(s) (up to 14)"
                 }),
+                "files": ("GEMINI_INPUT_FILES", ),
                 "system_prompt": ("STRING", {
                     "multiline": True,
                     "default": GEMINI_IMAGE_SYS_PROMPT,
@@ -109,7 +110,7 @@ class PDGeminiProImageGenAuthToken:
     FUNCTION = "generate_image"
     CATEGORY = "PD_Tools/Image_Generation"
 
-    def generate_image(self, auth_token, prompt, model, aspect_ratio, resolution, response_modalities, seed, unique_id, images=None, system_prompt=""):
+    def generate_image(self, auth_token, prompt, model, aspect_ratio, resolution, response_modalities, seed, unique_id, images=None, files=None, system_prompt=""):
         """生成图片"""
         
         print(f"[GEMINI_PRO_AUTHTOKEN] ========== 开始执行 ==========")
@@ -152,6 +153,15 @@ class PDGeminiProImageGenAuthToken:
             if images.shape[0] > 14:
                 print(f"[GEMINI_PRO_AUTHTOKEN] Warning: Only first 14 images will be used (provided {images.shape[0]})")
 
+        # Files 输入处理
+        if files:
+            for file_part in files:
+                if hasattr(file_part, "model_dump"):
+                    parts.append(file_part.model_dump(exclude_none=True))
+                else:
+                    parts.append(file_part)
+                print(f"[GEMINI_PRO_AUTHTOKEN] Added file part")
+
         # 生成配置
         generation_config = {
             "responseModalities": ["IMAGE"] if response_modalities == "IMAGE" else ["TEXT", "IMAGE"]
@@ -296,5 +306,5 @@ NODE_CLASS_MAPPINGS = {
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "PDGeminiProImageGenAuthToken": "PD: Gemini Pro Image (comfyui_AuthToken)"
+    "PDGeminiProImageGenAuthToken": "PD: Gemini Pro Image (ComfyUI AuthToken)"
 }