mirror of
https://github.com/alibaba/higress.git
synced 2026-03-04 16:40:50 +08:00
90 lines
2.6 KiB
Python
Executable File
90 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
import os
|
||
import sys
|
||
import json
|
||
import requests
|
||
|
||
def read_file(file_path):
|
||
"""Read file and return its content as a string."""
|
||
try:
|
||
with open(file_path, 'r', encoding='utf-8') as file:
|
||
return file.read()
|
||
except Exception as e:
|
||
print(f"Error reading file: {e}")
|
||
sys.exit(1)
|
||
|
||
def call_openai_api(content, base_url):
|
||
"""Call OpenAI API to translate content from Chinese to English."""
|
||
url = f"http://{base_url}/chat/completions"
|
||
|
||
# Prepare the prompt for OpenAI
|
||
prompt = f"""
|
||
请将以下中文文档翻译成英文。保持原始的Markdown格式，包括标题、列表、代码块等。
|
||
确保翻译准确、专业，并且保持技术术语的正确性。
|
||
|
||
以下是需要翻译的中文文档：
|
||
|
||
{content}
|
||
"""
|
||
|
||
# Prepare the API request
|
||
headers = {
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
data = {
|
||
"model": "gpt-4o",
|
||
"messages": [
|
||
{"role": "system", "content": "你是一个专业的技术文档翻译助手，擅长将中文技术文档翻译成英文。"},
|
||
{"role": "user", "content": prompt}
|
||
],
|
||
"temperature": 0.3
|
||
}
|
||
|
||
try:
|
||
response = requests.post(url, headers=headers, json=data)
|
||
response.raise_for_status()
|
||
|
||
result = response.json()
|
||
if "choices" in result and len(result["choices"]) > 0:
|
||
return result["choices"][0]["message"]["content"]
|
||
else:
|
||
print("Error: Unexpected API response format")
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(f"Error calling OpenAI API: {e}")
|
||
sys.exit(1)
|
||
|
||
def save_markdown(markdown_content, output_file):
|
||
"""Save the Markdown content to a file."""
|
||
try:
|
||
with open(output_file, 'w', encoding='utf-8') as file:
|
||
file.write(markdown_content)
|
||
except Exception as e:
|
||
print(f"Error saving Markdown file: {e}")
|
||
sys.exit(1)
|
||
|
||
def main():
|
||
if len(sys.argv) < 2:
|
||
print("Usage: python translate_readme.py <input_file_path> [output_file_path]")
|
||
sys.exit(1)
|
||
|
||
input_file = sys.argv[1]
|
||
output_file = sys.argv[2] if len(sys.argv) > 2 else "README.md"
|
||
base_url = "127.0.0.1:8080/v1"
|
||
|
||
# Read the Chinese content
|
||
chinese_content = read_file(input_file)
|
||
|
||
# Translate to English
|
||
english_content = call_openai_api(chinese_content, base_url)
|
||
|
||
# Save the translated content
|
||
save_markdown(english_content, output_file)
|
||
|
||
# Print the translated content to stdout
|
||
print(english_content)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|