Added genderAudio parameter (#30)

AgrI-Mitra · Sep 3, 2024 · 8e69d30 · 8e69d30
1 parent 45e6e82
commit 8e69d30
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/app.controller.ts b/src/app.controller.ts
@@ -224,7 +224,7 @@ export class AppController {
       schemeName:
         promptDto.schemeName && promptDto.schemeName.trim() !== ""
           ? promptDto.schemeName
-          : "PM Kisan",
+          : "All Schemes",
     };
 
     let conversation = await this.conversationService.getConversationState(
@@ -502,7 +502,8 @@ export class AppController {
       };
       res["audio"] = await this.aiToolsService.textToSpeech(
         res.text,
-        prompt.inputLanguage
+        prompt.inputLanguage,
+        promptDto.audioGender
       );
       if (res["audio"]["error"]) {
         await this.telemetryService.capture({
@@ -1135,7 +1136,8 @@ export class AppController {
         textToaudio = removeLinks(textToaudio);
         result["audio"] = await this.aiToolsService.textToSpeech(
           textToaudio,
-          isNumber ? Language.en : prompt.inputLanguage
+          isNumber ? Language.en : prompt.inputLanguage,
+          promptDto.audioGender
         );
         if (result["audio"]["error"]) {
           await this.telemetryService.capture({

diff --git a/src/modules/aiTools/ai-tools.service.ts b/src/modules/aiTools/ai-tools.service.ts
@@ -182,7 +182,7 @@ export class AiToolsService {
     }
   }
 
-  async textToSpeech(text: string, language: Language) {
+  async textToSpeech(text: string, language: Language, audioGender: string = 'male') {
     try {
       let config: any = await this.getBhashiniConfig("tts", {
         language: {
@@ -199,6 +199,7 @@ export class AiToolsService {
           language: {
             sourceLanguage: language,
           },
+          gender: audioGender
         },
         {
           input: [
@@ -385,7 +386,9 @@ export class AiToolsService {
     myHeaders.append("Content-Type", "application/json");
     config["serviceId"] = serviceId;
     if (task == "tts") {
-      config["gender"] = "male";
+      if(!config["gender"]){
+        config["gender"] = "male";
+      }
       config["samplingRate"] = 8000;
     }
     var raw = JSON.stringify({