File size: 2,958 Bytes
127f052
7b28cbc
95db9c2
9c37f19
127f052
7b28cbc
 
 
 
 
 
 
 
9c37f19
7b28cbc
 
 
 
 
 
 
 
 
9c37f19
7b28cbc
c378e49
7b28cbc
053b008
9d6764e
7b28cbc
672ad31
 
 
 
 
c378e49
 
 
 
 
 
 
 
7d82239
2efb0d5
c378e49
 
672ad31
 
 
 
 
 
 
 
 
 
 
 
c378e49
 
672ad31
7b28cbc
 
 
672ad31
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
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."

# Dictionary mapping grade levels to reading ease scores
"""
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()