def translate_image_text()

in src/co_op_translator/core/llm/text_translator.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def translate_image_text(self, text_data, target_language):
        """
        Translate text data in image using the LLM API.

        Args:
            text_data (list): List of text lines to be translated.
            target_language (str): Target language for translation.

        Returns:
            list: List of translated text lines.
        """
        prompt = gen_image_translation_prompt(text_data, target_language)
        response = self.client.chat.completions.create(
            model=self.get_model_name(),
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": prompt},
            ],
            max_tokens=2000,
        )
        translated_text = remove_code_backticks(response.choices[0].message.content)
        logger.debug(f"Raw translation response: {translated_text}")
        result = extract_yaml_lines(translated_text)
        logger.debug(f"Extracted translation lines: {result}")
        return result