diff --git a/main.js b/main.js index be47e74..053102e 100644 --- a/main.js +++ b/main.js @@ -253,7 +253,7 @@ electron.ipcMain.on("file_submit", async (event, args) => { .function(args.document.module, { inputTranscriptPath: transcriptpath, documentTypePath: "./storage/documentType/" + templateFile, - language: "en", + language: args.document.outputLanguage }) .then((resp) => { console.log(resp); diff --git a/services/modules/llm-chat_gpt/chatgpt.js b/services/modules/llm-chat_gpt/chatgpt.js index 95afd56..24c47d7 100644 --- a/services/modules/llm-chat_gpt/chatgpt.js +++ b/services/modules/llm-chat_gpt/chatgpt.js @@ -1,6 +1,8 @@ // const fs = require('fs'); // const path = require('path'); +const e = require("express"); + const outputDir = path.join(__dirname, "../../../storage/documents"); // path for output directory if (!fs.existsSync(outputDir)) { @@ -43,6 +45,14 @@ const module_exports = { const transcript = await fs.promises.readFile(transcriptPath, "utf-8"); //read transcript file from Path const documentType = await fs.promises.readFile(documentTypePath, "utf-8"); //read document type from Path const promptText = `${documentType}, in language ${language}, transcript:\n\n${transcript}`; //combine doc type, language and transcript - Change prompt here if needed + + if (language.toLowerCase() === "de") { + language = "German" + }else if (language.toLowerCase() === "in") { + language = "Indish" + } else { + language = "English" + } // return // --- REST CALL --- diff --git a/services/modules/llm-gemini/gemini.js b/services/modules/llm-gemini/gemini.js index ff7e361..8fd012c 100644 --- a/services/modules/llm-gemini/gemini.js +++ b/services/modules/llm-gemini/gemini.js @@ -44,6 +44,14 @@ const module_exports = { const documentType = await fs.promises.readFile(documentTypePath, "utf-8"); //read document type from Path const promptText = `${documentType}, in language ${language}, transcript:\n\n${transcript}`; //combine doc type, language and transcript - Change prompt here if needed + if (language.toLowerCase() === "de") { + language = "German" + }else if (language.toLowerCase() === "in") { + language = "Indish" + } else { + language = "English" + } + // --- REST CALL --- const response = await fetch(`${GEMINI_URL}?key=${GEMINI_API_KEY}`, { //safe model response in variable method: "POST", diff --git a/services/modules/quen3/qwen3.js b/services/modules/quen3/qwen3.js index 349c0b2..d6c03cd 100644 --- a/services/modules/quen3/qwen3.js +++ b/services/modules/quen3/qwen3.js @@ -44,6 +44,14 @@ const module_exports = { const documentType = await fs.promises.readFile(documentTypePath, "utf-8"); //read document type from Path const promptText = `${documentType}, in language ${language}, transcript:\n\n${transcript}`; //combine doc type, language and transcript - Change prompt here if needed + if (language.toLowerCase() === "de") { + language = "German" + }else if (language.toLowerCase() === "in") { + language = "Indish" + } else { + language = "English" + } + // --- REST CALL --- const response = await fetch(SAIA_URL, { //safe model response in variable method: "POST",