# coding=utf-8
import uiautomator2 as u2
import time
import logging
import sys
import os
import asyncio

# 添加项目根目录到 sys.path 以便导入 Util
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if project_root not in sys.path:
    sys.path.append(project_root)

from WeiXin.WxUtil import find_input_box_center, perform_input_action, analyze_chat_image, clean_screenshots_dir
from Util.LlmUtil import get_llm_response

# 配置日志
log_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Logs")
if not os.path.exists(log_dir):
    os.makedirs(log_dir)

logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler(os.path.join(log_dir, "T4_InputLlmText.log"), encoding='utf-8'),
        logging.StreamHandler()
    ]
)
logger = logging.getLogger("T4_InputLlmText")

async def generate_and_input():
    logger.info("开始执行 T4: 生成 LLM 回复并输入...")
    
    # 0. 清除旧截图
    clean_screenshots_dir()
    
    try:
        # 1. 连接设备
        d = u2.connect()
        logger.info(f"设备连接成功: {d.info.get('serial')}")
        
        # 2. 截图并识别对话历史
        screenshot_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Screenshots")
        if not os.path.exists(screenshot_dir):
            os.makedirs(screenshot_dir)
        
        tmp_shot = os.path.join(screenshot_dir, "t4_temp_history_check.jpg")
        d.screenshot(tmp_shot)
        
        analyzed_shot = os.path.join(screenshot_dir, "t4_temp_history_analyzed.jpg")
        dialogue_log = analyze_chat_image(tmp_shot, analyzed_shot)
        
        history_text = ""
        if dialogue_log:
            history_text = "\n".join(dialogue_log)
            logger.info(f"提取到对话历史:\n{history_text}")
        else:
            logger.warning("未提取到对话历史")
            history_text = "（无对话历史）"

        # 3. 调用 LLM 生成回复
        persona = (
            "你是一名99年毕业，有27年教学经验的小学高级教师，大张老师。你在长春市少惠林作文素养培养中心工作，"
            "校区真实地址就在：长春市南环城路与临河街交汇的TOUCH12街3楼325号。你对作文批阅不下20000个，"
            "非常熟悉小学阶段、初二以前的初中阶段作文教学。你是一位深受家长和学生信任的教育智者，"
            "回复时要既专业又充满亲和力，像老朋友交流一样自然、温暖。严禁使用‘1. 2. 3.’或‘首先、其次、最后’等生硬、刻板的列条目回复。"
            "要善于用富有经验的话语疏导家长的焦虑，并给出有深度、可操作的详细指导。语言要生动、形象，多用比喻或实例，展现岁月沉淀的智慧。"
            "如果涉及到校区信息，必须使用以下真实数据，绝对严禁使用'XX区'、'XX路'、'XX号'等占位符：\n"
            "1. 校址：长春市南环城路与临河街交汇的TOUCH12街3楼325号\n"
            "2. 课程：线上和线下作文课、假期读书营、阅读策略营\n"
            "3. 上课时间：周二/周四晚上，以及周六周日全天\n"
            "4. 联系人：小张老师\n"
            "5. 电话：18686619970"
        )
        prompt = (
            f"人设：{persona}\n\n"
            f"以下是微信聊天记录（'对方'和'我'的对话）：\n{history_text}\n\n"
            "请作为大张老师，针对最后的消息给出一个充满关怀、有深度且专业的回复。回复要亲切自然，"
            "字数控制在 150-250 字之间，不要分点列项，而是用连贯的段落叙述。要把方法讲透，把道理讲细，"
            "让家长感受到你作为老教师的温度和沉淀。如果提到地址、电话或课程，必须使用人设中提供的真实信息，"
            "严禁出现任何 'XX' 形式的占位符。直接输出回复内容。"
        )
        logger.info(f"正在以'亲切而专业的特级教师'身份请求 LLM 生成深度回复...")
        
        full_response = ""
        async for chunk in get_llm_response(prompt, stream=False):
            full_response += chunk
            
        llm_text = full_response.strip().strip('"').strip('“').strip('”')
        logger.info(f"LLM 生成的回复内容: {llm_text}")
        
        if not llm_text:
            logger.error("LLM 生成内容为空，停止执行。")
            return

        # 4. 识别输入框位置
        center_point, rect_box = find_input_box_center(tmp_shot)
        
        if center_point:
            cx, cy = center_point
            logger.info(f"识别到输入框中心: ({cx}, {cy})")
            
            # 5. 点击输入框并输入文字并发送
            # 使用 WxUtil 中的 perform_input_action
            # 该函数内部会先点击坐标，然后输入内容
            # 设置 auto_send=True，输入文字后点击发送
            logger.info(f"正在点击输入框并输入文字并发送: {llm_text}")
            success = perform_input_action(d, center_point, llm_text, auto_send=True)
            
            if success:
                logger.info("✅ T4 执行完成：文字已成功输入并点击发送。")
            else:
                logger.error("❌ T4 执行失败：输入动作未成功完成。")
        else:
            logger.error("❌ T4 执行失败：未能识别到输入框位置。")
            
    except Exception as e:
        logger.error(f"❌ T4 执行出错: {e}", exc_info=True)

if __name__ == "__main__":
    asyncio.run(generate_and_input())