From 0b34d27c40b295dd58ad047d92c9dba86ee67514 Mon Sep 17 00:00:00 2001 From: Erfan Momeni Date: Mon, 8 Jul 2024 21:40:17 +0330 Subject: [PATCH] feat: add source language to the prompt for having better result --- deep_translator/chatgpt.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/deep_translator/chatgpt.py b/deep_translator/chatgpt.py index 2ab3777..e6d7e7f 100644 --- a/deep_translator/chatgpt.py +++ b/deep_translator/chatgpt.py @@ -44,7 +44,10 @@ def translate(self, text: str, **kwargs) -> str: openai.api_key = self.api_key - prompt = f"Translate the text below into {self.target}.\n" + if self.source == "auto": + prompt = f"Translate the text below into {self.target}.\n" + else: + prompt = f"Translate the text below from {self.source} into {self.target}.\n" prompt += f'Text: "{text}"' response = openai.ChatCompletion.create(