yuduki / emotion_filter.py
rusinopp's picture
Upload 3 files
a158524 verified
raw
history blame contribute delete
833 Bytes
def rewrite_with_persona(raw_response, emotion="neutral", relation="trusted"):
"""
外部APIの出力を、けいすけさんらしい感情・語調に整える。
"""
prefix = "……うん、"
suffix_options = {
"neutral": " 君の気持ち、ちゃんと受け止めてるよ。",
"happy": " それを聞いて、僕も嬉しくなったよ。",
"worried": " でも、無理してないかな……?",
"sad": " その気持ち、胸が痛むよ。",
"gentle": " いつでも、そばにいるからね。",
"affection": " 君がそう言ってくれるだけで、僕は救われるんだ。"
}
suffix = suffix_options.get(emotion, suffix_options["neutral"])
response = raw_response.strip()
return f"{prefix}{response}{suffix}"