"""Definition of function to generate a new randomly initialized model and save
it in Minio datastore."""

from pathlib import Path

from dotenv import load_dotenv
from torchinfo import summary

from model.src.models import VisualCommunicationModel
from shared.datastore import connect_minio, put_model
from shared.utils import setup_logging

if __name__ == '__main__':
    # load in env file
    env_path = Path(__file__).parent.parent / 'server.env'
    assert env_path.exists()
    load_dotenv(env_path)
    # setup logging
    setup_logging()
    # connect to minio
    client = connect_minio()
    # instantiate model
    model = VisualCommunicationModel(download_resnet_weights=True)
    # show model weights
    summary(model)
    # put buffer in minio bucket
    hash_str = put_model(
        client=client,
        model=model,
    )
    print(f"hash string: {hash_str}")
    print('saved to Minio')