try {
    # Import admin credentials
    $serverAddress = Get-Content -Path ".\DefaultServer.txt"
    $credential = Import-Clixml -Path .\AdminCredential.xml

    # Establish session with Exchange Server
    Write-Host "Connecting to Exchange server $serverAddress..."
    $session = New-PSSession -ConfigurationName Microsoft.Exchange -ConnectionUri "http://$serverAddress/PowerShell/" -Authentication Kerberos -Credential $credential
    Import-PSSession $session -DisableNameChecking -WarningAction SilentlyContinue -ErrorAction Stop
    Write-Host "Successfully connected."

    # Define output file path in the user's Downloads folder
    $outputFile = "$env:USERPROFILE\Downloads\ExchangeDatabaseMailboxListReport.html"

    # --- Data Collection Variables ---
    $totalUserMailboxes = 0
    $totalSharedMailboxes = 0
    $databaseStats = @() # Array to hold stats for the column chart

    # --- HTML Report Setup ---
    # Define CSS styles and include Chart.js library via CDN
    $head = @"
<style>
body { font-family: Arial, sans-serif; font-size: 10pt; }
h1 { color: #336699; font-size: 18pt; text-align: center; } /* Center H1 */
/* MODIFIED: Center H2 */
h2 { color: #336699; font-size: 14pt; margin-top: 20px; border-bottom: 1px solid #cccccc; text-align: center; padding-bottom: 5px; }
h3 { color: #5588bb; font-size: 12pt; margin-top: 15px; text-align: center; }
table { border-collapse: collapse; width: 80%; margin: 10px auto 15px auto; }
th, td { border: 1px solid #cccccc; padding: 5px; text-align: left; }
th { background-color: #f2f2f2; font-weight: bold; }
/* MODIFIED: Center dbinfo, remove padding */
.dbinfo { margin-bottom: 10px; text-align: center; padding-left: 0; }
.note { font-style: italic; color: #666666; margin-bottom: 20px; text-align: center; }
.nodata { color: #888888; font-style: italic; text-align:center; } /* Center nodata text */

/* MODIFIED: Wrapper for side-by-side charts, ensure stretch alignment */
.charts-wrapper {
    display: flex;
    flex-wrap: wrap;
    justify-content: space-around;
    align-items: stretch; /* Default, but explicit. Tries to make containers equal height */
    width: 95%;
    margin: 20px auto;
}

/* Chart container styling */
.chart-container {
    border: 1px solid #ddd;
    border-radius: 5px;
    padding: 15px;
    box-sizing: border-box;
    display: flex; /* Use flexbox internally too */
    flex-direction: column; /* Stack title and canvas */
    align-items: center; /* Center title and canvas */
}

/* Specific styles for chart containers INSIDE the wrapper */
.charts-wrapper .chart-container {
    width: 48%;
    margin-bottom: 20px;
}

/* Ensure canvas is responsive within its container */
.chart-container canvas {
    max-width: 100%;
    height: auto !important;
}

/* Media Query for stacking charts on smaller screens */
@media (max-width: 800px) {
    .charts-wrapper {
        flex-direction: column;
        align-items: center;
    }
    .charts-wrapper .chart-container {
        width: 90%; /* Take up more width when stacked */
    }
    table { width: 95%; }
}
</style>
<script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
"@

    # Start building the HTML body content
    $bodyContent = "<h1>Exchange Database Mailbox Report</h1>"
    $bodyContent += "<p class='note'>Generated on: $(Get-Date)</p>"
    $bodyContent += "<p class='note'>Note: System mailboxes such as 'Discovery Search Mailbox' have been excluded from this report.</p>"

    # Get database size and mailboxes
    Write-Host "Retrieving database information..."
    $databases = Get-MailboxDatabase -Status | Select-Object Name, DatabaseSize
    Write-Host "Found $($databases.Count) databases."

    # --- Process Each Database ---
    # (Loop remains the same)
    foreach ($db in $databases) {
        Write-Host "Processing database: $($db.Name)"
        $dbName = $db.Name
        $dbSize = $db.DatabaseSize -replace '\s*\(.*\)', '' # Clean up size string

        # Add database details to the HTML body (these will now be centered by CSS)
        $bodyContent += "<h2>Database: $dbName</h2>"
        $bodyContent += "<p class='dbinfo'>Total Size: $dbSize</p>"

        # Retrieve mailboxes under the database and exclude "Discovery Search Mailbox"
        Write-Host "  Retrieving mailboxes for $dbName..."
        $mailboxes = Get-Mailbox -Database $dbName | Where-Object {$_.DisplayName -ne "Discovery Search Mailbox"} | Select-Object DisplayName, PrimarySmtpAddress, RecipientTypeDetails

        # Separate user mailboxes and shared mailboxes
        $userMailboxes = $mailboxes | Where-Object {$_.RecipientTypeDetails -eq "UserMailbox"} | Sort-Object DisplayName | Select-Object DisplayName, PrimarySmtpAddress
        $sharedMailboxes = $mailboxes | Where-Object {$_.RecipientTypeDetails -eq "SharedMailbox"} | Sort-Object DisplayName | Select-Object DisplayName, PrimarySmtpAddress

        # Accumulate Counts
        $currentUserCount = $userMailboxes.Count
        $currentSharedCount = $sharedMailboxes.Count
        $totalUserMailboxes += $currentUserCount
        $totalSharedMailboxes += $currentSharedCount
        $databaseStats += [PSCustomObject]@{
            DatabaseName = $dbName
            UserCount    = $currentUserCount
            SharedCount  = $currentSharedCount
        }

        # Add User Mailboxes Table
        $bodyContent += "<h3>User Mailboxes</h3>"
        if ($userMailboxes) {
            $bodyContent += $userMailboxes | ConvertTo-Html -As Table -Fragment -PreContent "<p style='text-align: center; margin-top:0;'>Count: $currentUserCount</p>"
        } else {
            $bodyContent += "<p class='nodata'>No user mailboxes found in this database.</p>"
        }

        # Add Shared Mailboxes Table
        $bodyContent += "<h3>Shared Mailboxes</h3>"
        if ($sharedMailboxes) {
            $bodyContent += $sharedMailboxes | ConvertTo-Html -As Table -Fragment -PreContent "<p style='text-align: center; margin-top:0;'>Count: $currentSharedCount</p>"
        } else {
            $bodyContent += "<p class='nodata'>No shared mailboxes found in this database.</p>"
        }
         Write-Host "  Finished processing $dbName."
    } # End foreach database

    # --- Add Chart Placeholders and Script ---
    $bodyContent += "<hr style='margin-top: 40px;'>"
    $bodyContent += "<h2 style='text-align: center;'>Mailbox Count Summary Charts</h2>"

    # Start the charts wrapper div
    $bodyContent += "<div class='charts-wrapper'>"

    # Chart 1: Pie Chart Container
    $bodyContent += "<div class='chart-container'>"
    $bodyContent += "<h3>Total Mailbox Distribution</h3>"
    $bodyContent += "<canvas id='mailboxPieChart'></canvas>"
    $bodyContent += "</div>"

    # Chart 2: Column Chart Container
    $bodyContent += "<div class='chart-container'>"
    $bodyContent += "<h3>Mailbox Counts per Database</h3>"
    $bodyContent += "<canvas id='mailboxColumnChart'></canvas>"
    $bodyContent += "</div>"

    # End the charts wrapper div
    $bodyContent += "</div>"

    # --- Prepare Data for JavaScript ---
    # (This part remains the same)
    $dbLabels = $databaseStats.DatabaseName | ForEach-Object { "'$_'" }
    $jsDbLabels = "[$($dbLabels -join ', ')]"
    $jsUserCounts = "[$($databaseStats.UserCount -join ', ')]"
    $jsSharedCounts = "[$($databaseStats.SharedCount -join ', ')]"

    # --- Embed JavaScript for Chart Generation ---
    # MODIFIED: Added aspectRatio to chart options
    $chartScript = @"
<script>
  // Data from PowerShell
  const totalUserData = $totalUserMailboxes;
  const totalSharedData = $totalSharedMailboxes;
  const dbLabels = $jsDbLabels;
  const dbUserData = $jsUserCounts;
  const dbSharedData = $jsSharedCounts;

  // Pie Chart Configuration
  const pieCtx = document.getElementById('mailboxPieChart').getContext('2d');
  const mailboxPieChart = new Chart(pieCtx, {
    type: 'pie',
    data: {
      labels: ['User Mailboxes (' + totalUserData + ')', 'Shared Mailboxes (' + totalSharedData + ')'],
      datasets: [{
        label: 'Total Mailbox Count',
        data: [totalUserData, totalSharedData],
        backgroundColor: [ 'rgba(54, 162, 235, 0.7)', 'rgba(255, 159, 64, 0.7)' ],
        borderColor: [ 'rgba(54, 162, 235, 1)', 'rgba(255, 159, 64, 1)' ],
        borderWidth: 1
      }]
    },
    options: {
      responsive: true,
      aspectRatio: 1.5, // MODIFIED: Suggest aspect ratio (Width:Height)
      plugins: {
        legend: { position: 'bottom' },
        title: { display: false }
      }
    }
  });

  // Column Chart Configuration
  const columnCtx = document.getElementById('mailboxColumnChart').getContext('2d');
  const mailboxColumnChart = new Chart(columnCtx, {
    type: 'bar',
    data: {
      labels: dbLabels,
      datasets: [
        {
          label: 'User Mailboxes',
          data: dbUserData,
          backgroundColor: 'rgba(54, 162, 235, 0.7)',
          borderColor: 'rgba(54, 162, 235, 1)',
          borderWidth: 1
        },
        {
          label: 'Shared Mailboxes',
          data: dbSharedData,
          backgroundColor: 'rgba(255, 159, 64, 0.7)',
          borderColor: 'rgba(255, 159, 64, 1)',
          borderWidth: 1
        }
      ]
    },
    options: {
      responsive: true,
      aspectRatio: 1.5, // MODIFIED: Suggest aspect ratio (Width:Height) - Same as Pie
      plugins: {
        legend: { position: 'top' },
        title: { display: false }
      },
      scales: {
        y: { beginAtZero: true, title: { display: true, text: 'Number of Mailboxes' } },
        x: { title: { display: true, text: 'Database Name' } }
      }
    }
  });
</script>
"@

    # Append the chart script to the body content
    $bodyContent += $chartScript

    # --- Finalize and Save HTML Report ---
    Write-Host "Generating HTML report with centered info and balanced charts..."
    ConvertTo-Html -Head $head -Body $bodyContent -Title "Exchange Database Mailbox Report" | Out-File $outputFile -Encoding UTF8

    Write-Host "Report saved to $outputFile" -ForegroundColor Green
    Invoke-Item -Path $outputFile

    # Clean up session
    Write-Host "Removing PSSession..."
    Remove-PSSession $session
    Write-Host "Script finished."

} catch {
    Write-Host "An error occurred:" -ForegroundColor Red
    Write-Host $_.Exception.Message -ForegroundColor Red
    Write-Host "Script failed."
    if ($PSBoundParameters.ContainsKey('session') -and $session -ne $null -and (Get-PSSession | Where-Object {$_.ID -eq $session.ID})) {
        Write-Host "Attempting to remove PSSession after error..."
        Remove-PSSession $session
    }
}