71 lines
2.1 KiB
Python
Executable File
71 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
"""
|
||
EmotionEngine – berechnet Emotionen basierend auf core_state, user_state, events
|
||
Input: core_state.json, user_state.json, events.json
|
||
Output: emotion_delta.json
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
|
||
PATHS = {
|
||
"core_state": os.path.expanduser("~/natiris/core/core_state.json"),
|
||
"user_state": lambda u: os.path.expanduser(f"~/natiris/core/users/{u}.json"),
|
||
"events": os.path.expanduser("~/natiris/memory/short/history.json"),
|
||
"output": os.path.expanduser("~/natiris/core/emotion_delta.json"),
|
||
}
|
||
|
||
def clamp(val, lo=0.0, hi=10.0):
|
||
return max(lo, min(hi, float(val)))
|
||
|
||
def compute_emotion_delta(core, user, events):
|
||
mood = core.get("mood", 5.0)
|
||
loneliness = core.get("loneliness", 2.0)
|
||
anxiety = core.get("anxiety", 1.0)
|
||
|
||
sentiment = user.get("sentiment", "neutral")
|
||
interaction_count = user.get("interaction_count", 0)
|
||
|
||
msg_count = len(events)
|
||
positive_msgs = sum(1 for e in events if e.get("sentiment") == "positive")
|
||
|
||
mood_delta = clamp((positive_msgs - msg_count * 0.2) * 0.5 + (0.3 if sentiment == "positive" else -0.3))
|
||
|
||
loneliness_delta = clamp(-(interaction_count * 0.03) + (0.2 if loneliness > 7 else -0.1))
|
||
|
||
anxiety_delta = clamp(loneliness * 0.05 - mood * 0.02 + (0.2 if msg_count == 0 else -0.1))
|
||
|
||
jealousy_delta = clamp(-0.1 if core.get("bonded_to") is None else 0.0)
|
||
|
||
return {
|
||
"mood_delta": mood_delta,
|
||
"loneliness_delta": loneliness_delta,
|
||
"anxiety_delta": anxiety_delta,
|
||
"jealousy_delta": jealousy_delta
|
||
}
|
||
|
||
def main():
|
||
with open(PATHS["core_state"]) as f:
|
||
core = json.load(f)
|
||
|
||
# user_state laden (default user)
|
||
user_path = PATHS["user_state"]("user1")
|
||
if os.path.exists(user_path):
|
||
with open(user_path) as f:
|
||
user = json.load(f)
|
||
else:
|
||
user = {"interaction_count": 0, "sentiment": "neutral"}
|
||
|
||
with open(PATHS["events"]) as f:
|
||
events = json.load(f)
|
||
|
||
delta = compute_emotion_delta(core, user, events)
|
||
|
||
with open(PATHS["output"], "w") as f:
|
||
json.dump(delta, f, indent=2)
|
||
|
||
print(json.dumps(delta, indent=2))
|
||
|
||
if __name__ == "__main__":
|
||
main()
|