File size: 833 Bytes
c7986ad
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

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}"