#!/bin/bash

pip install kaggle --upgrade

echo $1
echo $2

kaggle datasets download -d brunoalercon123/top-200-spotify-songs-dataset -p ./dataset -u $1 -p $2

unzip -o ./dataset/top-200-spotify-songs-dataset.zip -d ./dataset

shuf Spotify-200-Songs.csv -o shuffled_spotify.csv

head -n 100 shuffled_spotify.csv > subset1.csv
tail -n 100 shuffled_spotify.csv > subset2.csv

cut -d ',' -f 1,2,3 shuffled_spotify.csv > trimmed_spotify.csv

cut -d ',' -f 1,2,4,5,6 shuffled_spotify.csv > processed_spotify.csv

echo "Shuffled dataset:" > results.txt
head shuffled_spotify.csv >> results.txt
echo "" >> results.txt

echo "Subset 1:" >> results.txt
head subset1.csv >> results.txt
echo "" >> results.txt

echo "Subset 2:" >> results.txt
head subset2.csv >> results.txt
echo "" >> results.txt

echo "Trimmed dataset:" >> results.txt
head trimmed_spotify.csv >> results.txt
echo "" >> results.txt

echo "Processed dataset:" >> results.txt
head processed_spotify.csv >> results.txt
echo "" >> results.txt

mkdir -p artifacts
mv shuffled_spotify.csv subset1.csv subset2.csv trimmed_spotify.csv processed_spotify.csv results.txt artifacts/