#imageContainer {
            margin-top: 0.5rem;
            text-align: center;
            display: grid;
            grid-template-columns: repeat(2, 1fr);
            grid-template-rows: repeat(2, 1fr);
            gap: 10px;
        }
        .image-section {
            position: relative;
            border-radius: 8px;
            overflow: hidden;
            box-shadow: var(--shadow);
        }
        .generatedImage {
            max-width: 100%;
            height: auto;
            border-radius: 8px;
            object-fit: contain;
            cursor: pointer;
            transition: transform 0.3s ease;
        }
        .generatedImage:hover {
            transform: scale(1.05);
        }
        .loading-overlay, .overlay {
            position: absolute;
            top: 0;
            left: 0;
            width: 100%;
            height: 100%;
            display: flex;
            justify-content: center;
            align-items: center;
            color: white;
            font-size: 1.2rem;
            font-weight: bold;
            text-align: center;
            display: none;
        }
        .loading-overlay {
            background-color: rgba(0, 0, 0, 0.7);
        }
        .overlay {
            background-color: rgba(0, 0, 0, 0.5);
        }
        .download-button {
            width: 80px;
            position: absolute;
            top: 10px;
            right: 10px;
            background: rgb(0 0 0 / 58%);
            color: white;
            border: none;
            border-radius: 6px;
            padding: 4px 8px;
            cursor: pointer;
            box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
            transition: background 0.3s ease;
        }
        .download-button:hover {
            background: rgb(0 0 0 / 88%);
        }
        .download-button:disabled {
            background: rgba(0, 0, 0, 0.3);
            cursor: not-allowed;
        }
        .preview-overlay {
            position: fixed;
            top: 0;
            left: 0;
            width: 100%;
            height: 100%;
            background-color: rgba(0, 0, 0, 0.8);
            display: none;
            justify-content: center;
            align-items: center;
            z-index: 1000;
        }
        .preview-image {
            max-width: 90%;
            max-height: 90%;
            border-radius: 8px;
            box-shadow: var(--shadow);
        }