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,
    }
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()


st.set_page_config(
    page_title='Norwegian Bokmaal to Nynorsk',
    page_icon='translator-icon.png',
    layout='wide',
    initial_sidebar_state='expanded'
)
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",
    default="Biblioteker er flotte!"
)

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