Update core_creator/assets_manager.py
Browse files
core_creator/assets_manager.py
CHANGED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# assets_manager.py - Fetches or generates assets like avatars, audio, visuals
|
2 |
+
import os
|
3 |
+
import random
|
4 |
+
|
5 |
+
AVATAR_DIR = "assets/default_avatars"
|
6 |
+
|
7 |
+
# For now, we use static files. Later this can connect to DALL路E, Gemini image gen, etc.
|
8 |
+
CATEGORY_AVATARS = {
|
9 |
+
"educational": ["teacher_bot.png", "friendly_helper.png"],
|
10 |
+
"assistive": ["aid_bot.png"],
|
11 |
+
"entertainment": ["party_bot.png"],
|
12 |
+
"industrial": ["mech_bot.png"],
|
13 |
+
"healthcare": ["nurse_bot.png"],
|
14 |
+
"home automation": ["smart_home_avatar.png"],
|
15 |
+
"retail": ["greeter_bot.png"],
|
16 |
+
"creative": ["abstract_bot.png"]
|
17 |
+
}
|
18 |
+
|
19 |
+
def fetch_visual_assets(intent_category: str) -> dict:
|
20 |
+
avatars = CATEGORY_AVATARS.get(intent_category, CATEGORY_AVATARS["creative"])
|
21 |
+
selected = random.choice(avatars)
|
22 |
+
path = os.path.join(AVATAR_DIR, selected)
|
23 |
+
|
24 |
+
return {
|
25 |
+
"avatar_image": path,
|
26 |
+
"tts_voice": "gpt-4o-mini-tts",
|
27 |
+
"style": "friendly",
|
28 |
+
"theme_color": "#4F46E5" # Default vibrant purple
|
29 |
+
}
|
30 |
+
|
31 |
+
# Example
|
32 |
+
if __name__ == "__main__":
|
33 |
+
print(fetch_visual_assets("retail"))
|