Remove notes added by ChatGPT to translation

macopedia · May 9, 2023 · 085a68b · 085a68b
1 parent 766c72e
commit 085a68b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/Macopedia/OpenAiTranslator/Client/OpenAiClient/Response/ErrorResponse.php b/src/Macopedia/OpenAiTranslator/Client/OpenAiClient/Response/ErrorResponse.php
@@ -9,12 +9,17 @@
 class ErrorResponse extends Response
 {
     public function __construct(
-        private string $error
+        private string|array $error
     ) {
     }
 
     public function getError(): string
     {
-        return $this->error;
+        $error = $this->error;
+
+        if (is_array($error)) {
+            $error = json_encode($error);
+        }
+        return $error;
     }
 }
diff --git a/src/Macopedia/OpenAiTranslator/Translator/OpenAiTranslator.php b/src/Macopedia/OpenAiTranslator/Translator/OpenAiTranslator.php
@@ -17,8 +17,14 @@ public function __construct(
 
     public function translate(string $text, Language $targetLanguageCode): ?string
     {
-        return $this
+        $answer = $this
             ->openAiClient
             ->ask('user', sprintf(self::MESSAGE, $targetLanguageCode->asText(), $text));
+
+        if ($answer !== null) {
+            $answer = preg_replace('/(\(Note.*)/','',$answer);
+        }
+
+        return $answer;
     }
 }