import gradio as gr
from src.essay_evaluation import EssayEvaluation
from src.logger import logger

demo = gr.Interface(
    title="Essay Evaluation Assistant 📋",
    description="""Master English with our AI Essay Assistant!\n
                Elevate your writing skills or prepare for IELTS and TOEFL. Get precise feedback to polish your skills and conquer every challenge!""",
    fn = EssayEvaluation.evaluate,
    inputs = [
       gr.Radio(["IELTS", "TOEFL", "General"], label="Essay type", info="What type of essay do you have?"),
       gr.Textbox(label="Essay topic (optional)", placeholder="Enter your essay topic here...", container=True),
       gr.Textbox(label="Essay text", placeholder="Enter your essay here...", container=True)
    ],
    outputs = [
        gr.Textbox(label="Results & Comments")
    ],
    allow_flagging="never"
)

if __name__ == "__main__":
    demo.launch()