mgbam commited on
Commit
eaead5d
verified
1 Parent(s): 7319700

Update core_creator/assets_manager.py

Browse files
Files changed (1) hide show
  1. core_creator/assets_manager.py +33 -0
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"))