|
from typing import * |
|
import gradio as gr |
|
from model.common import Grader |
|
from model.IELTS_essay_scoring import IELTS_essay_scoring |
|
from model.Engessay_grading_ML import Engessay_grading_ML |
|
from model.mistral_7b_ielts_evaluator import Mistral_7b_IELTS_Evaluator |
|
|
|
models: Dict[str, Grader] = { |
|
"IELTS_essay_scoring": IELTS_essay_scoring(), |
|
"Engessay_grading_ML": Engessay_grading_ML(), |
|
"mistral_7b_ielts_evaluator": Mistral_7b_IELTS_Evaluator(), |
|
} |
|
|
|
|
|
|
|
def grade(question: str, answer: str, model: str) -> Tuple[float, str]: |
|
if len(question) < 30 or len(answer) < 30: |
|
raise gr.Error("Please enter more than 30 characters") |
|
|
|
if model not in models: |
|
raise gr.Error(f"Model {model} not found") |
|
|
|
grader = models[model] |
|
return grader.grade(question, answer) |
|
|
|
|
|
with gr.Blocks() as app: |
|
gr.Markdown("# Essay Scoring") |
|
|
|
with gr.Row(): |
|
with gr.Column(): |
|
question = gr.Textbox( |
|
label="Enter the question here", |
|
placeholder="Write the question here", |
|
lines=3, |
|
) |
|
essay = gr.Textbox( |
|
label="Enter your essay here", |
|
placeholder="Write your essay here", |
|
lines=10, |
|
) |
|
model = gr.Radio( |
|
label="Select the grading model", |
|
choices=list(models.keys()), |
|
value=list(models.keys())[0], |
|
) |
|
btn = gr.Button("Grade Essay", variant="primary") |
|
with gr.Column(): |
|
overall = gr.Number(label="Overall Score") |
|
comment = gr.Markdown(label="Comment") |
|
|
|
btn.click( |
|
fn=grade, |
|
inputs=[question, essay, model], |
|
outputs=[overall, comment], |
|
) |
|
|
|
gr.Examples( |
|
[ |
|
[ |
|
"It is important for all towns and cities to have large public spaces such as squares and parks. Do you agree or disagree with this statement?", |
|
( |
|
"It is crucial for all metropolitan cities and towns to " |
|
"have some recreational facilities like parks and squares because of their numerous benefits. A number of " |
|
"arguments surround my opinion, and I will discuss it in upcoming paragraphs. To commence with, the first " |
|
"and the foremost merit is that it is beneficial for the health of people because in morning time they can " |
|
"go for walking as well as in the evenings, also older people can spend their free time with their loved ones, " |
|
"and they can discuss about their daily happenings. In addition, young people do lot of exercise in parks and " |
|
"gardens to keep their health fit and healthy, otherwise if there is no park they glue with electronic gadgets " |
|
"like mobile phones and computers and many more. Furthermore, little children get best place to play, they play " |
|
"with their friends in parks if any garden or square is not available for kids then they use roads and streets " |
|
"for playing it can lead to serious incidents. Moreover, parks have some educational value too, in schools, " |
|
"students learn about environment protection in their studies and teachers can take their pupils to parks because " |
|
"students can see those pictures so lively which they see in their school books and they know about importance " |
|
"and protection of trees and flowers. In recapitulate, parks holds immense importance regarding education, health " |
|
"for people of every society, so government should build parks in every city and town." |
|
), |
|
], |
|
], |
|
inputs=[question, essay], |
|
) |
|
|
|
info = "\n\n".join([f"### {k}\n{grader.info()}" for k, grader in models.items()]) |
|
gr.Markdown(info) |
|
|
|
app.launch() |
|
|
