|
import gradio as gr |
|
from text_converter import generate_similar_sentence |
|
from text_converter import user_input_readability_level |
|
from text_converter import 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>''' |
|
|
|
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." |
|
|
|
|
|
""" |
|
reading_levels = { |
|
"5th Grade (90-100)": (90, 100), |
|
"6th Grade (80-90)": (80, 90), |
|
"7th Grade (70-80)": (70, 80), |
|
"8th - 9th Grade (60-70)": (60, 70), |
|
"10th - 12th Grade (50-60)": (50, 60), |
|
"College (30-50)": (30, 50), |
|
"College Graduate + Professionals (0-30)": (0, 30) |
|
} |
|
""" |
|
|
|
def convert_text(input_text, grade_level, input_reading_score): |
|
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)) |
|
return output_text, similarity, reading_level, message |
|
|
|
with gr.Blocks(css='styles.css') as app: |
|
gr.Markdown(APP_DESCRIPTION) |
|
|
|
with gr.Tab("Reading Level Calculator"): |
|
input_text = gr.Textbox(label="Input Text", placeholder="Type here...", lines=4) |
|
|
|
fetch_score_and_lvl_btn = gr.Button("Fetch Score and Level") |
|
|
|
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], |
|
outputs=[output_input_reading_score, output_input_reading_level] |
|
) |
|
|
|
grade_level = gr.Radio(choices=list(reading_levels.keys()), label="Target Reading Level", value=list(reading_levels.keys())[0]) |
|
|
|
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], |
|
outputs=[output_converted_text, output_similarity, output_reading_level, output_message] |
|
) |
|
|
|
|
|
if __name__ == '__main__': |
|
app.launch() |