Upload 2 files

app.py

@@ -17,15 +17,9 @@ with gr.Blocks(css='styles.css') as app:
     gr.Markdown(APP_DESCRIPTION)
 
     with gr.Tab("Reading Level Calculator"):
-        option_reading_level = {}
-        model_select = gr.Radio(choices=model_types, label="Readability Score Model", value=model_types[1], interactive=True, scale=2)
-        model_select_btn = gr.Button("Select Readability Score Model", scale=1)
-        
-        model_select_btn.click(
-            fn=set_reading_levels,
-            inputs=[model_select],
-            outputs=[option_reading_level]
-        )
+        with gr.Row():
+            model_select = gr.Radio(choices=model_types, label="Readability Score Model", value=model_types[1], interactive=True, scale=2)
+            model_select_btn = gr.Button("Select Readability Score Model", scale=1)
 
         input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4)
 
@@ -40,7 +34,13 @@ with gr.Blocks(css='styles.css') as app:
             outputs=[output_input_reading_score, output_input_reading_level]
         )
         
-        grade_level = gr.Radio(choices=list(option_reading_level.keys()), label="Target Reading Level", value=list(option_reading_level.keys())[0], interactive=True)
+        grade_level = gr.Radio(label="Target Reading Level", interactive=True)
+
+        model_select_btn.click(
+            fn=set_reading_levels,
+            inputs=[model_select],
+            outputs=[grade_level]
+        )
         
         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)

text_converter.py

@@ -62,7 +62,7 @@ def set_reading_levels(level_type):
         reading_levels = sbert_levels
     inverse_reading_levels = {v: k for k, v in reading_levels.items()}
     model_type = level_type
-    return reading_levels
+    return reading_levels.keys()
 
 def user_input_readability_level(input_text):
     current_score = ping_api(input_text, model_type) #flesch_reading_ease(input_text)