Spaces:
Runtime error
Runtime error
| import os | |
| import gradio as gr | |
| import requests | |
| def clamp(x, minimum, maximum): | |
| return max(minimum, min(x, maximum)) | |
| ################################################################################################################################################# | |
| # API calls | |
| ################################################################################################################################################# | |
| # read secret api key | |
| API_KEY = os.getenv('ApiKey') | |
| base_url = "https://skapi.polyglot-edu.com/" | |
| levels = ["Pre K", "Primary", "Middle School", "High School", "Academic"] | |
| def get_level_mapping(level): | |
| return { | |
| "Academic": 0, | |
| "Pre K": 1, | |
| "Primary": 2, | |
| "Middle School": 3, | |
| "High School": 4, | |
| }[level] | |
| def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, number_of_distractors, temperature): | |
| """ | |
| Generate a fill-gaps question from a given text. | |
| Parameters | |
| ---------- | |
| original_text : str | |
| The original text from which to generate the fill-gaps question. | |
| number_of_words : int | |
| The number of words of the generated text. | |
| number_of_gaps : int | |
| The number of gaps to generate. | |
| number_of_distractors : int | |
| The number of distractors to generate for each gap. | |
| temperature : float | |
| The temperature for the generation. | |
| Returns | |
| ------- | |
| str | |
| The fill-gaps question. | |
| """ | |
| match number_of_words: | |
| case "≈ 150 Words": | |
| number_of_words = 150 | |
| case "≈ 250 Words": | |
| number_of_words = 250 | |
| case "≈ 350 Words": | |
| number_of_words = 350 | |
| response = requests.post( | |
| base_url + "FillTheGaps/generateexercise", | |
| headers={ | |
| "ApiKey": API_KEY | |
| }, | |
| json={ | |
| "text": original_text, | |
| "level": get_level_mapping(level), | |
| "n_o_w": int(number_of_words), | |
| "n_o_g": int(number_of_gaps), | |
| "n_o_d": int(number_of_distractors), | |
| "temperature": int(temperature) * 0.2 | |
| }, | |
| timeout=20 | |
| ) | |
| if response.status_code == 200: | |
| return response.text | |
| raise Exception(f"API call failed with status code {response.status_code} and message {response.text}") | |
| def generate_open_question(original_text, level, temperature): | |
| """ | |
| Generate an open question from a given text. | |
| Parameters | |
| ---------- | |
| original_text : str | |
| The original text from which to generate the open question. | |
| temperature : float | |
| The temperature for the generation. | |
| Returns | |
| ------- | |
| str | |
| The open question. | |
| """ | |
| response = requests.post( | |
| base_url + "QuestionExercise/generateexercise", | |
| headers={ | |
| "ApiKey": API_KEY | |
| }, | |
| json={ | |
| "text": original_text, | |
| "level": get_level_mapping(level), | |
| "temperature": int(temperature) * 0.2 | |
| }, | |
| timeout=20 | |
| ) | |
| if response.status_code == 200: | |
| return response.text | |
| raise Exception(f"API call failed with status code {response.status_code} and message {response.text}") | |
| def generate_multiplechoice(original_text, level, number_of_options, number_of_easy_distractors, number_of_distractors, temperature): | |
| """ | |
| Generate a multiple-choice question from a given text. | |
| Parameters | |
| ---------- | |
| original_text : str | |
| The original text from which to generate the multiple-choice question. | |
| number_of_options : int | |
| The number of options to generate. | |
| number_of_easy_distractors : int | |
| The number of easy distractors to generate for each option. | |
| number_of_distractors : int | |
| The number of distractors to generate for each option. | |
| temperature : float | |
| The temperature for the generation. | |
| Returns | |
| ------- | |
| str | |
| The multiple-choice question. | |
| """ | |
| response = requests.post( | |
| base_url + "QuizExercise/generateexercise", | |
| headers={ | |
| "ApiKey": API_KEY | |
| }, | |
| json={ | |
| "text": original_text, | |
| "type": True, | |
| "level": get_level_mapping(level), | |
| "n_o_d": int(number_of_distractors), | |
| "nedd": int(number_of_easy_distractors), | |
| "temperature": int(temperature) * 0.2 | |
| }, | |
| timeout=20 | |
| ) | |
| if response.status_code == 200: | |
| return response.text | |
| raise Exception(f"API call failed with status code {response.status_code} and message {response.text}") | |
| ################################################################################################################################################# | |
| # Interface building | |
| ################################################################################################################################################# | |
| flag_link = "https://www.google.com" | |
| comment_link = "https://www.google.com" | |
| def build_fill_gaps_interface(): | |
| """ | |
| Build the fill-gaps interface. | |
| """ | |
| with gr.Blocks(title="Fill-gaps") as demo: | |
| with gr.Row(): | |
| with gr.Column(scale=5): | |
| input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)") | |
| submit_btn = gr.Button(value="Submit") | |
| with gr.Column(scale=4): | |
| output_text_length = gr.Radio(["≈ 150 Words", "≈ 250 Words", "≈ 350 Words"], label="Output text length", value="≈ 150 Words") | |
| level = gr.Radio(levels, label="Level", value="Primary") | |
| with gr.Row(): | |
| blanks = gr.Number(value=5, minimum=4, maximum=8, step=1, label="Number of blanks") | |
| distractors = gr.Number(value=5, minimum=4, maximum=8, step=1, label="Number of distractors") | |
| temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)") | |
| def update_numeric(output_text_length, blanks, distractors): | |
| if output_text_length == "≈ 150 Words": | |
| min_, max_ = 4, 8 | |
| elif output_text_length == "≈ 250 Words": | |
| min_, max_ = 6, 10 | |
| elif output_text_length == "≈ 350 Words": | |
| min_, max_ = 8, 12 | |
| return ( | |
| gr.Number(value=clamp(blanks, min_, max_), minimum=min_, maximum=max_, label="Number of blanks"), | |
| gr.Number(value=clamp(distractors, 0, blanks), minimum=0, maximum=blanks, label="Number of distractors") | |
| ) | |
| def update_blanks(blanks, distractors): | |
| return gr.Number(value=clamp(distractors, 0, blanks), minimum=0, maximum=blanks, label="Number of distractors") | |
| blanks.change(update_blanks, [blanks, distractors], [distractors]) | |
| output_text_length.change(update_numeric, [output_text_length, blanks, distractors], [blanks, distractors]) | |
| with gr.Row(): | |
| output = gr.TextArea(placeholder="Generated text", label="Output") | |
| with gr.Row() as button_row: | |
| upvote_btn = gr.Button(value="👍 Upvote") | |
| downvote_btn = gr.Button(value="👎 Downvote") | |
| comment_btn = gr.Button(value="💬 Comment", link=comment_link) | |
| flag_btn = gr.Button(value="⚠️ Flag", link=flag_link) | |
| submit_btn.click(generate_fill_gaps, [input_field, level, output_text_length, blanks, distractors, temperature], [output]) | |
| return demo | |
| def build_multiplechoice_interface(): | |
| """ | |
| Build the open question interface. | |
| """ | |
| with gr.Blocks(title="Open Question") as demo: | |
| with gr.Row(): | |
| with gr.Column(scale=5): | |
| input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)") | |
| submit_btn = gr.Button(value="Submit") | |
| with gr.Column(scale=4): | |
| level = gr.Radio(levels, label="Level", value="Primary") | |
| with gr.Row(): | |
| options = gr.Number(value=4, minimum=2, maximum=8, step=1, label="Number of blanks", interactive=False) | |
| easy_distractors = gr.Number(value=1, minimum=0, maximum=8, step=1, label="Number of easy distractors") | |
| distractors = gr.Number(value=1, minimum=0, maximum=8, step=1, label="Number of distractors") | |
| temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)") | |
| def update_options(options, easy_distractors, distractors): | |
| distractors = clamp(distractors, 0, options) | |
| easy_distractors = clamp(easy_distractors, 0, distractors) | |
| return ( | |
| gr.Number(value=easy_distractors, minimum=0, maximum=distractors, label="Number of easy distractors"), | |
| gr.Number(value=distractors, minimum=0, maximum=options, label="Number of distractors") | |
| ) | |
| def update_distractors(easy_distractors, distractors): | |
| easy_distractors = clamp(easy_distractors, 0, distractors) | |
| return gr.Number(value=easy_distractors, minimum=0, maximum=distractors, label="Number of easy distractors") | |
| options.change(update_options, [options, easy_distractors, distractors], [easy_distractors, distractors]) | |
| distractors.change(update_distractors, [easy_distractors, distractors], [easy_distractors]) | |
| with gr.Row(): | |
| output = gr.TextArea(placeholder="Generated text", label="Output") | |
| with gr.Row() as button_row: | |
| upvote_btn = gr.Button(value="👍 Upvote") | |
| downvote_btn = gr.Button(value="👎 Downvote") | |
| comment_btn = gr.Button(value="💬 Comment", link=comment_link) | |
| flag_btn = gr.Button(value="⚠️ Flag", link=flag_link) | |
| submit_btn.click(generate_multiplechoice, [input_field, level, options, easy_distractors, distractors, temperature], [output]) | |
| return demo | |
| def build_open_question_interface(): | |
| """ | |
| Build the multiple-choice interface. | |
| """ | |
| with gr.Blocks(title="Multiple choice") as demo: | |
| with gr.Row(): | |
| with gr.Column(scale=5): | |
| input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)") | |
| submit_btn = gr.Button(value="Submit") | |
| with gr.Column(scale=4): | |
| level = gr.Radio(levels, label="Level", value="Primary") | |
| with gr.Row(): | |
| pass | |
| temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)") | |
| with gr.Row(): | |
| output = gr.TextArea(placeholder="Generated text", label="Output") | |
| with gr.Row() as button_row: | |
| upvote_btn = gr.Button(value="👍 Upvote") | |
| downvote_btn = gr.Button(value="👎 Downvote") | |
| comment_btn = gr.Button(value="💬 Comment", link=comment_link) | |
| flag_btn = gr.Button(value="⚠️ Flag", link=flag_link) | |
| submit_btn.click(generate_open_question, [input_field, level, temperature], [output]) | |
| return demo | |
| def build_demo(): | |
| return gr.TabbedInterface( | |
| [build_fill_gaps_interface(), build_open_question_interface(), build_multiplechoice_interface()], | |
| ["Fill-gaps", "Open question", "Multiple-choice"], | |
| title="Education AI" | |
| ) | |
| if __name__ == "__main__": | |
| build_demo().launch(share=False) | |