brian/tangmo_trans_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only use HTML mode when necessary

Browse Source
This commit is contained in:
TangMo
2025-08-29 01:16:42 +00:00
parent b770ba83d6
commit c613af2b38
1 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/Commands/GenericmessageCommand.php
View File

@@ -27,7 +27,7 @@ class GenericmessageCommand extends SystemCommand
$message = $this->getMessage();
$chat_id = $message->getChat()->getId();
if(!in_array($chat_id, $allowed_chat_ids)) {
if (!in_array($chat_id, $allowed_chat_ids)) {
return $this->replyToChat("Chat ID `$chat_id` not allowed");
}
@@ -38,37 +38,45 @@ class GenericmessageCommand extends SystemCommand
$text_chara_count = count(preg_split('//u', $text, -1, PREG_SPLIT_NO_EMPTY));
$text_english_count = count(preg_split('~[^a-z 0-9\'\",.]*~i', $text, 0, PREG_SPLIT_NO_EMPTY));
if ($text_english_count >= ($text_chara_count*2 /3)) {
if ($text_english_count >= ($text_chara_count * 2 / 3)) {
$sourceLang = 'en';
$targetLang = 'zh-hans';
} else {
$sourceLang = 'zh';
$targetLang = 'en-US';
}
$htmlMode = $message->getEntities() && count($message->getEntities()) > 0;
$deeplOptions = [
TranslateTextOptions::FORMALITY => 'prefer_less',
TranslateTextOptions::MODEL_TYPE => 'prefer_quality_optimized',
TranslateTextOptions::TAG_HANDLING => 'html',
TranslateTextOptions::SPLIT_SENTENCES => 'on',
];
if ($htmlMode) {
$deeplOptions[TranslateTextOptions::TAG_HANDLING] = 'html';
}
if (isset($deepl_glossary_id)) {
$deeplOptions[TranslateTextOptions::GLOSSARY] = $deepl_glossary_id;
}
$this->translator = new Translator($deepl_api_key);
try {
$decoded_text = (new EntityDecoder('HTML'))->decode(json_decode($message->toJson()));
$translated = $this->translator->translateText($decoded_text, $sourceLang, $targetLang, $deeplOptions);
if ($htmlMode) {
$text = (new EntityDecoder('HTML'))->decode(json_decode($message->toJson()));
// $text = str_replace("\n", '<br />', $text);
}
$translated = $this->translator->translateText($text, $sourceLang, $targetLang, $deeplOptions);
$translated = $translated->text;
} catch (\Exception $e) {
return $this->replyToChat(get_class($e) . ': ' . $e->getMessage()); // ?? var_export($e, true)
}
$translated = str_replace('&lt;BR /&gt;', "\n", $translated);
// $translated = str_replace('<br />', "\n", $translated);
$message_forwarded_from = $message->getForwardFrom()?->getFirstName() ?? $message->getForwardSenderName();
if ($message_forwarded_from !== null) {
$translated = $message_forwarded_from . ': ' . $translated;
$forwardedFrom = $message->getForwardFrom();
if ($forwardedFrom) {
$translated = $forwardedFrom['first_name'] . ': ' . $translated;
}
return $this->replyToChat($translated, ['disable_notification' => true, 'parse_mode' => 'HTML']);
return $this->replyToChat($translated, ['disable_notification' => true, 'parse_mode' => $htmlMode ? 'HTML' : null]);
}
}
}