optimized

SynonymEditor.py

@@ -15,19 +15,21 @@ class SynonymEditor:
 
     # 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):
-        if "__QUOTE__" in sentence:
-            return "Replace exactly one word with a synonym while preserving __QUOTE__ in the following sentence:\n"+sentence+"\n"
+    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:"
+        elif "__QUOTE__" in sentence:
+            prompt = "Replace exactly one word with a synonym while preserving __QUOTE__ in the following sentence:\n"+sentence+"\n"
         else:
-            return "Replace exactly one word with a synonym in the following sentence:\n"+sentence+"\n"
+            prompt = "Replace exactly one word with a synonym in the following sentence:\n"+sentence+"\n"
+        return prompt
 
     # Call the OpenAI API here
 
     def __call_ai(self, sentence, few_shots):
-        if (few_shots):
-            prompt = few_shots + "\nInput:" + sentence + " Output:"
-        else:
-            prompt = self._get_prompt(sentence)
+        prompt = self._get_prompt(sentence, few_shots)
+        print(prompt)
         response = openai.Completion.create(
             model=self.model_engine,
             prompt=prompt,
@@ -40,8 +42,9 @@ class SynonymEditor:
         return self._post_process_sentence(response.choices[0].text.strip())
 
     # Split the paragraph to preserve quotation marks
-    def _split_into_sentences(self, text):
-        text = text.replace('"', '__QUOTE__')
+    def _split_into_sentences(self, text, few_shots):
+        if (few_shots == False):
+            text = text.replace('"', '__QUOTE__')
         text = re.sub(r'\s+', ' ', text)
         text = text.strip()
         sentences = sent_tokenize(text)
@@ -56,7 +59,7 @@ class SynonymEditor:
         paragraphs = text.split("\n\n")
         edited_paragraphs = []
         for paragraph in paragraphs:
-            sentences = self._split_into_sentences(paragraph)
+            sentences = self._split_into_sentences(paragraph, few_shots)
             edited_sentences = []
             for sentence in sentences:
                 new_sentence = self.__call_ai(sentence, few_shots)