Spaces:
Runtime error
Runtime error
| # importing libraries | |
| import gradio as gr | |
| import pickle | |
| import pandas as pd | |
| from huggingface_hub import hf_hub_download | |
| import json | |
| import requests | |
| pd.options.mode.chained_assignment = None # Turn off SettingWithCopyWarning | |
| songs_df = pd.read_csv( | |
| hf_hub_download('damilojohn/Personal_Playlist_Generator', | |
| repo_type='dataset', | |
| filename='spotify_transformed.csv')) | |
| mappings = pd.read_csv( | |
| hf_hub_download('damilojohn/Personal_Playlist_Generator', | |
| repo_type='dataset', | |
| filename='song_mappings.csv')) | |
| verses_df = pd.read_csv( | |
| hf_hub_download('damilojohn/Personal_Playlist_Generator', | |
| repo_type='dataset', | |
| filename='verses.csv')) | |
| song_embeddings = pickle.load( | |
| open(hf_hub_download('damilojohn/Personal_Playlist_Generator', | |
| repo_type='dataset', | |
| filename='embeddings.pkl'), 'rb')) | |
| verses_df.rename(columns={'0': 'verse'}, inplace=True) | |
| mappings.rename(columns={'Unnamed: 0': 'verse', '0': 'song_name'}, | |
| inplace=True) | |
| def generate_playlist(prompt): | |
| payload = {'prompt': prompt} | |
| response = requests.request('POST', | |
| url='https://xi5j0hwh1a.execute-api.eu-west-2.amazonaws.com/test/huh', | |
| data=json.dumps(payload)).json() | |
| hits = response['hits'] | |
| hits = pd.DataFrame.from_dict(hits[0]) | |
| verses_match = verses_df.iloc[hits['corpus_id']] | |
| songs_match = mappings[mappings['verse'].isin( | |
| verses_match['verse'].values)] | |
| songs_match = songs_df[songs_df['song_name'].isin( | |
| songs_match['song_name'].values)] | |
| songs_match = songs_match.sort_values('song_name') | |
| songs_match = songs_match.drop_duplicates(subset='song_name') | |
| songs_name = list(songs_match['song_name'][:9]) | |
| cover_art = list(songs_match['image'][:9]) | |
| images = [gr.Image.update(value=art, visible=True) for art in cover_art] | |
| return (gr.Radio.update(label='songs', interactive=True, | |
| choices=songs_name), | |
| *images) | |
| def set_example_prompt(examples): | |
| return gr.TextArea.update(value=examples[0]) | |
| def create_frontend(): | |
| demo = gr.Blocks() | |
| with demo: | |
| gr.Markdown( | |
| ''' | |
| # A Text based playlist Generator for Afrobeats | |
| ''' | |
| ) | |
| with gr.Row(): | |
| with gr.Column(): | |
| gr.Markdown( | |
| ''' | |
| Enter words describing your playlist | |
| ''' | |
| ) | |
| song_prompt = gr.TextArea( | |
| value='', | |
| placeholder=" Enter a sentence that describes how you're feeling or what you want your playlist to be about " | |
| ) | |
| example_prompts = gr.Dataset( | |
| components=[song_prompt], | |
| samples=[ | |
| ['heartbreak'], | |
| ['love at the beach'], | |
| ['uncertainty and bleak hopes'] | |
| ] | |
| ) | |
| with gr.Column(): | |
| fetch_songs = gr.Button( | |
| value='Enter to see playlist',).style(full_width=True) | |
| with gr.Column(): | |
| song_options = gr.Radio(label='songs', interactive=True, | |
| choices=None, type='value', | |
| visible=True) | |
| with gr.Column(): | |
| with gr.Row(): | |
| tile1 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| tile2 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| tile3 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| with gr.Row(): | |
| tile4 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| tile5 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| tile6 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| with gr.Row(): | |
| tile7 = gr.Image(value="/content/songs_cover.jpg", | |
| show_label=False, visible=True) | |
| tile8 = gr.Image(value='/content/songs_cover.jpg', | |
| show_label=False, visible=True) | |
| tiles = [tile1, tile2, tile3, tile4, tile5, tile6, tile7, tile8] | |
| fetch_songs.click( | |
| fn=generate_playlist, | |
| inputs=[song_prompt], | |
| outputs=[song_options, *tiles] | |
| ) | |
| example_prompts.click( | |
| fn=set_example_prompt, | |
| inputs=[example_prompts], | |
| outputs=example_prompts.components | |
| ) | |
| demo.launch(debug=True) | |
| def main(): | |
| create_frontend() | |
| if __name__ == "__main__": | |
| main() | |