import asyncio
from typing import AsyncGenerator

class FastSlowVoiceOrchestrator:
    def __init__(self):
        self.is_interrupted = False

    async def fast_filler_system(self) -> AsyncGenerator[str, None]:
        """快系统：收到复杂指令后，在 200ms 内先给出自然口语响应垫字"""
        fillers = ["收到，正在为您调取数据，请稍候...", "正在核对系统配置..."]
        for f in fillers:
            if self.is_interrupted:
                break
            yield f
            await asyncio.sleep(0.8)

    async def slow_reasoning_system(self, user_intent: str) -> str:
        """慢系统：执行多步深度推理与外部 API 调用"""
        print(f"[慢系统] 启动深度业务工具链: {user_intent}")
        # 模拟外部数据库长时查询
        for _ in range(4):
            if self.is_interrupted:
                print("[慢系统] 收到打断信号，主动中断执行！")
                return ""
            await asyncio.sleep(0.5)
        return "查询完成：当前网络负载正常，未见异常丢包。"

    async def handle_user_speech(self, audio_event: str):
        self.is_interrupted = False
        print(f"\n[用户开始说话]: {audio_event}")

        # 并行拉起快慢双系统
        slow_task = asyncio.create_task(self.slow_reasoning_system(audio_event))
        
        # 快系统率先响应用户
        async for filler_text in self.fast_filler_system():
            if slow_task.done():
                break
            print(f"[快系统即时发音]: {filler_text}")

        # 等待慢系统最终产出
        final_answer = await slow_task
        if final_answer and not self.is_interrupted:
            print(f"[慢系统输出最终结论并播报]: {final_answer}")

    def trigger_barge_in(self):
        """外部 VAD 监测到人类打断，强行截断当前会话"""
        print("\n>>> [VAD 拦截] 检测到用户打断！立即清空音频播放缓冲区 <<<")
        self.is_interrupted = True

async def main():
    orchestrator = FastSlowVoiceOrchestrator()
    # 场景 1: 完整执行
    await orchestrator.handle_user_speech("帮我全网分析近 24 小时的支付流水告警")
    
    # 场景 2: 中途打断测试
    print("\n--- 模拟打断场景 ---")
    speech_task = asyncio.create_task(orchestrator.handle_user_speech("分析慢查询日志"))
    await asyncio.sleep(0.6)
    orchestrator.trigger_barge_in()
    await speech_task

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