Commit 56058cca authored by 周成波's avatar 周成波

更换翻译接口为专业模型

parent 533cd168
...@@ -21,7 +21,7 @@ export default { ...@@ -21,7 +21,7 @@ export default {
}) })
.catch((err: any) => { .catch((err: any) => {
console.log(err); console.log(err);
return Promise.reject("与LLM通讯失败"); return Promise.reject(`与LLM通讯失败:${JSON.stringify(err.message.message)}`);
}); });
}, },
...@@ -64,7 +64,7 @@ export default { ...@@ -64,7 +64,7 @@ export default {
}) })
.catch((err: any) => { .catch((err: any) => {
console.log(err); console.log(err);
return Promise.reject("与 stable-diffusion-webui Api 通讯失败"); return Promise.reject(`与 stable-diffusion-webui Api 通讯失败:${JSON.stringify(err.message.message)}`);
}); });
}, },
...@@ -85,7 +85,7 @@ export default { ...@@ -85,7 +85,7 @@ export default {
}) })
.catch((err: any) => { .catch((err: any) => {
console.log(err); console.log(err);
return Promise.reject("gen_video接口通讯失败"); return Promise.reject(`gen_video接口通讯失败:${JSON.stringify(err.message.message)}`);
}); });
}, },
...@@ -106,7 +106,28 @@ export default { ...@@ -106,7 +106,28 @@ export default {
}) })
.catch((err: any) => { .catch((err: any) => {
console.log(err); console.log(err);
return Promise.reject("与pwd_check接口通讯失败"); return Promise.reject(`与pwd_check接口通讯失败:${JSON.stringify(err.message.message)}`);
});
},
submitTranslateToEn(input_string: string): Promise<string> {
if (!input_string) {
return Promise.reject("输入不能为空");
}
const post_data = { input_string: input_string }
return request.post('/text2video/translate_to_en', post_data)
.then((res: any) => {
// console.log(res);
if (res && res.code === 0) {
return res.data.result;
} else {
const errorMessage = res ? res.message : "未知错误";
return Promise.reject(errorMessage);
}
})
.catch((err: any) => {
console.log(err);
return Promise.reject(`与翻译接口通讯失败:${JSON.stringify(err.message.message)}`);
}); });
}, },
} }
...@@ -153,8 +153,10 @@ const onAdaptRolesKeywords = async () => { ...@@ -153,8 +153,10 @@ const onAdaptRolesKeywords = async () => {
one_role.角色关键词 = keywords; one_role.角色关键词 = keywords;
await delay(100); await delay(100);
const adapt_attribute_restrict_en = `你现在扮演专业的英语翻译的角色。请将这段文字“${keywords}”翻译为英语。\n要求:只返回英语即可,不要返回其他内容。`; // const adapt_attribute_restrict_en = `你现在扮演专业的英语翻译的角色。请将这段文字“${keywords}”翻译为英语。\n要求:只返回英语即可,不要返回其他内容。`;
let keywords_en = await text2videoService.submitLLM(adapt_attribute_restrict_en, role_keywords_llm.api); // let keywords_en = await text2videoService.submitLLM(adapt_attribute_restrict_en, role_keywords_llm.api);
let keywords_en = await text2videoService.submitTranslateToEn(keywords);
console.log(keywords_en)
keywords_en = utils.filterChineseAndPunctuation(keywords_en.replace(/"/g, '')); keywords_en = utils.filterChineseAndPunctuation(keywords_en.replace(/"/g, ''));
one_role.角色英文关键词 = keywords_en; one_role.角色英文关键词 = keywords_en;
} }
...@@ -253,10 +255,11 @@ const onAdaptOneScene = async (item: any) => { ...@@ -253,10 +255,11 @@ const onAdaptOneScene = async (item: any) => {
} }
// 推理场景 // 推理场景
try { try {
const adapt_restrict = `你现在扮演专业的英语翻译的角色。请将这段文字“${item.场景描述}”翻译为英语。\n要求:只返回英语即可,不要返回其他内容。`; // const adapt_restrict = `你现在扮演专业的英语翻译的角色。请将这段文字“${item.场景描述}”翻译为英语。\n要求:只返回英语即可,不要返回其他内容。`;
const keywords = await text2videoService.submitLLM(adapt_restrict, tuili_llm.api); // const keywords_en = await text2videoService.submitLLM(adapt_restrict, tuili_llm.api);
// console.log(keywords) let keywords_en = await text2videoService.submitTranslateToEn(item.场景描述);
item.场景关键词 = utils.filterChineseAndPunctuation(keywords); console.log(keywords_en)
item.场景关键词 = utils.filterChineseAndPunctuation(keywords_en);
} catch (error) { } catch (error) {
ElMessage({ ElMessage({
message: String(error), message: String(error),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment