File size: 2,992 Bytes
67d6834 9e39004 a04f574 67d6834 e7dc5d0 67d6834 9e39004 a04f574 9e39004 67d6834 988475c 67d6834 71e7460 |
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 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import os
import tempfile
import json
from flask import Flask, request, jsonify
from werkzeug.utils import secure_filename
from preprocess import process_audio_file
from pause import annotate_pauses
from repetition import annotate_repetitions
from syllable import annotate_syllables
from fillerword import annotate_fillerwords
from morpheme import annotate_morpheme
from morpheme_omission import annotate_morpheme_omission
from annotation import annotate_transcript
app = Flask(__name__)
@app.route('/process_old', methods=['POST'])
def process_audio_old():
data = request.get_json()
if not data or 'input_audio_file' not in data:
return jsonify({'error': 'Missing input_audio_file parameter'}), 400
input_audio_file = data['input_audio_file']
device = data.get('device', 'cuda')
pause_threshold = data.get('pause_threshold', 0.5)
num_speakers = data.get('num_speakers', 2)
app.logger.info(f"Processing audio file: {input_audio_file}")
session_id = process_audio_file(input_audio_file, num_speakers=num_speakers, device=device)
annotate_pauses(session_id, pause_threshold)
annotate_repetitions(session_id)
annotate_syllables(session_id)
annotate_fillerwords(session_id)
output_annotation = annotate_transcript(session_id)
result = {
'session_id': session_id,
'annotation_result': output_annotation
}
return jsonify(result), 200
@app.route('/process', methods=['POST'])
def process_audio():
if 'audio_file' not in request.files:
return jsonify({'error': 'Missing audio file '}), 400
audio_file = request.files['audio_file']
filename = secure_filename(audio_file.filename)
suffix = os.path.splitext(filename)[1] or '.wav'
with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
audio_path = tmp.name
audio_file.save(audio_path)
device = request.form.get('device', 'cuda')
pause_threshold = float(request.form.get('pause_threshold', 0.5))
num_speakers = int(request.form.get('num_speakers', 2))
app.logger.info(f"Processing uploaded audio: {audio_path}")
session_id = process_audio_file(audio_path, num_speakers=num_speakers, device=device)
annotate_pauses(session_id, pause_threshold)
annotate_repetitions(session_id)
# annotate_syllables(session_id)
annotate_fillerwords(session_id)
# annotate_transcript(session_id)
annotate_morpheme(session_id)
annotate_morpheme_omission(session_id)
json_path = f"session_data/{session_id}/{session_id}_transcriptionCW.json"
if not os.path.isfile(json_path):
return jsonify({'error': f"Annotation file {json_path} not found"}), 500
with open(json_path, 'r', encoding='utf-8') as f:
transcription = json.load(f)
try:
os.remove(audio_path)
except OSError:
pass
return jsonify(transcription), 200
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860, debug=True)
|