Spaces:
Running
Running
File size: 2,534 Bytes
5abbb8c 3fdcb38 28d0c5f 74a35d9 5abbb8c 74a35d9 28d0c5f 5abbb8c 28d0c5f 3fdcb38 28d0c5f 7b1ddd7 3fdcb38 7b1ddd7 3fdcb38 28d0c5f 3fdcb38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
import json
import webbrowser
from aip_trainer import PROJECT_ROOT_FOLDER, app_logger
from flask import Flask, render_template, request, Response
from flask_cors import CORS
from aip_trainer.lambdas import lambdaGetSample
from aip_trainer.lambdas import lambdaSpeechToScore
from aip_trainer.lambdas import lambdaTTS
app = Flask(__name__, template_folder="static")
cors = CORS(app)
app.config['CORS_HEADERS'] = '*'
rootPath = ''
@app.route(rootPath+'/')
def main():
return render_template('main.html')
@app.route(rootPath+'/getAllSamples')
def getDataDeEnAll():
import pickle
from pathlib import Path
sample_folder = Path(PROJECT_ROOT_FOLDER / "aip_trainer" / "lambdas")
with open(sample_folder / 'data_de_en_2.pickle', 'rb') as handle:
df = pickle.load(handle)
j = df.to_json()
return Response(j, mimetype='application/json')
@app.route(rootPath+'/getSampleSearch', methods=['POST'])
def getDataDeEnSearch():
import pickle
from pathlib import Path
sample_folder = Path(PROJECT_ROOT_FOLDER / "aip_trainer" / "lambdas")
with open(sample_folder / 'data_de_en_2.pickle', 'rb') as handle:
event = request.get_json(force=True)
df = pickle.load(handle)
lang = event.get('language')
filter_key = event.get('search')
df_by_language = df[f"{lang}_sentence"]
filter_obj = df_by_language.str.contains(filter_key)
filtered = df_by_language[filter_obj]
j = filtered.to_json()
return Response(j, mimetype='application/json')
@app.route(rootPath+'/getAudioFromText', methods=['POST'])
def getAudioFromText():
event = {'body': json.dumps(request.get_json(force=True))}
return lambdaTTS.lambda_handler(event, [])
@app.route(rootPath+'/getSample', methods=['POST'])
def getNext():
event = {'body': json.dumps(request.get_json(force=True))}
return lambdaGetSample.lambda_handler(event, [])
@app.route(rootPath+'/GetAccuracyFromRecordedAudio', methods=['POST'])
def GetAccuracyFromRecordedAudio():
try:
event = {'body': json.dumps(request.get_json(force=True))}
lambda_correct_output = lambdaSpeechToScore.lambda_handler(event, [])
return lambda_correct_output
except Exception as e:
import traceback
app_logger.error(e)
app_logger.error(traceback.format_exc())
raise e
if __name__ == "__main__":
language = 'de'
webbrowser.open_new('http://127.0.0.1:3000/')
app.run(host="0.0.0.0", port=3000) # , debug=True)
|