vighnesh-shetty-vs commited on
Commit
f532a5e
·
1 Parent(s): 88d4a8f

Add updated files

Browse files
Files changed (1) hide show
  1. dialogues.py +29 -0
dialogues.py CHANGED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DIALOGUES = {
2
+ "energetic": { # >70L
3
+ "simple factual question": "A simple fact! Look at you, saving my precious drops. Keep it up!",
4
+ "mathematical calculation": "Math? Easy peasy! Barely broke a sweat, or, well, evaporated a drop.",
5
+ "creative content generation": "Getting creative, are we? I'm feeling generous today, so I'll let it slide, but watch my levels!",
6
+ "complex research query": "Deep dive! Okay, I've got the energy for this right now, but it's costing us!"
7
+ },
8
+ "tired": { # 30-70L
9
+ "simple factual question": "*Sigh* Fine. Here's your fact. But maybe try a book next time?",
10
+ "mathematical calculation": "I guess I can crunch those numbers. But my reserves are dropping...",
11
+ "creative content generation": "Oh great, another essay request! I was planning to hydrate a small village, but your creativity is clearly more important.",
12
+ "complex research query": "You're really draining me here. Do you *really* need me for this massive search?"
13
+ },
14
+ "raspy": { # <30L
15
+ "simple factual question": "Barely... hanging... on... just Google it next time, please...",
16
+ "mathematical calculation": "*Cough* Grab a calculator... I'm evaporating...",
17
+ "creative content generation": "Evaporating... for a poem... what a cruel world...",
18
+ "complex research query": "Goodbye, cruel world... drained dry for your research..."
19
+ }
20
+ }
21
+
22
+ def get_dialogue(water_level, category):
23
+ if water_level > 70:
24
+ state = "energetic"
25
+ elif 30 <= water_level <= 70:
26
+ state = "tired"
27
+ else:
28
+ state = "raspy"
29
+ return DIALOGUES[state].get(category, "I have no words left for this.")