# -*- coding: utf-8 -*-
"""
CLI 入口 - 数据分析智能体
"""

import glob
import os
import sys
from datetime import datetime

from data_analysis_agent import DataAnalysisAgent
from config.llm_config import LLMConfig
from utils.create_session_dir import create_session_output_dir
from utils.logger import PrintCapture


def main():
    llm_config = LLMConfig()

    # 自动查找数据文件
    data_extensions = ["*.csv", "*.xlsx", "*.xls"]
    search_dirs = ["cleaned_data"]
    files = []

    for search_dir in search_dirs:
        for ext in data_extensions:
            pattern = os.path.join(search_dir, ext)
            files.extend(glob.glob(pattern))

    if not files:
        print("[WARN] 未在 cleaned_data 目录找到数据文件，尝试使用默认文件")
        files = ["./cleaned_data.csv"]
    else:
        print(f"[DIR] 自动识别到以下数据文件: {files}")

    analysis_requirement = """
基于所有运维工单，整理一份工单健康度报告，包括但不限于对所有车联网技术支持工单的全面数据分析，
深入挖掘工单处理过程中的关键问题、效率瓶颈及改进机会。请从车型，模块，功能角度，分别展示工单数据、问题类型、模块分布、严重程度、责任人负载、车型分布、来源渠道及处理时长等多个维度。
通过多轮交叉分析与趋势洞察，为提升车联网服务质量、优化资源配置及降低运营风险提供数据驱动的决策依据，问题总揽，高频问题、重点问题分析，输出若干个重要的统计指标，并绘制相关图表；
结合图表，总结一份，车联网运维工单健康度报告，汇报给我。
    """

    # 创建会话目录
    base_output_dir = "outputs"
    session_output_dir = create_session_output_dir(base_output_dir, analysis_requirement)

    # 使用 PrintCapture 替代全局 stdout 劫持
    log_path = os.path.join(session_output_dir, "log.txt")

    with PrintCapture(log_path):
        print(f"\n{'='*20} Run Started at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} {'='*20}\n")
        print(f"[DOC] 日志文件已保存至: {log_path}")

        agent = DataAnalysisAgent(llm_config, force_max_rounds=False)

        # 交互式分析循环
        while True:
            is_first_run = agent.current_round == 0 and not agent.conversation_history

            report = agent.analyze(
                user_input=analysis_requirement,
                files=files if is_first_run else None,
                session_output_dir=session_output_dir,
                reset_session=is_first_run,
                max_rounds=None if is_first_run else 10,
            )
            print("\n" + "=" * 30 + " 当前阶段分析完成 " + "=" * 30)

            print("\n[TIP] 你可以继续对数据提出分析需求，或者输入 'exit'/'quit' 结束程序。")
            user_response = input("[>] 请输入后续分析需求 (直接回车退出): ").strip()

            if not user_response or user_response.lower() in ["exit", "quit", "n", "no"]:
                print("[BYE] 分析结束，再见！")
                break

            analysis_requirement = user_response
            print(f"\n[LOOP] 收到新需求，正在继续分析...")


if __name__ == "__main__":
    main()