import numpy as np
import torch
import gradio as gr

from libs import *

with gr.Blocks(title="Signal Similarity Live Demo") as demo:

    default_fig, default_result = demo_fn()

    with gr.Row():
        signal_images = gr.Plot(default_fig,label="Visualize Signals")
    with gr.Row():
        singal_results = gr.Text(default_result, interactive=False)

    with gr.Row():
        with gr.Column():
            gr.Markdown("### Ground Truth Control")
            gt_button     = gr.Radio(["Sin", "Cos", "Random"], label="Signal Type", value="Sin")
            gt_scale_slider  = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Scaling Factor", value=1)
            gt_shift_slider  = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Shifting Factor",  value=0)
        with gr.Column():
            gr.Markdown("### Pred Signal Control")
            signal_button = gr.Radio(["Sin", "Cos", "Random"], label="Signal Type",  value="Sin")
            pred_scale_slider  = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Scaling Factor", value=1)
            pred_shift_slider  = gr.Slider(minimum=-2, maximum=2, step=0.1, label="Shifting Factor",  value=0)
            
    gt_button.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
    signal_button.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
    gt_scale_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
    gt_shift_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
    pred_scale_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])
    pred_shift_slider.change(fn=demo_fn, inputs=[gt_button, signal_button, gt_scale_slider, gt_shift_slider, pred_scale_slider, pred_shift_slider], outputs=[signal_images, singal_results])


demo.launch(share=True)