import gradio as gr

from utils.state import State
from utils.process_input import get_response
from utils.examples import examples


def clear_fields():
    app_state.clear_all()
    return None, None, None


def generate_response(instructions, input_mail):
    try:
        model = app_state.model
        response = get_response(
            model=model, instructions=instructions, input_texts=input_mail
        )

        return response
    except Exception as e:
        gr.Error(e)


with gr.Blocks() as demo:
    gr.Markdown("# Phi3 Mini 4k Model Fine-tuned")
    gr.Markdown(
        "**NOTE**: Please note that the response generated by the model might not be accurate. Always verify results."
    )
    with gr.Row(equal_height=True):
        with gr.Column():
            instructions = gr.TextArea(label="System Instructions", max_lines=2)
            input_mail = gr.TextArea(label="Input Email", max_lines=6)
            with gr.Row():
                submit_button = gr.Button("Submit", variant="primary")
                clear_button = gr.Button("Clear")

        with gr.Column():
            with gr.Row():
                extracted_fields = gr.TextArea(label="Extracted Fields")

    gr.Examples(examples=examples, inputs=[instructions, input_mail])

    clear_button.click(
        fn=clear_fields,
        outputs=[instructions, input_mail, extracted_fields],
    )
    submit_button.click(
        fn=generate_response,
        inputs=[instructions, input_mail],
        outputs=[extracted_fields],
    )

if __name__ == "__main__":
    try:
        app_state = State()
        demo.launch()
    except Exception as e:
        gr.Error(e)