import gradio as gr
from ps4_models.classifiers import *
from ps4_data.get_embeddings import generate_embedings


def pred(residue_seq):
    generate_embedings(residue_seq)
    model = PS4_Mega()
    return "Hello " + residue_seq + "!!"


iface = gr.Interface(fn=pred, title="Protein Secondary Structure Prediction with PS4-Mega",
                     inputs="text", outputs="text", examples=[
    ["HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA"],
    ["AHKLFIGGLPNYLNDDQVKELLTSFGPLKAFNLVKDSATGLSKGYAFCEYVDINVTDQAIAGLNGMQLGDKKLLVQRASVGAKNA"]
])
iface.launch()