import gradio as gr
from utils import ChemicalConverter, validate_smiles2iupac, plot_mol, login

def convert(chemical_name, style, validate, plot):
    # Initialize the ChemicalConverter
    converter = ChemicalConverter(mode="SMILES2IUPAC")
    converted_name = ""
    validation_score = ""
    plot_image = None
    style_prefix = "<" + style[:4] + ">"
    converted_name = converter.convert(style_prefix + chemical_name)
    if validate:
        validation_score = validate_smiles2iupac(chemical_name, converted_name)
    if plot:
        plot_image = plot_mol(chemical_name)
    return converted_name, validation_score, plot_image

smiles2iupac = gr.Interface(
    fn=convert,
    allow_flagging='auto',
    inputs=[
        gr.Textbox(label="Enter your SMILES name", placeholder="Enter your SMILES name here"),
        gr.Radio(
            choices=["BASE", "SYSTEMATIC", "TRADITIONAL"],
            label="Choose desired IUPAC style",
        ),
        gr.Checkbox(label="Validate with molecular similarity", value=False),
        gr.Checkbox(label="Plot molecule", value=True)
    ],
    outputs=[gr.Text(label="Converted Name"),
             gr.Text(label="Input-Target similarity score"),
             gr.Image(type='pil', label="Molecule Plot", height=170, width=890)],
    examples=[
        ["CCO", "BASE", True, False]
    ],
)