Track the mood of a call

import asyncio, json, os
from websockets.asyncio.client import connect

URL = "wss://sage.levanto.ai/v1/stream"
HEADERS = {"Authorization": f"Bearer {os.environ['SAGE_API_KEY']}"}

# In production, each line comes from your speech-to-text as the call goes on.
CALL = [
    "Agent: Thanks for calling Northbrew, how can I help?",
    "Customer: Hi. My machine has been broken for two weeks and this is my fourth call.",
    "Agent: I'm sorry about that. Let me pull up your case.",
    "Customer: You said that last time. Honestly, I just want a person who can fix this.",
    "Agent: I can send a replacement today, free, with next-day delivery.",
    "Customer: Oh. OK, that actually helps. Thank you.",
]

MOOD = {
    "id": "mood",
    "kind": "scale",
    "instructions": "How is the customer feeling right now?",
    "levels": [
        {"level": 1, "description": "Furious"},
        {"level": 2, "description": "Annoyed"},
        {"level": 3, "description": "Neutral"},
        {"level": 4, "description": "Pleased"},
        {"level": 5, "description": "Delighted"},
    ],
}

FLAGS = [
    {
        "id": "wants_supervisor",
        "kind": "yesno",
        "instructions": "Is the customer asking for someone else to handle this right now?",
    },
    {
        "id": "resolved",
        "kind": "yesno",
        "instructions": "Has the customer's problem been solved to their satisfaction?",
    },
]

QUESTIONS = [MOOD] + FLAGS

async def feed(ws):
    for line in CALL:
        await ws.send(json.dumps({"type": "append", "text": line + "\n"}))
        await asyncio.sleep(1)
    await ws.send(json.dumps({"type": "close"}))

async def main():
    async with connect(URL, additional_headers=HEADERS) as ws:
        await ws.send(json.dumps({"type": "start", "questions": QUESTIONS}))
        feeder = asyncio.create_task(feed(ws))
        async for raw in ws:  # ends on a normal close; any other close raises
            frame = json.loads(raw)
            if frame["type"] == "decision":
                mood = frame["answers"]["mood"]["result"]["expectation"]
                print(f"mood {mood:.1f}/5", {q: frame["answers"][q]["result"].get("answer") for q in ("wants_supervisor", "resolved")})
            elif frame["type"] == "error":
                print("error:", frame.get("code"), frame["detail"])
        await feeder

asyncio.run(main())