.search-container {
    position: relative; /* This is crucial for positioning the button */
    width: 433px;
    margin: 1.5rem auto;
}

.search-input {
    width: 100%;
    padding: 1rem 1.5rem;
    font-size: 1.2rem;
    border: none;
    border-radius: 30px;
    background-color: #000;
    color: #E0E0E0;
    outline: none;
    transition: box-shadow 0.3s ease;
    /* Make space for the button on the right */
    padding-right: 130px; /* A bit more padding */
}

.search-input:focus {
    box-shadow: 0 0 0 3px #BB86FC;
}

.ai-button {
    position: absolute;
    top: 50%;
    right: 10px; /* Position on the right inside the container */
    transform: translateY(-50%);
    background-color: #555;
    border: none;
    border-radius: 20px; /* This makes it pill-shaped */
    color: #202124;
    cursor: pointer;
    font-size: 14px;
    font-weight: 500;
    padding: 8px 16px; /* Adjust padding for a good size */
    width: auto;
    white-space: nowrap;
}