Upload 2 files

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from text_converter import fre_levels, sbert_levels, model_types, generate_similar_sentence, user_input_readability_level, set_reading_levels
+from text_converter import fre_levels, sbert_levels, generate_sim_fre, generate_sim_sbert, fetch_readability_score_fre, fetch_readability_score_sbert
 
 APP_DESCRIPTION = '''# Reading Level Converter
 <div id="content_align">Convert any text to a specified reading level while retaining the core text meaning</div>'''
@@ -8,10 +8,14 @@ MIN_ENTAILMENT = 0.5
 MAX_ITER = 5
 SYSTEM_PROMPT = "You are a writing assistant. You help convert complex texts to simpler texts while maintaining the core meaning of the text."
 
-def convert_text(input_text, grade_level, input_reading_score, is_fre):
-    reading_levels = fre_levels if is_fre else sbert_levels
-    min_level, max_level = reading_levels[grade_level]
-    output_text, similarity, reading_level, message = generate_similar_sentence(input_text, min_level, max_level, MIN_ENTAILMENT, SYSTEM_PROMPT, MAX_ITER, float(input_reading_score), is_fre)
+def convert_text_fre(input_text, grade_level, input_reading_score):
+    min_level, max_level = fre_levels[grade_level]
+    output_text, similarity, reading_level, message = generate_sim_fre(input_text, min_level, max_level, MIN_ENTAILMENT, SYSTEM_PROMPT, MAX_ITER, float(input_reading_score))
+    return output_text, similarity, reading_level, message
+
+def convert_text_sbert(input_text, grade_level, input_reading_score, is_fre):
+    min_level, max_level = sbert_levels[grade_level]
+    output_text, similarity, reading_level, message = generate_sim_sbert(input_text, min_level, max_level, MIN_ENTAILMENT, SYSTEM_PROMPT, MAX_ITER, float(input_reading_score))
     return output_text, similarity, reading_level, message
 
 with gr.Blocks(css='styles.css') as app:
@@ -36,8 +40,8 @@ with gr.Blocks(css='styles.css') as app:
         output_input_reading_level = gr.Textbox(label="Input Text Reading Level", placeholder="Input Text Reading Level...", lines=1, interactive=False)
 
         fetch_score_and_lvl_btn.click(
-            fn=user_input_readability_level,
-            inputs=[input_text, True],
+            fn=fetch_readability_score_fre,
+            inputs=[input_text],
             outputs=[output_input_reading_score, output_input_reading_level]
         )
         
@@ -55,8 +59,8 @@ with gr.Blocks(css='styles.css') as app:
         convert_button = gr.Button("Convert Text")
 
         convert_button.click(
-            fn=convert_text,
-            inputs=[input_text, grade_level_fre, output_input_reading_score, True],
+            fn=convert_text_fre,
+            inputs=[input_text, grade_level_fre, output_input_reading_score],
             outputs=[output_converted_text, output_similarity, output_reading_level, output_message]
         )
     #with gr.Tab("SBERT"):

text_converter.py

@@ -61,6 +61,12 @@ def set_reading_levels(level_type):
     inverse_reading_levels = {v: k for k, v in reading_levels.items()}
     return level_type
 
+def fetch_readability_score_fre(input_text):
+    return user_input_readability_level(input_text, True)
+
+def fetch_readability_score_sbert(input_text):
+    return user_input_readability_level(input_text, False)
+
 def user_input_readability_level(input_text, is_fre):
     current_score = ping_api(input_text, model_types[0] if is_fre else model_types[1])
     print(f'Reading score for user input is: {current_score} for model type: {model_types[0] if is_fre else model_types[1]}')
@@ -72,6 +78,12 @@ def user_input_readability_level(input_text, is_fre):
             break
     return current_score, current_level
 
+def generate_sim_fre(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level):
+    return generate_similar_sentence(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level, True)
+
+def generate_sim_sbert(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level):
+    return generate_similar_sentence(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level, False)
+
 def generate_similar_sentence(input_text, min_reading_level, max_reading_level, min_entailment, system_prompt, max_iter, curr_reading_level, is_fre):
     i = 0
     completed = False