main_diffsynth.py

# 主入口
# main_diffsynth.py
# !/usr/bin/env python3
"""
Auto-DiffSynth-Studio 主入口
基于aacode和DiffSynth-Studio的AI图片和视频创作助手
工作目录固定为 my_diffsynth_studio
"""

import argparse
import asyncio
import os
import subprocess
import sys
import time
from pathlib import Path
from typing import Optional, Dict, Any, List
if __package__ in (None, ""):
    sys.path.insert(0, str(Path(__file__).parent.parent))
    from aacode.core.main_agent import MainAgent
    from aacode.utils.context_manager import ContextManager
    from aacode.utils.safety import SafetyGuard
    from aacode.config import settings
else:
    from .core.main_agent import MainAgent
    from .utils.context_manager import ContextManager
    from .utils.safety import SafetyGuard
    from .config import settings

WORK_DIR_NAME = "my_diffsynth_studio"


class AICoder:
    """Auto-DiffSynth-Studio AI创作助手主类"""

    def __init__(self, model_config: Optional[Dict] = None):
        """
        初始化Auto-DiffSynth-Studio

        Args:
            model_config: 模型配置
        """
        project_path = Path(__file__).parent / WORK_DIR_NAME
        project_path = project_path.absolute()

        self.project_path = project_path
        self.target_project = project_path

        print(f"🎨 Auto-DiffSynth-Studio 工作目录: {self.project_path}")

        # 检查并创建项目目录，处理权限问题
        try:
            self.project_path.mkdir(parents=True, exist_ok=True)
        except PermissionError as e:
            print(f"❌ 权限错误: 无法创建项目目录 '{self.project_path}'")
            print(f"   错误信息: {e}")
            print(f"   请检查目录权限或使用有写入权限的目录")
            raise

        # 检查目录写入权限
        test_file = self.project_path / ".permission_test"
        try:
            test_file.touch()
            test_file.unlink()
        except PermissionError as e:
            print(f"❌ 权限错误: 对目录 '{self.project_path}' 没有写入权限")
            print(f"   错误信息: {e}")
            print(f"   请使用 'chmod' 命令修改目录权限或选择其他目录")
            raise

        # 不使用chdir，避免路径混乱问题
        # 所有工具都会使用target_project作为基准路径

        # 初始化核心组件（使用target_project作为安全护栏的基准）
        self.safety_guard = SafetyGuard(
            self.target_project,
            dangerous_command_action=settings.safety.dangerous_command_action,
        )
        # 上下文管理器使用aacode工作目录（存放日志等）
        self.context_manager = ContextManager(self.project_path)
        # 主Agent使用目标项目目录（实际操作目录）
        self.main_agent = MainAgent(
            project_path=self.target_project,
            context_manager=self.context_manager,
            safety_guard=self.safety_guard,
            model_config=model_config or settings.DEFAULT_MODEL,
        )

        # 加载项目初始化指令
        self._load_init_instructions()

        # 初始化类方法映射器
        self._init_class_method_mapper()

    def _load_init_instructions(self):
        """加载项目初始化指令"""
        # 优先从目标项目加载init.md，如果不存在则从工作目录加载
        init_file = self.target_project / "init.md"
        if not init_file.exists():
            init_file = self.project_path / "init.md"
        if not init_file.exists():
            # 创建默认指令
            default_init = """# 项目指导原则

## 核心规则
1. 每个代码文件顶部必须用注释标注路径：`# {relative_path}`
2. 优先修改现有文件而非创建新文件
3. 所有文件操作必须在项目目录内
4. 危险命令需要用户确认

## 工作流程
1. 先分析需求，制定计划
2. 小步快跑，频繁测试
3. 编写自包含的测试函数
4. 使用工具前检查安全性

## 代码质量
- 遵循PEP 8/Python最佳实践
- 函数尽量不超过60行
- 添加必要的文档字符串
- 错误处理要优雅"""
            init_file.write_text(default_init)

        self.init_instructions = init_file.read_text()

    def _init_class_method_mapper(self):
        """初始化类方法映射器"""
        try:
            # 尝试使用增强版映射器
            if __package__ in (None, ""):
                from utils.class_method_mapper import EnhancedClassMethodMapper
            else:
                from .utils.class_method_mapper import EnhancedClassMethodMapper

            self.class_method_mapper = EnhancedClassMethodMapper(self.target_project)
            print("✅ 增强版类方法映射器初始化成功（支持多语言）")
        except ImportError as e:
            print(f"⚠️  无法导入增强版类方法映射器: {e}")
            try:
                # 回退到基础版映射器
                if __package__ in (None, ""):
                    from utils.class_method_mapper import ClassMethodMapper
                else:
                    from .utils.class_method_mapper import ClassMethodMapper

                self.class_method_mapper = ClassMethodMapper(self.target_project)
                print("✅ 基础版类方法映射器初始化成功（仅Python）")
            except ImportError as e2:
                print(f"⚠️  无法导入类方法映射器: {e2}")
                self.class_method_mapper = None

        # 初始化to-do-list管理器（使用aacode工作目录）
        if __package__ in (None, ""):
            from utils.todo_manager import get_todo_manager
        else:
            from .utils.todo_manager import get_todo_manager

        self.todo_manager = get_todo_manager(self.project_path)

    def analyze_project_structure(self) -> str:
        """分析项目结构并生成类方法映射"""
        if not self.class_method_mapper:
            return "类方法映射器未初始化"

        try:
            print("🔍 开始分析项目结构...")
            summary = self.class_method_mapper.analyze_project()

            # 尝试使用增强版方法
            try:
                # 增强版映射器
                map_file = self.class_method_mapper.save_enhanced_map()
                # 获取语言摘要用于提示
                language_summary = self.class_method_mapper.get_language_summary()

                print(f"✅ 项目结构分析完成:")
                if "multi_lang_analysis" in summary:
                    lang_stats = summary["multi_lang_analysis"]["languages"]
                    for lang, stats in lang_stats.items():
                        print(
                            f"   - {lang}: {stats['file_count']} 个文件, {stats['total_lines']} 行"
                        )
                print(f"   - 结构文件: {map_file.name}")

                # 返回语言摘要和映射内容
                map_content = map_file.read_text(encoding="utf-8")
                return f"{language_summary}\n\n完整结构见文件: {map_file.name}"

            except AttributeError:
                # 回退到基础版方法
                map_file = self.class_method_mapper.save_class_method_map()
                map_content = map_file.read_text(encoding="utf-8")

                print(f"✅ Python项目结构分析完成:")
                print(f"   - 类数量: {summary.get('class_count', 0)}")
                print(f"   - 函数数量: {summary.get('function_count', 0)}")
                print(f"   - 文件数量: {summary.get('file_count', 0)}")
                print(f"   - 映射文件: {map_file.name}")

                return map_content[:2000]  # 返回前2000字符作为摘要

        except Exception as e:
            error_msg = f"项目结构分析失败: {e}"
            print(f"❌ {error_msg}")
            return error_msg

    def update_class_method_map(
        self, changed_files: Optional[List[Path]] = None
    ) -> str:
        """更新类方法映射"""
        if not self.class_method_mapper:
            return "类方法映射器未初始化"

        try:
            print("🔄 更新类方法映射...")
            # 处理None情况
            files_to_update = changed_files if changed_files is not None else []

            # 将字符串转换为Path对象（相对路径转换为绝对路径）
            files_to_update = [
                (self.project_path / f) if isinstance(f, str) else f
                for f in files_to_update
            ]

            # 尝试使用增强版方法
            try:
                success = self.class_method_mapper.update_analysis(files_to_update)
                map_file = self.project_path / "project_structure.md"
            except AttributeError:
                # 回退到基础版方法
                success = self.class_method_mapper.update_class_method_map(
                    files_to_update
                )
                map_file = self.project_path / "class_method_map.md"

            if success:
                if map_file.exists():
                    map_content = map_file.read_text(encoding="utf-8")
                    return f"类方法映射已更新\n\n{map_content[:1000]}..."
                else:
                    return "类方法映射更新失败：文件未生成"
            else:
                return "类方法映射更新失败"

        except Exception as e:
            error_msg = f"更新类方法映射失败: {e}"
            print(f"❌ {error_msg}")
            return error_msg

    async def run(self, task: str, max_iterations: int = 30) -> Dict[str, Any]:
        """
        执行任务

        Args:
            task: 任务描述
            max_iterations: 最大迭代次数

        Returns:
            执行结果
        """
        print(f"\n🎯 开始任务: {task}")
        print(f"📁 aacode工作目录: {self.project_path}")
        print(f"🎯 目标项目目录: {self.target_project}")
        print(f"📝 初始化指令已加载 ({len(self.init_instructions.split())} 字)")

        # 跳过项目结构分析（DiffSynth-Studio文件过多，会造成上下文膨胀）
        print("\n⏭️ 跳过项目结构分析")
        # analysis_result = self.analyze_project_structure()
        analysis_result = "项目结构分析已跳过，请参考init.md中的资源路径速查"

        # 创建任务目录
        task_dir = (
            self.project_path
            / ".aacode"
            / f"task_{int(asyncio.get_event_loop().time())}"
        )
        task_dir.mkdir(parents=True, exist_ok=True)

        # 创建to-do-list，并同步到上下文管理器
        print("\n📋 创建任务待办清单...")
        todo_file = await self.todo_manager.create_todo_list(
            task, context_manager=self.context_manager
        )
        print(f"✅ 待办清单已创建: {todo_file}")

        try:
            # 运行主Agent，传递类方法映射信息
            result = await self.main_agent.execute(
                task=task,
                init_instructions=self.init_instructions,
                task_dir=task_dir,
                max_iterations=max_iterations,
                project_analysis=analysis_result,
                todo_manager=self.todo_manager,  # 传递todo管理器
            )
            return result
        except asyncio.CancelledError:
            # 异步任务被取消
            print("\n⏹️ 任务被取消")
            return {
                "status": "cancelled",
                "error": "任务被用户取消",
                "iterations": 0,
                "execution_time": 0,
                "session_id": f"cancelled_{int(asyncio.get_event_loop().time())}",
            }
        except Exception as e:
            # 捕获并处理异常，避免程序崩溃
            print(f"\n❌ 任务执行失败: {e}")
            import traceback

            traceback.print_exc()
            # 返回一个包含错误信息的结果，而不是抛出异常
            return {
                "status": "error",
                "error": str(e),
                "iterations": 0,
                "execution_time": 0,
                "session_id": f"error_{int(asyncio.get_event_loop().time())}",
            }
        finally:
            # 确保资源被清理
            try:
                # 清理web_tools资源
                if hasattr(self.main_agent, "web_tools"):
                    await self.main_agent.web_tools.cleanup()
            except Exception as e:
                print(f"⚠️  清理资源时出错: {e}")


async def continue_session(coder, project_dir):
    """继续会话，执行追加任务"""
    print("\n" + "=" * 50)
    print("🔁 继续会话模式")
    print("=" * 50)

    # 检查是否有待办清单
    todo_dir = project_dir / ".aacode" / "todos"
    if todo_dir.exists():
        todo_files = list(todo_dir.glob("*.md"))
        if todo_files:
            print(f"\n📋 发现 {len(todo_files)} 个待办清单:")
            for i, todo_file in enumerate(todo_files[-3:], 1):  # 显示最近3个
                print(f"  {i}. {todo_file.name}")
            print("💡 输入 'todo' 查看待办清单详情")

    # 检查是否有会话日志
    log_dir = project_dir / ".aacode" / "logs"
    if log_dir.exists():
        log_files = list(log_dir.glob("*.log"))
        if log_files:
            print(f"📝 发现 {len(log_files)} 个会话日志")

    while True:
        try:
            print("\n当前项目目录:", project_dir)
            print("可用命令:")
            print("  - 输入任务描述继续工作")
            print("  - 输入 'list' 查看项目文件")
            print("  - 输入 'todo' 查看待办清单")
            print("  - 输入 'logs' 查看会话日志")
            print("  - 输入 'exit' 或 'quit' 退出")
            print("  - 输入 'clear' 清空项目目录")
            print("  - 输入 '继续' 查看恢复任务说明")
            print("  - 输入 'help' 查看帮助")

            user_input = input("\n> ").strip()

            if user_input.lower() in ["exit", "quit", "q"]:
                print("👋 退出会话")
                break
            elif user_input.lower() == "list":
                # 列出项目文件
                print("\n📁 项目文件:")
                files = list(project_dir.glob("*"))
                if not files:
                    print("  (空目录)")
                else:
                    for file in files:
                        if file.is_file():
                            size = file.stat().st_size
                            print(f"  - {file.name} ({size} bytes)")
                continue
            elif user_input.lower() == "todo":
                # 查看待办清单
                if todo_dir.exists():
                    todo_files = list(todo_dir.glob("*.md"))
                    if todo_files:
                        print("\n📋 待办清单:")
                        for i, todo_file in enumerate(todo_files, 1):
                            with open(todo_file, "r", encoding="utf-8") as f:
                                first_line = f.readline().strip()
                            print(f"  {i}. {todo_file.name}")
                            print(f"     内容: {first_line[:80]}...")
                        print("\n💡 输入待办清单编号查看详情，或输入 'back' 返回")
                        choice = input("选择待办清单 (编号/back): ").strip()
                        if choice.lower() != "back" and choice.isdigit():
                            idx = int(choice) - 1
                            if 0 <= idx < len(todo_files):
                                with open(todo_files[idx], "r", encoding="utf-8") as f:
                                    print(f"\n📄 {todo_files[idx].name}:")
                                    print(f.read())
                    else:
                        print("📭 没有待办清单")
                else:
                    print("📭 待办目录不存在")
                continue
            elif user_input.lower() == "logs":
                # 查看会话日志
                if log_dir.exists():
                    log_files = list(log_dir.glob("*.log"))
                    if log_files:
                        print("\n📝 会话日志:")
                        for i, log_file in enumerate(log_files[-5:], 1):  # 显示最近5个
                            size = log_file.stat().st_size
                            print(f"  {i}. {log_file.name} ({size} bytes)")
                        print("\n💡 输入日志编号查看最后几行，或输入 'back' 返回")
                        choice = input("选择日志 (编号/back): ").strip()
                        if choice.lower() != "back" and choice.isdigit():
                            idx = int(choice) - 1
                            if 0 <= idx < len(log_files):
                                with open(log_files[idx], "r", encoding="utf-8") as f:
                                    lines = f.readlines()
                                    print(f"\n📄 {log_files[idx].name} (最后20行):")
                                    for line in lines[-20:]:
                                        print(line.rstrip())
                    else:
                        print("📭 没有会话日志")
                else:
                    print("📭 日志目录不存在")
                continue
            elif user_input.lower() == "clear":
                # 确认清空项目
                confirm = (
                    input("⚠️  确认清空项目目录? (输入 'yes' 确认): ").strip().lower()
                )
                if confirm == "yes":
                    for file in project_dir.glob("*"):
                        if file.is_file() and file.name != ".env":
                            file.unlink()
                    print("✅ 项目目录已清空")
                continue
            elif user_input.lower() in ["继续", "continue"]:
                # 处理"继续"命令
                print("\n" + "=" * 50)
                print("🔄 恢复任务说明")
                print("=" * 50)
                print("\n要恢复之前中断的任务，有以下几种方式:")
                print("\n1. 🎯 输入具体的任务描述")
                print("   例如: '完成用户注册功能'")
                print("   系统会自动参考之前的待办清单继续工作")
                print("\n2. 🔄 使用会话ID恢复")
                print("   重新运行程序时使用: --session <session_id>")
                print("   会话ID会在任务开始时显示")
                print(
                    "   例如: python main_diffsynth.py --session session_20260212_123548_3"
                )
                print("\n3. 📋 基于待办清单继续")
                print("   输入 'todo' 查看现有待办清单")
                print("   选择待办清单后，输入相关任务描述")
                print("\n4. 🔍 查看项目状态")
                print("   输入 'list' 查看项目文件")
                print("   输入 'logs' 查看会话日志")
                print("\n💡 建议: 输入具体的任务描述是最直接的方式")
                continue
            elif user_input.lower() == "help":
                print("\n" + "=" * 50)
                print("📚 帮助文档")
                print("=" * 50)
                print("\n🔧 常用命令:")
                print("  list    - 查看项目文件")
                print("  todo    - 查看待办清单")
                print("  logs    - 查看会话日志")
                print("  clear   - 清空项目目录")
                print("  exit    - 退出会话")
                print("  help    - 显示帮助")
                print("\n🎯 任务执行:")
                print("  直接输入任务描述即可开始工作")
                print("  例如: '添加用户登录功能'")
                print("  系统会自动分析项目并制定计划")
                print("\n🔄 恢复任务:")
                print("  输入 '继续' 查看恢复任务说明")
                print("  或直接输入任务描述继续工作")
                print("\n⚠️  注意事项:")
                print("  1. 确保API密钥已正确设置")
                print("  2. 大型项目可能需要较长时间")
                print("  3. 可以使用Ctrl+C中断当前任务")
                print("  4. 中断后输入 'y' 可以继续会话")
                continue
            elif user_input:
                # 检查是否是"继续任务"或类似命令
                if user_input.lower() in ["继续任务", "继续之前的任务", "恢复任务"]:
                    # 尝试恢复最近的任务
                    print(f"\n🔄 尝试恢复最近的任务...")

                    # 检查待办清单目录
                    todo_dir = project_dir / ".aacode" / "todos"
                    if todo_dir.exists():
                        todo_files = list(todo_dir.glob("*.md"))
                        if todo_files:
                            # 获取最新的待办清单
                            latest_todo = max(
                                todo_files, key=lambda f: f.stat().st_mtime
                            )
                            print(f"📋 找到待办清单: {latest_todo.name}")

                            # 读取待办清单内容
                            with open(latest_todo, "r", encoding="utf-8") as f:
                                todo_content = f.read()

                            # 提取任务描述
                            import re

                            task_match = re.search(r"\*\*任务\*\*: (.+)", todo_content)
                            if task_match:
                                original_task = task_match.group(1)
                                print(f"🎯 原始任务: {original_task}")

                                # 询问用户是否继续这个任务
                                confirm = (
                                    input(f"是否继续这个任务? (y/n): ").strip().lower()
                                )
                                if confirm == "y":
                                    user_input = original_task
                                    print(f"🔄 继续任务: {original_task}")
                                else:
                                    print("请输入新的任务描述:")
                                    user_input = input("> ").strip()
                            else:
                                print("❌ 无法从待办清单中提取任务描述")
                                print("请输入任务描述:")
                                user_input = input("> ").strip()
                        else:
                            print("📭 没有找到待办清单")
                            print("请输入任务描述:")
                            user_input = input("> ").strip()
                    else:
                        print("📭 待办目录不存在")
                        print("请输入任务描述:")
                        user_input = input("> ").strip()

                # 执行任务
                print(f"\n🎯 开始执行任务: {user_input}")
                print("正在准备...")

                try:
                    result = await coder.run(user_input)

                    # 检查任务是否成功
                    if result.get("status") == "error":
                        print(f"\n❌ 任务执行失败: {result.get('error', '未知错误')}")
                        print("💡 建议: 检查错误信息，或输入新任务重试")
                    else:
                        print(f"\n✅ 任务完成!")
                        print(f"迭代次数: {result.get('iterations', 0)}")
                        print(f"执行时间: {result.get('execution_time', 0):.2f}秒")

                        # 显示会话ID以便后续恢复
                        session_id = result.get("session_id")
                        if session_id and not session_id.startswith("error_"):
                            print(f"会话ID: {session_id}")
                            print(f"使用 --session {session_id} 可以继续此会话")
                except Exception as e:
                    print(f"\n❌ 任务执行失败: {e}")
                    print("💡 建议: 检查错误信息，或输入新任务重试")

                continue
            else:
                print("❌ 请输入有效命令")

        except KeyboardInterrupt:
            print("\n\n⏸️  会话中断")
            print("输入 'y' 继续当前会话，输入 'n' 退出程序")
            choice = input("继续? (y/n): ").strip().lower()
            if choice == "y":
                continue
            else:
                print("👋 退出程序")
                break
        except Exception as e:
            print(f"\n❌ 执行出错: {e}")
            import traceback

            traceback.print_exc()
            print("\n💡 建议: 检查错误信息，或输入新任务重试")
            print("   输入 'help' 查看帮助文档")


async def main():
    """命令行主函数"""
    import time as time_module

    project_path = Path(__file__).parent / WORK_DIR_NAME

    parser = argparse.ArgumentParser(
        description="🎨 Auto-DiffSynth-Studio - AI图片视频创作助手",
        formatter_class=argparse.RawTextHelpFormatter,
    )
    parser.add_argument(
        "task",
        nargs="*",
        help='任务描述，例如 "生成一张科技感的城市夜景图片"',
    )
    parser.add_argument("--continue", action="store_true", help="继续上一个会话")
    parser.add_argument("--session", help="指定会话ID继续任务")
    parser.add_argument(
        "--plan-first", action="store_true", help="先规划再执行模式（复杂任务推荐）"
    )
    parser.add_argument("--interactive", action="store_true", help="交互式连续对话模式")

    args = parser.parse_args()

    # 加载环境变量配置
    env_file = Path(".env")
    if env_file.exists():
        with open(env_file, "r", encoding="utf-8") as f:
            for line in f:
                line = line.strip()
                if line and "=" in line and not line.startswith("#"):
                    key, value = line.split("=", 1)
                    os.environ[key.strip()] = value.strip()

    # 基础任务
    base_task = "请遵循init.md完成用户任务。"
    # 解析任务
    task_parts = args.task

    # 如果命令行没给任务，就交互式询问
    task = " ".join(task_parts).strip()
    if not task:
        task = input("请输入任务: ").strip()

    task = f"{base_task}\n用户任务：{task}"

    # 创建Auto-DiffSynth-Studio实例（工作目录固定为my_diffsynth_studio）
    coder = AICoder()

    # 获取工作目录
    work_dir = coder.project_path

    # 运行任务
    try:
        result = await coder.run(task)

        print(f"\n✅ 任务完成!")
        print(f"📋 会话ID: {result.get('session_id', 'N/A')}")
        print(f"迭代次数: {result.get('iterations', 0)}")
        print(f"最终状态: {result.get('status', 'unknown')}")
        print(f"执行时间: {result.get('execution_time', 0):.2f}秒")

        # 检查生成的文件
        project_dir = work_dir
        created_files = []

        # 递归查找项目目录下的Python文件（排除.aacode和虚拟环境）
        exclude_dirs = {
            ".aacode",
            ".venv",
            "venv",
            "__pycache__",
            ".git",
            "node_modules",
            "DiffSynth-Studio"
        }

        for py_file in project_dir.rglob("*.py"):
            # 排除特定目录和主程序
            if py_file.name not in ["main.py", "main_diffsynth.py"] and not any(
                excluded in py_file.parts for excluded in exclude_dirs
            ):
                # 检查文件是否是最近创建的（5分钟内）
                import time

                if time.time() - py_file.stat().st_mtime < 300:  # 5分钟 = 300秒
                    created_files.append(py_file)
        """
        # 这个后续动作有点生硬，先注释
        if created_files:
            print("\n📁 生成的文件:")
            for file in created_files:
                # 显示相对路径
                rel_path = file.relative_to(project_dir)
                print(f"  - {rel_path}")
                
                # 询问是否运行Python文件
                try:
                    response = input(f"是否运行 {rel_path}? (y/n): ").strip().lower()
                    if response == 'y':
                        print(f"🚀 运行 {rel_path}:")
                        
                        # 方法1: 尝试使用项目根目录作为工作目录
                        result = subprocess.run([sys.executable, str(rel_path)], 
                                              cwd=project_dir, 
                                              capture_output=True, text=True,
                                              env={**os.environ, 'PYTHONPATH': str(project_dir)})
                        
                        if result.returncode == 0:
                            print(f"✅ 输出: {result.stdout.strip()}")
                        else:
                            error_msg = result.stderr.strip()
                            print(f"❌ 错误: {error_msg}")
                            
                            # 提供更详细的错误信息和解决建议
                            if error_msg:
                                print(f"\n💡 错误分析:")
                                if "No module named" in error_msg:
                                    print(f"   - 导入错误: 缺少依赖模块")
                                    print(f"   - 建议: 检查是否需要安装依赖包")
                                    print(f"   - 或者: 文件可能需要在特定目录运行")
                                elif "FileNotFoundError" in error_msg or "No such file" in error_msg:
                                    print(f"   - 文件路径错误")
                                    print(f"   - 建议: 检查文件中使用的相对路径")
                                else:
                                    print(f"   - 建议: 检查代码逻辑和语法")
                except (KeyboardInterrupt, EOFError):
                    print("\n跳过运行")
                    break
        """

        # 任务完成后，默认进入继续会话模式
        print("\n" + "=" * 50)
        print("✅ 任务已完成！")
        print("=" * 50)

        # 询问是否继续执行追加任务
        try:
            response = (
                input("\n🔁 是否继续执行其他任务? (y/n，默认y): ").strip().lower()
            )
            if not response or response == "y":
                await continue_session(coder, project_dir)
        except (KeyboardInterrupt, EOFError):
            print("\n👋 退出程序")

    except KeyboardInterrupt:
        print("\n⏹️ 用户中断")
        # 即使中断，也询问是否继续
        try:
            response = (
                input("\n🔁 任务被中断，是否继续执行其他任务? (y/n): ").strip().lower()
            )
            if response == "y":
                project_dir = work_dir
                await continue_session(coder, project_dir)
        except:
            pass
    except Exception as e:
        print(f"\n❌ 错误: {e}")
        import traceback

        traceback.print_exc()
        # 错误后也询问是否继续
        try:
            response = (
                input("\n🔁 发生错误，是否继续执行其他任务? (y/n): ").strip().lower()
            )
            if response == "y":
                project_dir = work_dir
                await continue_session(coder, project_dir)
        except:
            pass


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n\n⏹️ 程序被用户中断，正常退出")
        sys.exit(0)