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="songs_cover.jpg", | |
show_label=False, visible=True) | |
tile2 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
tile3 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
with gr.Row(): | |
tile4 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
tile5 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
tile6 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
with gr.Row(): | |
tile7 = gr.Image(value="songs_cover.jpg", | |
show_label=False, visible=True) | |
tile8 = gr.Image(value='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() | |