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(Navitas).html"

    # --- Data Collection Variables ---
    $totalUserMailboxes = 0
    $totalSharedMailboxes = 0
    $databaseStats = @() # Array to hold stats for the column chart
    # Array to hold detailed info for sorting and report generation
    $allDatabaseDetails = @()

    # --- HTML Report Setup ---
    # Define CSS styles and include Chart.js library via CDN
    # MODIFIED: Added fixed table layout and column width definitions
    $head = @"
<style>
body { font-family: Arial, sans-serif; font-size: 10pt; }
h1 { color: #336699; font-size: 18pt; text-align: center; }
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%; /* Keep overall table width */
    margin: 10px auto 15px auto;
    /* NEW: Add table-layout fixed to better enforce column widths */
    table-layout: fixed;
}
th, td {
    border: 1px solid #cccccc;
    padding: 5px;
    text-align: left;
    /* NEW: Ensure long text wraps within the defined column width */
    overflow-wrap: break-word;
    word-wrap: break-word; /* Older browser fallback */
}
th { background-color: #f2f2f2; font-weight: bold; }

/* NEW: Define specific column widths for the 2-column mailbox tables */
/* Target first column (DisplayName) */
th:nth-child(1), td:nth-child(1) {
    width: 40%; /* Adjust percentage as needed */
}
/* Target second column (PrimarySmtpAddress) */
th:nth-child(2), td:nth-child(2) {
    width: 60%; /* Adjust percentage as needed */
}

.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; }
.charts-wrapper {
    display: flex;
    flex-wrap: wrap;
    justify-content: space-around;
    align-items: stretch;
    width: 95%;
    margin: 20px auto;
}
.chart-container {
    border: 1px solid #ddd;
    border-radius: 5px;
    padding: 15px;
    box-sizing: border-box;
    display: flex;
    flex-direction: column;
    align-items: center;
}
.charts-wrapper .chart-container {
    width: 48%;
    margin-bottom: 20px;
}
.chart-container canvas {
    max-width: 100%;
    height: auto !important;
}
@media (max-width: 800px) {
    .charts-wrapper {
        flex-direction: column;
        align-items: center;
    }
    .charts-wrapper .chart-container {
        width: 90%;
    }
    table { width: 95%; } /* Adjust table width for smaller screens too */
}
</style>
<script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
"@

    # --- Collect Data from All Databases First ---
    Write-Host "Retrieving database information..."
    $databases = Get-MailboxDatabase -Status | Select-Object Name, DatabaseSize
    Write-Host "Found $($databases.Count) databases. Processing mailbox data..."

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

        # Retrieve mailboxes under the database and exclude "Discovery Search Mailbox"
        $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

        # Get Counts for this DB
        $currentUserCount = $userMailboxes.Count
        $currentSharedCount = $sharedMailboxes.Count # Corrected simple assignment

        # Accumulate Totals (for Pie Chart)
        $totalUserMailboxes += $currentUserCount
        $totalSharedMailboxes += $currentSharedCount

        # Store stats for Column Chart
        $databaseStats += [PSCustomObject]@{
            DatabaseName = $dbName
            UserCount    = $currentUserCount
            SharedCount  = $currentSharedCount
        }

        # Store detailed info for report generation and sorting
        $allDatabaseDetails += [PSCustomObject]@{
            Name            = $dbName
            Size            = $dbSize
            UserMailboxes   = $userMailboxes   # Store the actual collection
            SharedMailboxes = $sharedMailboxes # Store the actual collection
            UserCount       = $currentUserCount
            SharedCount     = $currentSharedCount
            HasMailboxes    = ($currentUserCount + $currentSharedCount) -gt 0
        }
         Write-Host "    Found $currentUserCount user mailboxes, $currentSharedCount shared mailboxes."
    } # End foreach database data collection

    # --- Start Building HTML Body ---
    $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>"

    # --- Generate HTML for Databases WITH Mailboxes ---
    $databasesWithMailboxes = $allDatabaseDetails | Where-Object { $_.HasMailboxes } | Sort-Object Name
    if ($databasesWithMailboxes) {
        Write-Host "Generating report section for databases with mailboxes..."
        $bodyContent += "<h2 style='border-bottom: 2px solid #336699;'>Databases with Mailboxes</h2>" # Distinct separator

        foreach ($dbInfo in $databasesWithMailboxes) {
            $bodyContent += "<h2>Database: $($dbInfo.Name)</h2>"
            $bodyContent += "<p class='dbinfo'>Total Size: $($dbInfo.Size)</p>"

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

            # Add Shared Mailboxes Table
            $bodyContent += "<h3>Shared Mailboxes</h3>"
            if ($dbInfo.SharedMailboxes) {
                $bodyContent += $dbInfo.SharedMailboxes | ConvertTo-Html -As Table -Fragment -PreContent "<p style='text-align: center; margin-top:0;'>Count: $($dbInfo.SharedCount)</p>"
            } else {
                $bodyContent += "<p class='nodata'>No shared mailboxes found in this database.</p>"
            }
        }
    } else {
        Write-Host "No databases found with user or shared mailboxes."
        $bodyContent += "<h2>Databases with Mailboxes</h2>"
        $bodyContent += "<p class='nodata'>No databases found containing user or shared mailboxes (excluding system mailboxes).</p>"
    }

    # --- Generate HTML for Databases WITHOUT Mailboxes ---
    $databasesWithoutMailboxes = $allDatabaseDetails | Where-Object { -not $_.HasMailboxes } | Sort-Object Name
    if ($databasesWithoutMailboxes) {
        Write-Host "Generating report section for databases without mailboxes..."
        $bodyContent += "<h2 style='margin-top: 40px; border-bottom: 2px solid #336699;'>Databases without Mailboxes</h2>" # Distinct separator

        foreach ($dbInfo in $databasesWithoutMailboxes) {
            $bodyContent += "<h2>Database: $($dbInfo.Name)</h2>"
            $bodyContent += "<p class='dbinfo'>Total Size: $($dbInfo.Size)</p>"
            $bodyContent += "<p class='nodata'>No user or shared mailboxes found in this database.</p>" # Simplified message
        }
    } else {
         Write-Host "All databases have mailboxes or no empty databases were found."
    }


    # --- Add Summary Chart Placeholders and Script ---
    $bodyContent += "<hr style='margin-top: 40px; border-top: 2px solid #336699;'>"
    $bodyContent += "<h2 style='text-align: center;'>Mailbox Count Summary Charts</h2>"
    $bodyContent += "<p class='note'>These charts summarize mailbox counts across ALL databases included in this report.</p>" # Clarifying note

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

    # Chart 1: Pie Chart Container (Overall Totals)
    $bodyContent += "<div class='chart-container'>"
    $bodyContent += "<h3>Total Mailbox Distribution (All Databases)</h3>" # Clarified title
    $bodyContent += "<canvas id='mailboxPieChart'></canvas>"
    $bodyContent += "</div>"

    # Chart 2: Column Chart Container (Per Database Comparison)
    $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 ---
    # Sort databaseStats by name to match the column chart labels with potential report sorting
    $databaseStats = $databaseStats | Sort-Object DatabaseName

    $dbLabels = $databaseStats.DatabaseName | ForEach-Object { "'$_'" }
    $jsDbLabels = "[$($dbLabels -join ', ')]"
    $jsUserCounts = "[$($databaseStats.UserCount -join ', ')]"
    $jsSharedCounts = "[$($databaseStats.SharedCount -join ', ')]"

    # --- Embed JavaScript for Chart Generation ---
    # (JavaScript code is unchanged as it correctly generates summary charts)
    $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,
      plugins: {
        legend: { position: 'bottom' },
        title: { display: false } // Title is handled by H3 above canvas
      }
    }
  });

  // 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,
      plugins: {
        legend: { position: 'top' },
        title: { display: false } // Title is handled by H3 above canvas
      },
      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 reordered sections, summary charts, and fixed table columns..."
    ConvertTo-Html -Head $head -Body $bodyContent -Title "Exchange Database Mailbox Report" | Out-File $outputFile -Encoding UTF8

    Write-Host "Report saved to $outputFile" -ForegroundColor Green
    # Optional: Automatically open the report
    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
    # Optional: More detailed error info
    # Write-Host $_.ScriptStackTrace
    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
    }
}