import toml
import os
import secrets

CONFIG_FILE = 'config.toml'

def load_or_create_config():
    if not os.path.exists(CONFIG_FILE):
        admin_password = secrets.token_urlsafe(16)
        config = {
            'admin': {
                'password': admin_password
            },
            'directories': {
                'upload': 'uploads',
                'thumbnail': 'thumbnails'
            }
        }
        with open(CONFIG_FILE, 'w') as f:
            toml.dump(config, f)
        print(f"Generated new config file with admin password: {admin_password}")
    else:
        with open(CONFIG_FILE, 'r') as f:
            config = toml.load(f)
    return config