feat: add slider for adequacy and fluency

app.py

@@ -2,9 +2,15 @@ import gradio as gr
 from frasaria.frasaria import paraphrase_text
 
 demo = gr.Interface(
-    fn=paraphrase_text,
-    inputs=gr.Textbox(lines=5, placeholder=None, label='Text'),
-    outputs=gr.Textbox(label=None)
+    paraphrase_text,
+    [
+        gr.Slider(0.1, 0.99, value=0.90, label="Adequacy",
+                  info='Adequacy menunjukkan sejauh mana kalimat yang dihasilkan memiliki makna yang sama dengan kalimat asli. Kurangi nilai ini untuk mendapatkan kalimat yang lebih berbeda.'),
+        gr.Slider(0.1, 0.99, value=0.90, label="Fluency",
+                  info='Fluency menunjukkan sejauh mana kalimat yang dihasilkan terdengar alami.'),
+        gr.Textbox(lines=5, label="Text Input")
+    ],
+    'textbox'
 )
 
 if __name__ == '__main__':

frasaria/frasaria.py

@@ -8,26 +8,19 @@ def split_sentences(text: str):
     return [sentence.strip() for sentence in text.split(".")]
 
 
-def paraphrase(phrases):
-    para_phrases = parrot.augment(phrases, max_return_phrases=10, do_diverse=True)
-    max_score = 0
-    paraphrased = para_phrases[0][0]
-    for para_phrase in para_phrases:
-        score = para_phrase[-1]
-        if score > max_score:
-            max_score = score
-            paraphrased = para_phrase[0]
-    return paraphrased
-
-
-def paraphrase_text(phrase):
+def paraphrase(adequacy, fluency, phrases):
+    para_phrases = parrot.augment(phrases, max_return_phrases=1, adequacy_threshold=adequacy, fluency_threshold=fluency)
+    return para_phrases[0][0] if para_phrases else phrases
+
+
+def paraphrase_text(adequacy, fluency, phrase):
     phrase = phrase.replace('"', "'")
     result = translator.translate_text(phrase, target_lang="EN-US")
 
     # paraphrase
     sentences = []
     for translated in split_sentences(result.text)[:-1]:
-        sentences.append(paraphrase(translated))
+        sentences.append(paraphrase(adequacy, fluency, translated))
 
     text = '. '.join(list(map(lambda x: x.capitalize(), sentences))) + '.'
     if result.detected_source_lang == 'EN':