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
+from text_converter import fre_levels, sbert_levels, model_types, generate_similar_sentence, user_input_readability_level, set_reading_levels
 
 APP_DESCRIPTION = '''# Reading Level Converter
 <div id="content_align">Convert any text to a specified reading level while retaining the core text meaning</div>'''
@@ -21,36 +21,14 @@ with gr.Blocks(css='styles.css') as app:
     gr.Markdown(APP_DESCRIPTION)
 
     with gr.Tab("FRE"):
-        with gr.Row():
-            input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4, scale = 2)
-            fetch_score_and_lvl_btn = gr.Button("Fetch Score and Level", scale = 1)
-
-        output_input_reading_score = gr.Textbox(label="Input Text Reading Score", placeholder="Input Text Reading Score...", lines=1)
-        output_input_reading_level = gr.Textbox(label="Input Text Reading Level", placeholder="Input Text Reading Level...", lines=1)
-
-        fetch_score_and_lvl_btn.click(
-            fn=user_input_readability_level,
-            inputs=[input_text, "FRE"],
-            outputs=[output_input_reading_score, output_input_reading_level]
-        )
-        
-        grade_level = gr.Radio(choices=list(fre_levels.keys()), label="Target Reading Level", value = list(fre_levels.keys())[0], interactive=True)
-        
-        output_reading_level = gr.Textbox(label="Output Reading Level", placeholder="Output Reading Level...", lines=1)
-        output_similarity = gr.Textbox(label="Similarity", placeholder="Similarity Score...", lines=1)
-        output_converted_text = gr.Textbox(label="Converted Text", placeholder="Results will appear here...", lines=4)
-                    
-        output_message = gr.Textbox(label="Message", placeholder="System Message...", lines=2)
-        
-        convert_button = gr.Button("Convert Text")
-
-        convert_button.click(
-            fn=convert_text,
-            inputs=[input_text, grade_level, output_input_reading_score, "FRE"],
-            outputs=[output_converted_text, output_similarity, output_reading_level, output_message]
+        #with gr.Row():
+        readability_model = gr.Radio(choices=model_types, label="Readability Model", value = model_types[0], interactive=True)
+        readability_model.click(
+            fn=set_reading_levels,
+            inputs=[readability_model],
+            outputs=[]
         )
 
-    with gr.Tab("SBERT"):
         with gr.Row():
             input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4, scale = 2)
             fetch_score_and_lvl_btn = gr.Button("Fetch Score and Level", scale = 1)
@@ -60,11 +38,11 @@ with gr.Blocks(css='styles.css') as app:
 
         fetch_score_and_lvl_btn.click(
             fn=user_input_readability_level,
-            inputs=[input_text, "SBERT"],
+            inputs=[input_text, readability_model],
             outputs=[output_input_reading_score, output_input_reading_level]
         )
         
-        grade_level = gr.Radio(choices=list(sbert_levels.keys()), label="Target Reading Level", value = list(sbert_levels.keys())[0], interactive=True)
+        grade_level = gr.Radio(choices=list(fre_levels.keys()) if readability_model == model_types[0] else list(sbert_levels.keys()), label="Target Reading Level", interactive=True)
         
         output_reading_level = gr.Textbox(label="Output Reading Level", placeholder="Output Reading Level...", lines=1)
         output_similarity = gr.Textbox(label="Similarity", placeholder="Similarity Score...", lines=1)
@@ -76,7 +54,7 @@ with gr.Blocks(css='styles.css') as app:
 
         convert_button.click(
             fn=convert_text,
-            inputs=[input_text, grade_level, output_input_reading_score, "SBERT"],
+            inputs=[input_text, grade_level, output_input_reading_score, readability_model],
             outputs=[output_converted_text, output_similarity, output_reading_level, output_message]
         )
 

text_converter.py

@@ -61,7 +61,6 @@ def set_reading_levels(level_type):
     inverse_reading_levels = {v: k for k, v in reading_levels.items()}
 
 def user_input_readability_level(input_text, model_type):
-    set_reading_levels(model_type)
     current_score = ping_api(input_text, model_type)
     print(f'Reading score for user input is: {current_score} for model type: {model_type}')
     current_level = ''