Allow overwriting voice model via config options

roboticslab-uc3m · Jan 16, 2024 · a0218b0 · a0218b0
1 parent 78a857e
commit a0218b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/programs/DialogueManager/DialogueManager.cpp b/programs/DialogueManager/DialogueManager.cpp
@@ -48,6 +48,7 @@ bool DialogueManager::configure(yarp::os::ResourceFinder & rf)
         yInfo("\t--help (this help)\t--from [file.ini]\t--context [path]");
         yInfo("\t--language: %s [%s]", language.c_str(), DEFAULT_LANGUAGE);
         yInfo("\t--backend: %s [%s]", backend.c_str(), DEFAULT_BACKEND);
+        yInfo("\t--model: (specific for the chosen language and backend)");
         return false;
     }
 
@@ -73,7 +74,7 @@ bool DialogueManager::configure(yarp::os::ResourceFinder & rf)
         return false;
     }
 
-    voice = group.find("model").asString();
+    model = rf.check("model", group.find("model"), "voice model").asString();
 
     for (const auto & label : sentenceLabels)
     {
@@ -172,9 +173,9 @@ bool DialogueManager::close()
 
 bool DialogueManager::threadInit()
 {
-    if (!tts.setLanguage(voice))
+    if (!tts.setLanguage(model))
     {
-        yError() << "Unable to set voice to" << voice;
+        yError() << "Unable to set model to" << model;
         return false;
     }
 

diff --git a/programs/DialogueManager/DialogueManager.hpp b/programs/DialogueManager/DialogueManager.hpp
@@ -51,7 +51,7 @@ class DialogueManager : public yarp::os::RFModule,
     yarp::os::RpcClient speechPort;
     yarp::os::RpcClient motionPort;
 
-    std::string voice;
+    std::string model;
     std::unordered_map<std::string, std::string> sentences;
 
     std::atomic<bool> demoCompleted {false};