Spaces:
Runtime error
Runtime error
| import time | |
| import json | |
| import os | |
| import pickle | |
| import gradio as gr | |
| import numpy as np | |
| import yt_dlp | |
| import sys | |
| import csv | |
| from huggingface_hub import hf_hub_download | |
| embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/' | |
| embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>' | |
| # NO GPU | |
| os.environ['CUDA_VISIBLE_DEVICES'] = '-1' | |
| os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' | |
| import tensorflow | |
| tensorflow.config.threading.set_intra_op_parallelism_threads(2) | |
| tensorflow.config.threading.set_inter_op_parallelism_threads(2) | |
| # | |
| python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'], | |
| use_auth_token=os.environ['TOKEN']) | |
| print("PATH : ", python_path) | |
| sys.path.append(os.environ['PRIVATE_DIR']) | |
| from models import * | |
| ind = get_index() | |
| ind_filenames = get_audio_names() | |
| catalog = get_catalog() | |
| url_dict = get_durl_myma() | |
| ###dict_catalog = get_dict_catalog() # dict containing track infos : key = track name (without ".mp3") | |
| def download_audio_(link): | |
| with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(title)s.mp3'}) as video: | |
| #with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(id).mp3'}) as video: | |
| info_dict = video.extract_info(link, download = True) | |
| video_title = info_dict['title'] | |
| #video_title = info_dict['id'] # KeyError !!!!! | |
| video.download(link) | |
| return video_title #"tmp" | |
| def download_audio(id_video): | |
| id = id_video.split("?v=")[-1][:11] | |
| audio_file = download_audio_(id_video) | |
| audio_file = audio_file+'.mp3' | |
| embed_html_all = embed_html1 + id +embed_html2 | |
| return audio_file, audio_file, embed_html_all | |
| def process_url(input_path): | |
| # setup the client | |
| #try : | |
| audio_file, audio_file, embed_html_all = download_audio(input_path) | |
| return process(audio_file, embed_html_all) | |
| #except: | |
| # return "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "" | |
| def process_file(input_path): | |
| return process(input_path, '')[1:] | |
| def process(audio_file, embed_html_all): | |
| #try : | |
| timestart = time.time() | |
| emb, ts = get_embed(audio_file) | |
| print("Embed time :", time.time()-timestart) | |
| D, I = do_search(emb, ind) | |
| print("Search + embed time :", time.time()-timestart) | |
| prin("DEBUG DISTANCES : ", D) | |
| #top1, top2, top3, top4, top5, top6, top7, top8, top9, top10 = get_top(I, ind_filenames, url_dict, catalog) | |
| top1, top2, top3, top4, top5, top6, top7, top8, top9, top10 = get_topN(I, ind_filenames, url_dict, catalog, 10) | |
| print("Total time :", time.time()-timestart) | |
| return embed_html_all, ind_filenames[I[0][0]].split('.')[0], top1, ind_filenames[I[0][1]].split('.')[0], top2, ind_filenames[I[0][2]].split('.')[0], top3, ind_filenames[I[0][3]].split('.')[0], top4, ind_filenames[I[0][4]].split('.')[0], top5, ind_filenames[I[0][5]].split('.')[0], top6, ind_filenames[I[0][6]].split('.')[0], top7, ind_filenames[I[0][7]].split('.')[0], top8, ind_filenames[I[0][8]].split('.')[0], top9, ind_filenames[I[0][9]].split('.')[0], top10 | |
| #except: | |
| # return embed_html_all, "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "", "Erreur Input", "" | |
| topname=[] | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| with gr.Column(): | |
| with gr.Row(): | |
| #gr.HTML(embed_html) | |
| html = gr.HTML() | |
| with gr.Row(): | |
| with gr.Column(): | |
| audio_url_input = gr.Textbox(placeholder='YouTube video URL', label='YouTube video URL') | |
| analyze_url_btn = gr.Button('Search from URL') | |
| with gr.Row(): | |
| with gr.Column(): | |
| audio_input_file = gr.Audio(type="filepath", label='Audio Input') | |
| analyze_file_btn = gr.Button('Search from file') | |
| with gr.Row(): | |
| with gr.Column(): | |
| ''' | |
| gr.HTML("<h3>Top 1</h3>") | |
| top1 = gr.Textbox(label="top1", show_label=False) | |
| gr.HTML("<h3>Top 2</h3>") | |
| top2 = gr.Textbox(label="top2", show_label=False) | |
| gr.HTML("<h3>Top 3</h3>") | |
| top3 = gr.Textbox(label="top3", show_label=False) | |
| gr.HTML("<h3>Top 4</h3>") | |
| top4 = gr.Textbox(label="top4", show_label=False) | |
| gr.HTML("<h3>Top 5</h3>") | |
| top5 = gr.Textbox(label="top5", show_label=False) | |
| ''' | |
| topname.append(gr.Textbox(label="top"+str(1)+"track name", show_label=True)) | |
| topname.append(gr.Audio(label="top"+str(1), show_label=False)) | |
| #top1_name = gr.Textbox(label="top1 track name", show_label=True) | |
| #top1 = gr.Audio(label="top1", show_label=False) | |
| top2_name = gr.Textbox(label="top2 track name", show_label=True) | |
| top2 = gr.Audio(label="top2", show_label=False) | |
| top3_name = gr.Textbox(label="top3 track name", show_label=True) | |
| top3 = gr.Audio(label="top3", show_label=False) | |
| top4_name = gr.Textbox(label="top4 track name", show_label=True) | |
| top4 = gr.Audio(label="top4", show_label=False) | |
| top5_name = gr.Textbox(label="top5 track name", show_label=True) | |
| top5 = gr.Audio(label="top5", show_label=False) | |
| top6_name = gr.Textbox(label="top6 track name", show_label=True) | |
| top6 = gr.Audio(label="top6", show_label=False) | |
| top7_name = gr.Textbox(label="top7 track name", show_label=True) | |
| top7 = gr.Audio(label="top7", show_label=False) | |
| top8_name = gr.Textbox(label="top8 track name", show_label=True) | |
| top8 = gr.Audio(label="top8", show_label=False) | |
| top9_name = gr.Textbox(label="top9 track name", show_label=True) | |
| top9 = gr.Audio(label="top9", show_label=False) | |
| top10_name = gr.Textbox(label="top10 track name", show_label=True) | |
| top10 = gr.Audio(label="top10", show_label=False) | |
| analyze_url_btn.click(process_url, inputs=[audio_url_input], | |
| outputs=[html]+topname+[top2_name, top2, top3_name, top3, top4_name, top4, top5_name, top5, top6_name, top6, top7_name, top7, top8_name, top8, top9_name, top9, top10_name, top10]) | |
| gr.Examples( | |
| examples = [ | |
| "https://www.youtube.com/watch?v=aNzCDt2eidg", | |
| "https://www.youtube.com/watch?v=NBE-uBgtINg", | |
| "https://www.youtube.com/watch?v=5NV6Rdv1a3I", | |
| "https://www.youtube.com/watch?v=OiC1rgCPmUQ",# | |
| "https://www.youtube.com/watch?v=dRX0wDNK6S4",# | |
| "https://www.youtube.com/watch?v=Guzu9aAeDIc"], | |
| inputs = [audio_url_input], | |
| outputs = [html]+topname+[top2_name, top2, top3_name, top3, top4_name, top4, top5_name, top5, top6_name, top6, top7_name, top7, top8_name, top8, top9_name, top9, top10_name, top10], | |
| fn = process_url, | |
| cache_examples=False, | |
| ) | |
| analyze_file_btn.click(process_file, inputs=[audio_input_file], | |
| outputs=topname+[top2_name, top2, top3_name, top3, top4_name, top4, top5_name, top5, top6_name, top6, top7_name, top7, top8_name, top8, top9_name, top9, top10_name, top10]) | |
| demo.launch(debug=False) |