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, genres, instruments, themes, str(dict_desc['Vocal/Instrumental']), str(dict_desc['Vocal/Instrumental'])
    
'''
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():
                with gr.Column():
                    dict_moods=gr.Label(label="Moods", show_label=False)
                    gr.Textbox("Hello", label="music desc")
                themes=gr.Label(label="Themes", show_label=False)
                genres  = gr.Label(label="Genres", show_label=False)
                instruments  = gr.Label(label="Instruments", show_label=False)
            
                #vocalinstru = gr.Textbox(label="Vocal/Instrumental"),
                
                '''
                themes = gr.Dataset(components=[gr.Textbox(visible=False)],
                    label="type de tags",
                    samples=[
                        ["Tag1"],
                        ["Tag2"],
                        ["Tag3"],
                        ["Tag4"]
                    ],
                )
                '''
                #themes = gr.Dataset(components=[gr.Textbox(visible=False)],label="type de tags")
        
                with gr.Column():
                    dict_desc = gr.Textbox(label="music desc")
                    vocalinstru = gr.Textbox(label="Vocal/Instrumental")
                

    analyze_btn.click(process, inputs=[audio_input], outputs=[dict_moods, genres, instruments, themes, dict_desc, vocalinstru])
    
demo.launch(debug=False)