mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-05 19:15:34 +00:00
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import json
|
|
import re
|
|
import time
|
|
|
|
import requests
|
|
from rich import print
|
|
|
|
from .base_translator import Base
|
|
|
|
|
|
class Caiyun(Base):
|
|
"""
|
|
caiyun translator
|
|
"""
|
|
|
|
def __init__(self, key, language, **kwargs) -> None:
|
|
super().__init__(key, language)
|
|
self.api_url = "http://api.interpreter.caiyunai.com/v1/translator"
|
|
self.headers = {
|
|
"content-type": "application/json",
|
|
"x-authorization": f"token {key}",
|
|
}
|
|
# caiyun api only supports: zh2en, zh2ja, en2zh, ja2zh
|
|
self.translate_type = "auto2zh"
|
|
if self.language == "english":
|
|
self.translate_type = "auto2en"
|
|
elif self.language == "japanese":
|
|
self.translate_type = "auto2ja"
|
|
|
|
def rotate_key(self):
|
|
pass
|
|
|
|
def translate(self, text):
|
|
print(text)
|
|
# for caiyun translate src issue #279
|
|
text_list = text.splitlines()
|
|
num = None
|
|
if len(text_list) > 1:
|
|
if text_list[0].isdigit():
|
|
num = text_list[0]
|
|
payload = {
|
|
"source": text,
|
|
"trans_type": self.translate_type,
|
|
"request_id": "demo",
|
|
"detect": True,
|
|
}
|
|
response = requests.request(
|
|
"POST",
|
|
self.api_url,
|
|
data=json.dumps(payload),
|
|
headers=self.headers,
|
|
)
|
|
try:
|
|
t_text = response.json()["target"]
|
|
except Exception as e:
|
|
print(str(e), response.text, "will sleep 60s for the time limit")
|
|
if "limit" in response.json()["message"]:
|
|
print("will sleep 60s for the time limit")
|
|
time.sleep(60)
|
|
response = requests.request(
|
|
"POST",
|
|
self.api_url,
|
|
data=json.dumps(payload),
|
|
headers=self.headers,
|
|
)
|
|
t_text = response.json()["target"]
|
|
|
|
print("[bold green]" + re.sub("\n{3,}", "\n\n", t_text) + "[/bold green]")
|
|
# for issue #279
|
|
if num:
|
|
t_text = str(num) + "\n" + t_text
|
|
return t_text
|