add auth to speetch to text server

This commit is contained in:
2023-01-04 23:39:40 +01:00
parent 12202b1baf
commit f00c4c2e23
+11 -1
View File
@@ -5,15 +5,25 @@ from flask import Flask
from flask import jsonify from flask import jsonify
from flask import request from flask import request
from flask_cors import CORS, cross_origin from flask_cors import CORS, cross_origin
from flask_httpauth import HTTPTokenAuth
from src.workers.SpeechToText import SpeechToText from src.workers.SpeechToText import SpeechToText
app = Flask(__name__) app = Flask(__name__)
cors = CORS(app) cors = CORS(app)
app.config['CORS_HEADERS'] = 'Content-Type' app.config['CORS_HEADERS'] = 'Content-Type'
auth = HTTPTokenAuth(scheme='Bearer')
secret_token = "97e36f7e-340e-4c02-b329-9415faee38c3"
@auth.verify_token
def verify_token(token):
return token == secret_token
@app.route('/', methods=['POST']) @app.route('/', methods=['POST'])
@auth.login_required
@cross_origin() @cross_origin()
def post(): def post():
file_name = str(uuid.uuid4()) + ".wav" file_name = str(uuid.uuid4()) + ".wav"
@@ -30,4 +40,4 @@ def post():
if __name__ == '__main__': if __name__ == '__main__':
app.run(host='0.0.0.0', port=8000, debug=True) app.run(host='0.0.0.0', port=8000, debug=False)