.ai-builder-image {
    max-width: 100%;
    height: auto;
    border-radius: 8px;
    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
}

.wp-block-ai-builder-image-ia {
    margin: 1em 0;
    position: relative;
}

.wp-block-ai-builder-image-ia .ai-builder-image {
    max-width: 100%;
    height: auto;
    border-radius: 8px;
    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
    display: block;
}

.wp-block-ai-builder-image-ia .image-preview {
    position: relative;
    display: inline-block;
}

.wp-block-ai-builder-image-ia .remove-image {
    position: absolute;
    top: 10px;
    right: 10px;
    background: rgba(255, 255, 255, 0.9) !important;
}

.wp-block-ai-builder-image-ia .image-controls {
    display: flex;
    flex-direction: column;
    gap: 10px;
    padding: 20px;
    background: #f8f9fa;
    border-radius: 8px;
    text-align: center;
}

.wp-block-ai-builder-image-ia .generate-button {
    width: 100%;
    justify-content: center;
}

.wp-block-ai-builder-image-ia .error-message {
    color: #d63638;
    margin-top: 10px;
    padding: 10px;
    background: #f8d7da;
    border-radius: 4px;
}

@media (min-width: 600px) {
    .wp-block-ai-builder-image-ia .image-controls {
        flex-direction: row;
        justify-content: center;
    }
    
    .wp-block-ai-builder-image-ia .generate-button {
        width: auto;
    }
}

.wp-block-ai-builder-image-ia .components-button {
    margin: 5px;
}