.attention-container {
            background: white;
            border: 2px solid #ddd;
            border-radius: 0;
            padding: 20px;
            margin-bottom: 20px;
            min-height: 300px;
        }

        .sequence-container {
            display: flex;
            gap: 10px;
            justify-content: center;
            margin: 20px 0;
            flex-wrap: wrap;
        }

        .vertical-sequence-container {
            display: flex;
            flex-direction: column;
            gap: 6px;
            min-width: 150px;
        }

        .token {
            display: flex;
            flex-direction: column;
            align-items: center;
            background: #f0f8ff;
            border: 2px solid #4a90e2;
            border-radius: 0;
            padding: 10px;
            min-width: 80px;
        }

        .vertical-sequence-container .token {
            min-width: 120px;
            align-items: flex-start;
            padding: 4px 8px;
            min-height: 40px;
            display: flex;
            flex-direction: column;
            justify-content: center;
        }


        .token.selected {
            background: #e6f3ff;
            border-color: #2c5aa0;
            box-shadow: 0 0 10px rgba(44, 90, 160, 0.5);
        }

        .token-text {
            font-weight: bold;
            font-size: 12px;
            margin-bottom: 2px;
            color: #2c5aa0;
        }

        .token-embedding {
            font-size: 8px;
            font-family: monospace;
            color: #666;
            text-align: left;
            line-height: 1.1;
        }

        .matrix-container {
            display: flex;
            justify-content: space-around;
            gap: 20px;
            margin: 20px 0;
            flex-wrap: wrap;
        }

        .matrix-section {
            flex: 1;
            min-width: 200px;
            text-align: center;
        }

        .vertical-matrix-section {
            flex: 1;
            text-align: center;
            min-width: 140px;
        }

        .matrix-title {
            font-weight: bold;
            margin-bottom: 10px;
            padding: 5px;
            border-radius: 0;
            color: white;
        }

        .matrix-title.query { background: #4a90e2; }
        .matrix-title.key { background: #7cb342; }
        .matrix-title.value { background: #f44336; }

        .matrix-display {
            border: 1px solid #ddd;
            border-radius: 0;
            background: white;
            min-height: 100px;
            display: flex;
            align-items: center;
            justify-content: center;
            font-family: monospace;
            font-size: 12px;
        }

        .attention-heatmap {
            background: white;
            border: 2px solid #ddd;
            border-radius: 0;
            padding: 20px;
            text-align: center;
        }

        .step-indicator {
            display: flex;
            justify-content: center;
            gap: 10px;
            margin: 20px 0;
        }

        .step {
            padding: 8px 16px;
            border: 2px solid #ddd;
            border-radius: 0;
            background: white;
            font-size: 12px;
            font-weight: bold;
            transition: all 0.2s ease;
        }

        .step.active {
            background: #4a90e2;
            color: white;
            border-color: #4a90e2;
        }

        .step.completed {
            background: #7cb342;
            color: white;
            border-color: #7cb342;
        }

        .math-explanation {
            background: #f8f9fa;
            border: 1px solid #dee2e6;
            border-radius: 0;
            padding: 15px;
            margin: 10px 0;
            font-size: 14px;
        }

        .dimension-controls {
            display: grid;
            grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
            gap: 15px;
            margin-bottom: 20px;
        }

        .weight-matrix-controls {
            background: #f8f9fa;
            border-radius: 0;
            padding: 15px;
            margin: 10px 0;
        }

        .weight-controls-grid {
            display: grid;
            grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
            gap: 10px;
            margin-top: 10px;
        }