lambda_function.py

import bf2spl
import json
from writers.default_writer import RandomWriter

def lambda_handler(event, context):
    """

    :param event: schema is
    {
        'body': string,
        'ai_mode': boolean?
    }
    :param context:
    :return:
    """
    body = event.get('body', None)
    ai_mode = event.get('ai_mode', False)
    if not body:
        return json.dumps({
            'statusCode': 400,
            'body': 'No body provided'
        })
    if ai_mode:
        from writers.ai_writer import ChatGptWriter, NullAIResponseLogger
        writer = ChatGptWriter(logger=NullAIResponseLogger())
    else:
        writer = RandomWriter()
    try:
        play = bf2spl.Play(body, writer=writer)
        return {
            'statusCode': 200,
            'body': play.render_spl()
        }
    except Exception as e:
        return json.dumps({
            'error': str(e)
        })