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.append((d['name'],d['probability'])) print(themes) instruments = [] for d in res['instruments']: instruments.append((d['name'],d['probability'])) print(instruments) 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 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(): moods=gr.outputs.Label(label="Moods") moods=gr.outputs.Label(label="Moods") analyze_btn.click(process, inputs=[audio_input], outputs=[dict_moods, dict_desc, genres]) #audio_input.submit(process, inputs=[audio_input], outputs=[moods]) # pas de sens ! demo.launch(debug=False)