mirror of
https://github.com/HiMeditator/auto-caption.git
synced 2026-02-04 04:14:42 +08:00
0825e48902
- 新增GLM-ASR云端语音识别引擎实现 - 扩展配置界面添加GLM相关参数设置 - Ollama支持自定义域名和Apikey以支持云端和其他LLM - 修改音频处理逻辑以支持新引擎 - 更新依赖项和构建配置 - 修复Ollama翻译功能相关问题
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
from ollama import chat, Client
|
|
from ollama import ChatResponse
|
|
try:
|
|
from openai import OpenAI
|
|
except ImportError:
|
|
OpenAI = None
|
|
import asyncio
|
|
from googletrans import Translator
|
|
from .sysout import stdout_cmd, stdout_obj
|
|
|
|
lang_map = {
|
|
'en': 'English',
|
|
'es': 'Spanish',
|
|
'fr': 'French',
|
|
'de': 'German',
|
|
'it': 'Italian',
|
|
'ru': 'Russian',
|
|
'ja': 'Japanese',
|
|
'ko': 'Korean',
|
|
'zh': 'Chinese',
|
|
'zh-cn': 'Chinese'
|
|
}
|
|
|
|
def ollama_translate(model: str, target: str, text: str, time_s: str, url: str = '', key: str = ''):
|
|
content = ""
|
|
try:
|
|
if url:
|
|
if OpenAI:
|
|
client = OpenAI(base_url=url, api_key=key if key else "ollama")
|
|
response = client.chat.completions.create(
|
|
model=model,
|
|
messages=[
|
|
{"role": "system", "content": f"/no_think Translate the following content into {lang_map[target]}, and do not output any additional information."},
|
|
{"role": "user", "content": text}
|
|
]
|
|
)
|
|
content = response.choices[0].message.content or ""
|
|
else:
|
|
client = Client(host=url)
|
|
response: ChatResponse = client.chat(
|
|
model=model,
|
|
messages=[
|
|
{"role": "system", "content": f"/no_think Translate the following content into {lang_map[target]}, and do not output any additional information."},
|
|
{"role": "user", "content": text}
|
|
]
|
|
)
|
|
content = response.message.content or ""
|
|
else:
|
|
response: ChatResponse = chat(
|
|
model=model,
|
|
messages=[
|
|
{"role": "system", "content": f"/no_think Translate the following content into {lang_map[target]}, and do not output any additional information."},
|
|
{"role": "user", "content": text}
|
|
]
|
|
)
|
|
content = response.message.content or ""
|
|
except Exception as e:
|
|
stdout_cmd("warn", f"Translation failed: {str(e)}")
|
|
return
|
|
|
|
if content.startswith('<think>'):
|
|
index = content.find('</think>')
|
|
if index != -1:
|
|
content = content[index+8:]
|
|
stdout_obj({
|
|
"command": "translation",
|
|
"time_s": time_s,
|
|
"text": text,
|
|
"translation": content.strip()
|
|
})
|
|
|
|
def google_translate(model: str, target: str, text: str, time_s: str):
|
|
translator = Translator()
|
|
try:
|
|
res = asyncio.run(translator.translate(text, dest=target))
|
|
stdout_obj({
|
|
"command": "translation",
|
|
"time_s": time_s,
|
|
"text": text,
|
|
"translation": res.text
|
|
})
|
|
except Exception as e:
|
|
stdout_cmd("warn", f"Google translation request failed, please check your network connection...")
|