from src.llm_models.utils_model import LLMRequest
from src.common.logger import get_logger
from src.config.config import model_config
from src.plugin_system.apis import send_api

logger = get_logger(__name__)

head_actions_list = ["不做额外动作", "点头一次", "点头两次", "摇头", "歪脑袋", "低头望向一边"]


async def yes_or_no_head(text: str, emotion: str = "", chat_history: str = "", chat_id: str = ""):
    prompt = f"""
{chat_history}
以上是对方的发言：

对这个发言，你的心情是：{emotion}
对上面的发言，你的回复是：{text}
请判断时是否要伴随回复做头部动作，你可以选择：

不做额外动作
点头一次
点头两次
摇头
歪脑袋
低头望向一边

请从上面的动作中选择一个，并输出，请只输出你选择的动作就好，不要输出其他内容。"""
    model = LLMRequest(model_set=model_config.model_task_config.emotion, request_type="motion")

    try:
        # logger.info(f"prompt: {prompt}")
        response, _ = await model.generate_response_async(prompt=prompt, temperature=0.7)
        logger.info(f"response: {response}")

        head_action = response if response in head_actions_list else "不做额外动作"
        await send_api.custom_to_stream(
            message_type="head_action",
            content=head_action,
            stream_id=chat_id,
            storage_message=False,
            show_log=True,
        )

    except Exception as e:
        logger.error(f"yes_or_no_head error: {e}")
        return "不做额外动作"