run format
This commit is contained in:
parent
4231cc0059
commit
71d0eeed73
2 changed files with 11 additions and 6 deletions
|
|
@ -10,9 +10,9 @@ use async_openai::types::{
|
||||||
ChatCompletionRequestMessage, ChatCompletionRequestUserMessageArgs,
|
ChatCompletionRequestMessage, ChatCompletionRequestUserMessageArgs,
|
||||||
CreateChatCompletionRequestArgs,
|
CreateChatCompletionRequestArgs,
|
||||||
};
|
};
|
||||||
|
use ferrous_opencc::{config::BuiltinConfig, OpenCC};
|
||||||
use log::{debug, error};
|
use log::{debug, error};
|
||||||
use once_cell::sync::Lazy;
|
use once_cell::sync::Lazy;
|
||||||
use ferrous_opencc::{OpenCC, config::BuiltinConfig};
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use std::time::Instant;
|
use std::time::Instant;
|
||||||
|
|
@ -168,13 +168,16 @@ async fn maybe_convert_chinese_variant(
|
||||||
// Check if language is set to Simplified or Traditional Chinese
|
// Check if language is set to Simplified or Traditional Chinese
|
||||||
let is_simplified = settings.selected_language == "zh-Hans";
|
let is_simplified = settings.selected_language == "zh-Hans";
|
||||||
let is_traditional = settings.selected_language == "zh-Hant";
|
let is_traditional = settings.selected_language == "zh-Hant";
|
||||||
|
|
||||||
if !is_simplified && !is_traditional {
|
if !is_simplified && !is_traditional {
|
||||||
debug!("selected_language is not Simplified or Traditional Chinese; skipping translation");
|
debug!("selected_language is not Simplified or Traditional Chinese; skipping translation");
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
|
|
||||||
debug!("Starting Chinese translation using OpenCC for language: {}", settings.selected_language);
|
debug!(
|
||||||
|
"Starting Chinese translation using OpenCC for language: {}",
|
||||||
|
settings.selected_language
|
||||||
|
);
|
||||||
|
|
||||||
// Use OpenCC to convert based on selected language
|
// Use OpenCC to convert based on selected language
|
||||||
let config = if is_simplified {
|
let config = if is_simplified {
|
||||||
|
|
@ -184,7 +187,7 @@ async fn maybe_convert_chinese_variant(
|
||||||
// Convert Simplified Chinese to Traditional Chinese
|
// Convert Simplified Chinese to Traditional Chinese
|
||||||
BuiltinConfig::S2twp
|
BuiltinConfig::S2twp
|
||||||
};
|
};
|
||||||
|
|
||||||
match OpenCC::from_config(config) {
|
match OpenCC::from_config(config) {
|
||||||
Ok(converter) => {
|
Ok(converter) => {
|
||||||
let converted = converter.convert(transcription);
|
let converted = converter.convert(transcription);
|
||||||
|
|
|
||||||
|
|
@ -354,14 +354,16 @@ impl TranscriptionManager {
|
||||||
let whisper_language = if settings.selected_language == "auto" {
|
let whisper_language = if settings.selected_language == "auto" {
|
||||||
None
|
None
|
||||||
} else {
|
} else {
|
||||||
let normalized = if settings.selected_language == "zh-Hans" || settings.selected_language == "zh-Hant" {
|
let normalized = if settings.selected_language == "zh-Hans"
|
||||||
|
|| settings.selected_language == "zh-Hant"
|
||||||
|
{
|
||||||
"zh".to_string()
|
"zh".to_string()
|
||||||
} else {
|
} else {
|
||||||
settings.selected_language.clone()
|
settings.selected_language.clone()
|
||||||
};
|
};
|
||||||
Some(normalized)
|
Some(normalized)
|
||||||
};
|
};
|
||||||
|
|
||||||
let params = WhisperInferenceParams {
|
let params = WhisperInferenceParams {
|
||||||
language: whisper_language,
|
language: whisper_language,
|
||||||
translate: settings.translate_to_english,
|
translate: settings.translate_to_english,
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue