60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
# Copyright (c) 2025 Patrick Motsch
|
|
# All rights reserved.
|
|
|
|
"""
|
|
Translate Document action for AI operations.
|
|
Translates documents to a target language while preserving formatting and structure.
|
|
"""
|
|
|
|
import logging
|
|
from typing import Dict, Any
|
|
from modules.workflows.methods.methodBase import action
|
|
from modules.datamodels.datamodelChat import ActionResult
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@action
|
|
async def translateDocument(self, parameters: Dict[str, Any]) -> ActionResult:
|
|
"""
|
|
GENERAL:
|
|
- Purpose: Translate documents to a target language while preserving formatting and structure.
|
|
- Input requirements: documentList (required); targetLanguage (required).
|
|
- Output format: Translated document in same format as input (default) or specified resultType.
|
|
|
|
Parameters:
|
|
- documentList (list, required): Document reference(s) to translate.
|
|
- targetLanguage (str, required): Target language code or name (e.g., "de", "German", "French", "es").
|
|
- sourceLanguage (str, optional): Source language if known (e.g., "en", "English"). If not provided, AI will detect.
|
|
- preserveFormatting (bool, optional): Whether to preserve original formatting. Default: True.
|
|
- resultType (str, optional): Output file extension. If not specified, uses same format as input.
|
|
"""
|
|
documentList = parameters.get("documentList", [])
|
|
if not documentList:
|
|
return ActionResult.isFailure(error="documentList is required")
|
|
|
|
targetLanguage = parameters.get("targetLanguage")
|
|
if not targetLanguage:
|
|
return ActionResult.isFailure(error="targetLanguage is required")
|
|
|
|
sourceLanguage = parameters.get("sourceLanguage")
|
|
preserveFormatting = parameters.get("preserveFormatting", True)
|
|
resultType = parameters.get("resultType")
|
|
|
|
aiPrompt = f"Translate the provided document(s) to {targetLanguage}."
|
|
if sourceLanguage:
|
|
aiPrompt += f" The source language is {sourceLanguage}."
|
|
if preserveFormatting:
|
|
aiPrompt += " Preserve all formatting, structure, tables, and layout exactly as they appear in the original document."
|
|
else:
|
|
aiPrompt += " Focus on accurate translation of content."
|
|
aiPrompt += " Maintain the same document structure, headings, and organization."
|
|
|
|
processParams = {
|
|
"aiPrompt": aiPrompt,
|
|
"documentList": documentList
|
|
}
|
|
if resultType:
|
|
processParams["resultType"] = resultType
|
|
|
|
return await self.process(processParams)
|
|
|