main.py

"""
Attribution: https://github.com/AIPI540/AIPI540-Deep-Learning-Applications/

Jon Reifschneider
Brinnae Bent 

"""

import streamlit as st
from PIL import Image
import numpy as np
import os
import numpy as np
import pandas as pd
import pandas as pd
import os
import json
import pandas as pd
import torch
import numpy as np
import pandas as pd
import torch.nn as nn
import torch.nn.functional as F
import matplotlib.pyplot as plt


    
if __name__ == '__main__':  
    
    st.header('Spotify Playlists')
    
    img1, img2 = st.columns(2)

    music_notes = Image.open('assets/music_notes.png')
    img1.image(music_notes, use_column_width=True)

    trumpet = Image.open('assets/trumpet.png')
    img2.image(trumpet, use_column_width=True)
    
    with st.sidebar:
        playlist_name = st.selectbox(
            "Playlist Selection",
            (   list(set([1,2]))            )
        )
    
    col1, col2 = st.columns(2)
    with col1:
        st.write(f'Artist')
    with col2:
        st.write(f'Album')