import os

import requests
import streamlit as st


API_URL = "https://api-inference.huggingface.co/models/andrek/nb2nn"

def translate(text):
    headers = {"Authorization": f"Bearer {os.environ['BEARER']}"}
    payload  = {
        "inputs": text,
    }
    try:
        response = requests.post(API_URL, headers=headers, json=payload)
        json_response = response.json()[0]
        if "error" in json_response:
            return f"Model still loading: {json_response}"
        elif "translation_text" in json_response:
            return json_response["translation_text"]
    except Exception as exc:
        return f"Oops, something went wrong: {exc}"


st.set_page_config(
    page_title='Norwegian Bokmaal to Nynorsk',
    page_icon='translator-icon.png',
)
st.title("Language Translator from Norwegian Bokmaal to Nynorsk :balloon:")
text = st.text_area("Enter text:",
    height=None,
    max_chars=None,
    key=None,
    help="Enter your text here",
)

if st.button('Translate'):
    if str(text).strip() == "":
        st.warning('Please **enter text** for translation')
    else:
        st.info(str(translate(text)))