from flask import Flask, request, jsonify
from transformers import pipeline
import firebase_admin
from firebase_admin import credentials, firestore
import os

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

app = Flask(__name__)

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

# Thiết lập Firebase
cred = credentials.Certificate('whisper-1da8a-firebase-adminsdk-1bfxq-fe43b4015d.json')
firebase_admin.initialize_app(cred)
db = firestore.client()

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

    # Upload transcription lên Firebase
    doc_ref = db.collection('transcriptions').add({'text': transcription})

    return jsonify({'transcription': transcription})

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