import os
from flask import Flask, render_template, request
import random
from transformers import pipeline

#Cachen kanssa oli ongelmia, tällä lähti toimimaan
os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'

#Helsinki-NLP-malli
pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fi")

app = Flask(__name__)

#Flask-api GET
@app.route('/')
def home():
   return render_template('index.html')

#Flast-api POST
@app.route('/submit', methods=['POST'])
def submit():
   kaannos_pyynto = request.form["translation_request" ]

   #Testi alussa
   #random_number = random.randint(0,100)


    #Malli toisti puutaheinäää kun sille lähetti tyhjän merkkijonon.... 
   if len(kaannos_pyynto) > 0: 
       kaannos = pipe(kaannos_pyynto)
       return kaannos[0]['translation_text']
   else:
       return ""

if __name__ == '__main__':
   app.run(debug=True)