from flask import Flask, request, jsonify
from flask_cors import CORS, cross_origin
from transformers import pipeline

import os

# Kiểm tra biến môi trường TRANSFORMERS_CACHE
print(f"TRANSFORMERS_CACHE: {os.getenv('TRANSFORMERS_CACHE')}")

app = Flask(__name__)

CORS(app)
app.config['CORS_HEADERS'] = 'Content-type'

# Tải mô hình Hugging Face
transcriber = pipeline(model="Phong1807/whisper-datn11")

@app.route('/transcribe', methods=['POST'])
@cross_origin(origin='*')
def transcribe():
    audio_data = request.files['audio']
    transcription = transcriber(audio_data)['text']

    return jsonify({'transcription': transcription})

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860)