meraj12 commited on
Commit
f962af2
·
verified ·
1 Parent(s): 5e80325

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +7 -4
utils.py CHANGED
@@ -11,12 +11,15 @@ def save_uploaded_audio(uploaded_file, filename):
11
  f.write(uploaded_file.read())
12
  return output_path
13
 
14
- def get_voice_preset(gender):
15
  presets = {
16
- "Male": "male-preset-1",
17
- "Female": "female-preset-1"
 
 
18
  }
19
- return presets.get(gender, "male-preset-1")
 
20
 
21
  def save_chat_history(chat_data, filename=None):
22
  os.makedirs("history", exist_ok=True)
 
11
  f.write(uploaded_file.read())
12
  return output_path
13
 
14
+ def get_voice_preset(gender, emotion):
15
  presets = {
16
+ ("Male", "Happy"): "male-happy-preset",
17
+ ("Male", "Sad"): "male-sad-preset",
18
+ ("Female", "Happy"): "female-happy-preset",
19
+ ("Female", "Sad"): "female-sad-preset",
20
  }
21
+ return presets.get((gender, emotion), "male-happy-preset")
22
+
23
 
24
  def save_chat_history(chat_data, filename=None):
25
  os.makedirs("history", exist_ok=True)