German

SynonymEditor.py

@@ -7,22 +7,33 @@ nltk.download('punkt')
 
 class SynonymEditor:
 
-    def __init__(self, api_key, model_engine, max_tokens, temperature):
+    def __init__(self, api_key, model_engine, max_tokens, temperature, language):
         openai.api_key = api_key
         self.model_engine = model_engine
         self.max_tokens = max_tokens
         self.temperature = temperature
+        self.language = language
 
     # Play with the prompts here and change the return index to change and see the effect of the prompt on the output quality
     # Note that the longer the prompt, higher the token used and hence the billing
     def _get_prompt(self, sentence, few_shots):
         if (few_shots):
-            prompt = "Replace exactly one word with a synonym while preserving the overall sentence structure and meaning.\n" + \
-                few_shots + "\nInput:" + sentence + " Output:"
+            if (self.language == "de"):
+                prompt = 'Modernisiere den Text. Behalte die Inhalte von Klammern "" bei.\n' + \
+                    few_shots + "\nEingang:" + sentence + " Ausgang:"
+            else:
+                prompt = "Replace exactly one word with a synonym while preserving the overall sentence structure and meaning.\n" + \
+                    few_shots + "\nInput:" + sentence + " Output:"
         elif "__QUOTE__" in sentence:
-            prompt = "Replace exactly one word with a synonym while preserving __QUOTE__ in the following sentence:\n"+sentence+"\n"
+            if (self.language == "de"):
+                prompt = 'Modernisiere den Text. Behalte die Inhalte von Klammern "" bei.\n'+sentence+'\n'
+            else:
+                prompt = "Replace exactly one word with a synonym while preserving __QUOTE__ in the following sentence:\n"+sentence+"\n"
         else:
-            prompt = "Replace exactly one word with a synonym in the following sentence:\n"+sentence+"\n"
+            if (self.language == "de"):
+                prompt = 'Modernisiere den Text. Behalte die Inhalte von Klammern "" bei.\n'+sentence+'\n'
+            else:
+                prompt = "Replace exactly one word with a synonym in the following sentence:\n"+sentence+"\n"
         return prompt
 
     # Call the OpenAI API here

app.py

@@ -1,11 +1,14 @@
 import gradio as gr
 from SynonymEditor import SynonymEditor
 
+LANGS = ["de", "en"]
 
-def replace_synonyms(api_key, text, use_few_shots, few_shots, temperature):
+
+def replace_synonyms(api_key, language, text, use_few_shots, few_shots, temperature):
     model_engine = "text-davinci-003"
     max_tokens = 500
-    editor = SynonymEditor(api_key, model_engine, max_tokens, temperature)
+    editor = SynonymEditor(api_key, model_engine,
+                           max_tokens, temperature, language)
     if (use_few_shots):
         return editor._edit_text(text, few_shots)
     else:
@@ -15,6 +18,9 @@ def replace_synonyms(api_key, text, use_few_shots, few_shots, temperature):
 api_key = gr.inputs.Textbox(label="OpenAI API Key", lines=1,
                             default="YOUR_API_KEY_HERE")
 
+language = gr.inputs.Dropdown(
+    label="Language", choices=LANGS, default="en")
+
 use_few_shots = gr.inputs.Checkbox(label="Use Few Shots", default=True)
 
 few_shots = gr.inputs.Textbox(
@@ -28,6 +34,6 @@ temperature_slider = gr.inputs.Slider(
 
 output_text = gr.outputs.Textbox(label="Output Text")
 
-io = gr.Interface(fn=replace_synonyms, inputs=[api_key, input_text, use_few_shots, few_shots, temperature_slider], outputs=output_text, title="Synonym Replacer",
+io = gr.Interface(fn=replace_synonyms, inputs=[api_key, language, input_text, use_few_shots, few_shots, temperature_slider], outputs=output_text, title="Synonym Replacer",
                   description="Replace words in a text with their synonyms.")
 io.launch()