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)