from flask import Flask
from flask_restx import Api, Resource, fields
from werkzeug.datastructures import FileStorage
import wav2vec2 as wave

flask_app = Flask(__name__)
app = Api(app = flask_app,
          version="1.0",
          title="Model deployment test",
          description="Test your models and feel rad")

upload_parser = app.parser()
upload_parser.add_argument('file',
                           location='files',
                           type=FileStorage)


model = app.model('Soundfile',
		  {'name': fields.String(required = True,
					 description="The soudnfile you want to decode",
					 help="File is necessary")})

@app.route('/transcribe', methods=['POST'])
@app.expect(upload_parser)
class Transcribe(Resource):
    def post(self):
        args = upload_parser.parse_args()
        file = args.get('file')
        text = wave.predict(file)
        return text