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翻译功能相关问题
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import sys
|
|
import json
|
|
import sherpa_onnx
|
|
|
|
display_caption = False
|
|
caption_index = -1
|
|
display = sherpa_onnx.Display()
|
|
|
|
def stdout(text: str):
|
|
stdout_cmd("print", text)
|
|
|
|
def stdout_err(text: str):
|
|
stdout_cmd("error", text)
|
|
|
|
def stdout_cmd(command: str, content = ""):
|
|
msg = { "command": command, "content": content }
|
|
sys.stdout.write(json.dumps(msg) + "\n")
|
|
sys.stdout.flush()
|
|
|
|
def change_caption_display(val: bool):
|
|
global display_caption
|
|
display_caption = val
|
|
|
|
def caption_display(obj):
|
|
global display_caption
|
|
global caption_index
|
|
global display
|
|
|
|
if caption_index >=0 and caption_index != int(obj['index']):
|
|
display.finalize_current_sentence()
|
|
caption_index = int(obj['index'])
|
|
full_text = f"{obj['text']}\n{obj['translation']}"
|
|
if obj['translation']:
|
|
full_text += "\n"
|
|
display.update_text(full_text)
|
|
display.display()
|
|
|
|
def translation_display(obj):
|
|
global original_caption
|
|
global display
|
|
full_text = f"{obj['text']}\n{obj['translation']}"
|
|
if obj['translation']:
|
|
full_text += "\n"
|
|
display.update_text(full_text)
|
|
display.display()
|
|
display.finalize_current_sentence()
|
|
|
|
def stdout_obj(obj):
|
|
global display_caption
|
|
if obj['command'] == 'caption' and display_caption:
|
|
caption_display(obj)
|
|
return
|
|
if obj['command'] == 'translation' and display_caption:
|
|
translation_display(obj)
|
|
return
|
|
sys.stdout.write(json.dumps(obj) + "\n")
|
|
sys.stdout.flush()
|
|
|
|
def stderr(text: str):
|
|
sys.stderr.write(text + "\n")
|
|
sys.stderr.flush()
|