#!/usr/bin/env python3
"""
Create a fresh simplified version of the 2020 CSV with only Rank, Artist, and Album columns.
"""

import csv

def main():
    # Read the 2020 CSV and extract only needed columns
    simplified_albums = []
    
    with open('rolling_stone_top_500_albums_2020.csv', 'r', encoding='utf-8') as file:
        reader = csv.DictReader(file)
        for row in reader:
            simplified_albums.append({
                'Rank': row['Rank'],
                'Artist': row['Artist'],
                'Album': row['Album']
            })
    
    # Write simplified CSV
    with open('rolling_stone_2020_simple.csv', 'w', newline='', encoding='utf-8') as file:
        fieldnames = ['Rank', 'Artist', 'Album']
        writer = csv.DictWriter(file, fieldnames=fieldnames)
        writer.writeheader()
        writer.writerows(simplified_albums)
    
    print(f"✅ Created fresh simplified 2020 CSV with {len(simplified_albums)} albums")

if __name__ == "__main__":
    main()