julius_baer_onboarding/app.py

import logging

from flask import Flask, request
from flask_cors import cross_origin

import config
from dto.requests import GameStartRequestDTO, GameDecisionRequestDTO
from dto.responses import GameStartResponseWithBotDecisionDTO, GameDecisionResponseWithBotDecisionDTO
from services.julius_baer_api_client import JuliusBaerApiClient
from services.advisor import Advisor
from utils.storage.game_files_manager import store_game_round_data

app = Flask(__name__)
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - [%(module)s] - %(message)s')
jb_client = JuliusBaerApiClient()


@app.route('/new-game', methods=['POST'])
@cross_origin()  # allow all origins all methods
def new_game():
    game_start_request = GameStartRequestDTO(player_name=config.API_TEAM)
    res = jb_client.start_game(game_start_request)
    bot_decision = Advisor().make_decision(res.client_data)

    res_with_bot_decision = GameStartResponseWithBotDecisionDTO(
        message=res.message,
        session_id=res.session_id,
        player_id=res.player_id,
        client_id=res.client_id,
        client_data=res.client_data,
        score=res.score,
        bot_decision=bot_decision.answer,
        bot_reason=bot_decision.reason,
    )

    store_game_round_data(bot_decision.answer, res_with_bot_decision, res.score, str(res.session_id), 'active')

    return res_with_bot_decision.model_dump_json()


@app.route('/next', methods=['POST'])
@cross_origin()  # allow all origins all methods
def next_client():
    body = request.get_json()

    decision = body.get("decision")
    client_id = body.get("client_id")
    session_id = body.get("session_id")

    make_decision_request = GameDecisionRequestDTO(decision=decision, client_id=client_id, session_id=session_id)
    res = jb_client.send_decision(make_decision_request)
    bot_decision = Advisor().make_decision(res.client_data)

    res_with_bot_decision = GameDecisionResponseWithBotDecisionDTO(
        status=res.status,
        score=res.score,
        client_id=res.client_id,
        client_data=res.client_data,
        bot_decision=bot_decision.answer,
        bot_reason=bot_decision.reason,
    )

    store_game_round_data(bot_decision.answer, res_with_bot_decision, res.score, str(session_id), res.status)

    return res_with_bot_decision.model_dump_json()


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