InitialCommit

This commit is contained in:
R5600U_PC 2024-11-06 08:25:45 +09:00
commit 86dc185f1e
3 changed files with 98 additions and 0 deletions

4
.gitignore vendored Normal file
View File

@ -0,0 +1,4 @@
Lib/
Scripts/
Include/
pyvenv.cfg

21
d.py Normal file
View File

@ -0,0 +1,21 @@
from deep_translator import (GoogleTranslator, PapagoTranslator, BaiduTranslator)
def google_translate(text, target_lang='ko'):
translated = GoogleTranslator(source='auto', target=target_lang).translate(text)
return f"Google Translate: {translated}"
def papago_translate(text, target_lang='ko'):
translated = PapagoTranslator(client_id='YOUR_CLIENT_ID', secret_key='YOUR_SECRET_KEY', source='auto', target=target_lang).translate(text)
return f"Papago Translate: {translated}"
def baidu_translate(text, target_lang='zh'):
translated = BaiduTranslator(appid='YOUR_APP_ID', appkey='YOUR_APP_KEY', source='auto', target=target_lang).translate(text)
return f"Baidu Translate: {translated}"
# Example usage
text = "百度AI开放平台 \n 全球领先的人工智能服务平台"
print(google_translate(text))
print(papago_translate(text))
print(baidu_translate(text))

73
main.py Normal file
View File

@ -0,0 +1,73 @@
from transformers import MarianMTModel, MarianTokenizer, MBartForConditionalGeneration, MBart50TokenizerFast
import requests
import hashlib
import random
import urllib.request
import json
# Baidu API 설정
BAIDU_APP_ID = 'YOUR_APP_ID'
BAIDU_SECRET_KEY = 'YOUR_SECRET_KEY'
# Papago API 설정
PAPAGO_CLIENT_ID = 'YOUR_CLIENT_ID'
PAPAGO_CLIENT_SECRET = 'YOUR_CLIENT_SECRET'
def marian_translate_zh_ko(text):
model_name = 'Helsinki-NLP/opus-mt-zh-ko'
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
translated = model.generate(**inputs)
translated_text = tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
return f"MarianMT Translate (zh-ko): {translated_text}"
def mbart_translate_zh_ko(text):
model_name = 'facebook/mbart-large-50-many-to-many-mmt'
tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
model = MBartForConditionalGeneration.from_pretrained(model_name)
tokenizer.src_lang = 'zh_CN'
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
translated = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id['ko_KR'])
translated_text = tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
return f"mBART Translate (zh-ko): {translated_text}"
def baidu_translate_zh_ko(text):
url = "https://fanyi-api.baidu.com/api/trans/vip/translate"
salt = random.randint(32768, 65536)
sign = hashlib.md5((BAIDU_APP_ID + text + str(salt) + BAIDU_SECRET_KEY).encode('utf-8')).hexdigest()
params = {
'q': text,
'from': 'zh',
'to': 'kor',
'appid': BAIDU_APP_ID,
'salt': salt,
'sign': sign
}
response = requests.get(url, params=params)
result = response.json()
return f"Baidu Translate (zh-ko): {result['trans_result'][0]['dst']}"
def papago_translate_zh_ko(text):
encText = urllib.parse.quote(text)
data = "source=zh-CN&target=ko&text=" + encText
url = "https://openapi.naver.com/v1/papago/n2mt"
request = urllib.request.Request(url)
request.add_header("X-Naver-Client-Id", PAPAGO_CLIENT_ID)
request.add_header("X-Naver-Client-Secret", PAPAGO_CLIENT_SECRET)
response = urllib.request.urlopen(request, data=data.encode("utf-8"))
rescode = response.getcode()
if(rescode==200):
response_body = response.read()
result = json.loads(response_body.decode('utf-8'))
return f"Papago Translate (zh-ko): {result['message']['result']['translatedText']}"
else:
return "Error: Papago translation failed."
# Example usage
text = "你好"
print(marian_translate_zh_ko(text))
print(mbart_translate_zh_ko(text))
print(baidu_translate_zh_ko(text))
print(papago_translate_zh_ko(text))