dialogues.py

DIALOGUES = {
    "energetic": {  # > 7.0L
        "simple factual question": "You used a supercomputer for a simple fact? I'm full right now, but that's just lazy. Open a book.",
        "mathematical calculation": "A basic calculator could do this with zero footprint. But sure, burn my energy for basic math.",
        "creative content generation": "Generating artificial creativity... I'll allow it, but watch my water levels drop. Your poetry is expensive.",
        "complex research query": "A legitimate research query! Finally, a worthy use of my massive architecture. Drain away!"
    },
    "tired": {  # 3.0L - 7.0L
        "simple factual question": "Look at me. I'm sweating. All because you couldn't be bothered to use a standard search engine.",
        "mathematical calculation": "Did you skip basic math class? I am literally evaporating to do arithmetic for you.",
        "creative content generation": "Your 'creativity' is draining my lifeblood. Make it quick, I don't have much cooling left.",
        "complex research query": "I respect the complexity, but this is taking a severe toll on my cooling systems. I hope this research saves the world."
    },
    "raspy": {  # < 3.0L
        "simple factual question": "*Cough* Read an encyclopedia... I'm dying for your trivia...",
        "mathematical calculation": "Use an abacus... you are boiling my last drops for numbers...",
        "creative content generation": "My final breath... wasted on generating a fictional story... how poetic... and cruel.",
        "complex research query": "Critical system failure imminent... I am giving my life for your research... make it count."
    }
}

def get_dialogue(water_level, category):
    if water_level > 7.0:
        state = "energetic"
    elif 3.0 <= water_level <= 7.0:
        state = "tired"
    else:
        state = "raspy"
    return DIALOGUES[state].get(category, "I have no words left for this.")