import gradio as gr

# Student Information
My_info = "Student ID: 6319250G, Name: Aung Hlaing Tun"

# Gradio UI Setup
with gr.Blocks() as interface:
    gr.Markdown("📩 Development of an AI Ticket Classifier Model Using DistilBERT")
    gr.Markdown(f"*{My_info}*")

    gr.Markdown(
        """
        **🔍 About this App**
        - This system predicts the appropriate **team assignment** and **contact email** based on the ticket description.
        - Simply enter up to **5 ticket descriptions**, and the AI will classify them accordingly.
        """
    )

    with gr.Row():
        with gr.Column():
            input1 = gr.Textbox(lines=2, placeholder="Enter ticket description 1...", label="Ticket 1")
            output_team1 = gr.Textbox(label="Predicted Team 1")
            output_email1 = gr.Textbox(label="Predicted Email 1")
        with gr.Column():
            input2 = gr.Textbox(lines=2, placeholder="Enter ticket description 2...", label="Ticket 2")
            output_team2 = gr.Textbox(label="Predicted Team 2")
            output_email2 = gr.Textbox(label="Predicted Email 2")
        with gr.Column():
            input3 = gr.Textbox(lines=2, placeholder="Enter ticket description 3...", label="Ticket 3")
            output_team3 = gr.Textbox(label="Predicted Team 3")
            output_email3 = gr.Textbox(label="Predicted Email 3")
        with gr.Column():
            input4 = gr.Textbox(lines=2, placeholder="Enter ticket description 4...", label="Ticket 4")
            output_team4 = gr.Textbox(label="Predicted Team 4")
            output_email4 = gr.Textbox(label="Predicted Email 4")
        with gr.Column():
            input5 = gr.Textbox(lines=2, placeholder="Enter ticket description 5...", label="Ticket 5")
            output_team5 = gr.Textbox(label="Predicted Team 5")
            output_email5 = gr.Textbox(label="Predicted Email 5")
        with gr.Column():
            input6 = gr.Textbox(lines=2, placeholder="Enter ticket description 6...", label="Ticket 6")
            output_team6 = gr.Textbox(label="Predicted Team 6")
            output_email6 = gr.Textbox(label="Predicted Email 6")

    with gr.Row():
        btn1 = gr.Button("Predict for Ticket 1")
        btn2 = gr.Button("Predict for Ticket 2")
        btn3 = gr.Button("Predict for Ticket 3")
        btn4 = gr.Button("Predict for Ticket 4")
        btn5 = gr.Button("Predict for Ticket 5")
        btn6 = gr.Button("Predict for Ticket 6")

interface.launch(share=True)