import streamlit as st
import requests
import os

# Установка API URL и заголовков
API_URL_tra = "https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-en-ru"
headers = {"Authorization": os.getenv("api_token")}


# Функция для перевода слова
def translate_word(payload):    
	response = requests.post(API_URL_tra, headers=headers, json=payload)
	return response.json()

# Настраеваем заголовок и название страницы
st.set_page_config(layout="wide", page_title="Students' Personal Assistant")
st.markdown(' # :female-student: Персональный помощник для студентов')
st.divider()
st.markdown('## :robot_face: Переводчик ')

col1, col2 = st.columns(2)
translated = ''
with col1:
    words_from_tarea = st.text_area('English', value='deep learning')
    
    traslate_btn = st.button("Перевести")
with col2:
    if traslate_btn:
         translated = translate_word({"inputs": words_from_tarea})
         translated = translated[0]['translation_text']
    st.text_area('Перевод', value=translated)