Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
xunbu
2025-06-10 19:47:09 +08:00
parent 8eb77e192a
commit a7de5bc5b9
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
docutranslate/translater.py
View File

@@ -274,7 +274,7 @@ class FileTranslater:
if isinstance(output_dir, str): if isinstance(output_dir, str):
output_dir = Path(output_dir) output_dir = Path(output_dir)
if filename is None: if filename is None:
filename=f"{self.document.stem}_translated.md" filename=f"{self.document.stem}.md"
# 确保输出目录存在 # 确保输出目录存在
output_dir.mkdir(parents=True, exist_ok=True) output_dir.mkdir(parents=True, exist_ok=True)
full_name = output_dir / filename full_name = output_dir / filename
@@ -297,7 +297,7 @@ class FileTranslater:
output_dir = Path(output_dir) output_dir = Path(output_dir)
if filename is None: if filename is None:
if filename is None: if filename is None:
filename = f"{self.document.stem}_translated.html" filename = f"{self.document.stem}.html"
# 确保输出目录存在 # 确保输出目录存在
output_dir.mkdir(parents=True, exist_ok=True) output_dir.mkdir(parents=True, exist_ok=True)
full_name = output_dir / filename full_name = output_dir / filename
@@ -373,13 +373,12 @@ class FileTranslater:
if refine: if refine:
self.refine_markdown_by_agent(refine_agent) self.refine_markdown_by_agent(refine_agent)
self.translate_markdown_by_agent(translate_agent, to_lang=to_lang, custom_prompt=custom_prompt_translate) self.translate_markdown_by_agent(translate_agent, to_lang=to_lang, custom_prompt=custom_prompt_translate)
if save: if save:
if output_format == "markdown": if output_format == "markdown":
filename = f"{file_path.stem}_{to_lang}.md" self.save_as_markdown(f"{self.document.stem}_{to_lang}.md",output_dir=output_dir)
self.save_as_markdown(filename=filename, output_dir=output_dir)
elif output_format == "html": elif output_format == "html":
filename = f"{file_path.stem}_{to_lang}.html" self.save_as_html(f"{self.document.stem}_{to_lang}.html", output_dir=output_dir)
self.save_as_html(filename=filename, output_dir=output_dir)
return self return self
async def translate_file_async(self, file_path: Path | str | None = None, to_lang="中文", output_dir="./output", async def translate_file_async(self, file_path: Path | str | None = None, to_lang="中文", output_dir="./output",