Hugging Face's logo

Spaces:
Simbals
/
AudioSimilarity
Runtime error

App Files Community
AudioSimilarity
File size: 4,475 Bytes 
3b04551
 
 
 
 
 
f95be78
7c1d20a
0423904
3b04551
 
 
7d21ed5
 
 
 
03e603a
ee994fc
 
365609b
 
7d21ed5
f95be78
 
 
 
 
 
3b04551
 
 
f95be78
 
3b04551
 
 
 
 
 
341f056
3b04551
 
341f056
3b04551
 
7761659
f81ad55
 
0eff579
f81ad55
03e603a
341f056
3b04551
415e92e
 
3b04551
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6468b0f
 
6f33556
128bd47
 
 
ebf2b7b
 
8d5f163
6f33556
 
 
128bd47
6468b0f
3b04551
 
 
415e92e
8d5f163
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
import time
import json
import os
import pickle
import gradio as gr
import numpy as np
import yt_dlp
import sys
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'

#
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'])
sys.path.append(os.environ['PRIVATE_DIR'])
from models import *

def download_audio_(link):
  with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(title)s.mp3'}) as video:
    info_dict = video.extract_info(link, download = True)
    video_title = info_dict['title']
    video.download(link)    
    return video_title

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
    audio_file, audio_file, embed_html_all = download_audio(input_path)
    return process(audio_file, embed_html_all)

def process_file(input_path):
    return process(input_path, '')[1:]

def process(audio_file, embed_html_all):
    emb, ts = get_embed(audio_file)
    ind = get_index()
    ind_filenames = get_audio_names()
    catalog = get_catalog()
    _, I = do_search(emb, ind)
    top1, top2, top3, top4, top5 = get_top(I, ind_filenames, catalog)   
    return embed_html_all, top1, top2, top3, top4, top5
            
poc_examples = [["https://www.youtube.com/watch?v=aNzCDt2eidg"]]

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)
                    '''
                    top1 = gr.Audio(label="top1", show_label=True)
                    top2 = gr.Audio(label="top2", show_label=True)
                    top3 = gr.Audio(label="top3", show_label=True)
                    top4 = gr.Audio(label="top4", show_label=True)
                    top5 = gr.Audio(label="top5", show_label=True)
            
    
    analyze_url_btn.click(process_url, inputs=[audio_url_input], 
                      outputs=[html,top1, top2, top3, top4, top5])
    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,top1, top2, top3, top4, top5],
        fn = process_url,
        cache_examples=False,
    )
    analyze_file_btn.click(process_file, inputs=[audio_input_file], 
                      outputs=[top1, top2, top3, top4, top5])


demo.launch(debug=True)