Spaces:
Runtime error
Runtime error
from simbals_apis_public_clients.clients.services import SimbalsAPIClient | |
import time | |
import json | |
import os | |
import gradio as gr | |
SIMBALS_GLOBAL_DB = 1 | |
SIMBALS_MAIN_DATABASE = 2 | |
def process(input_path): | |
# setup the client | |
client = SimbalsAPIClient(os.environ['TOKEN'], debug=True) | |
parameters = client.prepare_request("31415", audio_file=input_path) | |
# add audio features service with an available database | |
#parameters = client.add_audio_features(parameters, SIMBALS_MAIN_DATABASE) | |
parameters = client.add_tags(parameters, SIMBALS_MAIN_DATABASE) | |
# launch the request and test for ok/code values | |
ok, code, job_id = client.launch_request(parameters) | |
# try to get the results with obtained job_id. If code is 4, job is not finished. In all other cases, job is finished | |
for i in range(1000): | |
results=client.get_results(job_id) | |
if results[0] != 4: | |
print(json.dumps(results[1], indent=1)) | |
#return json.dumps(results[1], indent=1) | |
output = json.dumps(results[1], indent=1) | |
break | |
time.sleep(1) | |
res = results[1]["response"]["tagging"]["2"] | |
dict_moods = {} | |
for m in res['moods']: | |
dict_moods[m['name']] = m['probability'] | |
dict_desc={} | |
for d in res['music_descriptors']: | |
dict_desc[d['name']] = d['value'] | |
genres={} | |
for d in res['genres']: | |
genres[d['name']] = d['probability'] | |
themes = {} | |
for d in res['themes']: | |
themes[d['name']]= d['probability'] | |
instruments = {} | |
for d in res['instruments']: | |
instruments[d['name']] = d['probability'] | |
timbres = [] | |
for d in res['timbres']: | |
timbres.append((d['name'],d['probability'])) | |
print(timbres) | |
vocalgender = [res['vocal_gender'][0]['name'], res['vocal_gender'][0]['probability']] | |
print(vocalgender) | |
audioquality = [res['audio_quality'][0]['name'], res['audio_quality'][0]['probability']] | |
print(audioquality) | |
return dict_moods, dict_desc, genres, instruments, themes | |
demo = gr.Interface(fn=process, | |
inputs=gr.Audio(type="filepath"), | |
outputs=gr.outputs.Label(label="Moods"), | |
#outputs=gr.outputs.Textbox(label="Generated Text") | |
#examples=example_list, | |
#cache_examples=False | |
) | |
with gr.Blocks() as demo: | |
with gr.Row(): | |
with gr.Column(): | |
with gr.Row(): | |
audio_input = gr.Audio(type="filepath", label='Audio Input') | |
with gr.Row(): | |
analyze_btn = gr.Button('Analyze File') | |
with gr.Row(): | |
dict_moods=gr.outputs.Label(label="Moods") | |
themes=gr.outputs.Label(label="Themes") | |
genres = gr.outputs.Label(label="Genres") | |
instruments = gr.outputs.Label(label="Instruments") | |
analyze_btn.click(process, inputs=[audio_input], outputs=[dict_moods, dict_desc, genres, instruments, themes]) | |
#audio_input.submit(process, inputs=[audio_input], outputs=[moods]) # pas de sens ! | |
demo.launch(debug=False) |