from flask import Flask, request, Response, json
from cat_detection import detect_cat
from language_label_mapper import translate
from validator import validate

""" 
    Flask Rest API application to cat recognition.
    If request is valid then send response with results of recognition.
    If key named 'Image' in body does not occurred then returns 400 (BAD REQUEST).
    Otherwise returns 200 with results of recognition.
    Format of response:
        {
            "lang": {users_lang},
            "results": {
                {filename}: {
                    "isCat": {is_cat},
                    "results": {
                        "1": {result}
                        "2": {result}
                        "3": {result}
                        ...
                        "10" {result}
                    }
                },
                ...
            },
            errors[
                {error_message},
                {error_message},
                ...
            ]
        }
    To see result format -> cat_detection.py
"""


# Define flask app
app = Flask(__name__)
app.secret_key = 'secret_key'

# Available cats
list_of_labels = [
    'lynx',
    'lion',
    'tiger',
    'cheetah',
    'leopard',
    'jaguar',
    'tabby',
    'Egyptian_cat',
    'cougar',
    'Persian_cat',
    'Siamese_cat',
    'snow_leopard',
    'tiger_cat'
]

# Available languages
languages = {'pl', 'en'}


@app.route('/api/v1/detect-cat', methods=['POST'])
def upload_file():
    # Validate request
    error_messages = validate(request)

    # If any errors occurred, return 400 (BAD REQUEST)
    if len(error_messages) > 0:
        errors = json.dumps(
            {
                'errors': error_messages
            }
        )
        return Response(errors, status=400, mimetype='application/json')

    # Get files from request
    files = request.files.getlist('image')

    # Get user's language (Value in header 'Accept-Language'). Default value is English
    lang = request.accept_languages.best_match(languages, default='en')

    # Define JSON structure for results
    results = {
        'lang': lang,
        'results': {},
        'errors': []
    }

    # Generate results
    for file in files:
        predictions = detect_cat(file, list_of_labels)
        if predictions is not None:
            predictions, error_messages = translate(predictions, lang)
        results['results'][file.filename] = {
            'isCat': False if not predictions else True,
            **({'predictions': predictions} if predictions is not None else {})
        }
        if len(error_messages) > 1:
            results['errors'].append(error_messages)

    # Send response with 200 (Success)
    return Response(json.dumps(results), status=200, mimetype='application/json')


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