import streamlit as st
import torch
from model import (DistilBertTokenizerFast, TARGET_IND2LABEL, DistilBERTClassifier,
                    load_model, load_tokenizer,
                    get_preds, get_verdict)


st.markdown("## Hello, my dear friend!")
st.markdown("### This service can classify article topic by it's title and abstract")
st.markdown("##### You can specify only title or only abstract, just leave the field empty, but it can work worse in that case.")

title = st.text_area("Title:")
abstract = st.text_area("Abstract:")

device = 'cpu'
tokenizer = load_tokenizer()
model = load_model(device)

text = title + abstract
if (not text):
    verdict = "Both fields are empty"
else:
    verdict = get_verdict(get_preds(text, model, tokenizer, device))

st.markdown("#### Verdict:")
st.markdown(verdict)