import gradio as gr
from dotenv import load_dotenv

from mcq import mcq_generation
from lesson_plan import lesson_plan
from worksheet import worksheet_generation
from content import content_generation

load_dotenv()

with gr.Blocks() as app:
    gr.Markdown("# API Demo")
    with gr.Tabs():
        with gr.TabItem("MCQ Generation"):
            mcq_generation()
        with gr.TabItem("Lesson Plan Generation"):
            lesson_plan()
        with gr.TabItem("Worksheet Generation"):
            worksheet_generation()
        with gr.TabItem("Content Generation"):
            content_generation()

app.launch()