import numpy as np
import pandas as pd
import panel as pn

pn.extension('vizzu')

data = {
    'Name': ['Alice', 'Bob', 'Ted', 'Patrick', 'Jason', 'Teresa', 'John'],
    'Weight': 50+np.random.randint(0, 10, 7)*10
}

vizzu = pn.pane.Vizzu(
    data, config={'geometry': 'rectangle', 'x': 'Name', 'y': 'Weight', 'title': 'Weight by person'},
    duration=400, height=400, sizing_mode='stretch_width', tooltip=True
)

vizzu

vizzu.animate({'geometry': 'circle'})

vizzu.animate({'geometry': 'area'})

print(vizzu.config)

vizzu

df = pd.DataFrame(np.random.randn(50), columns=list('Y')).cumsum()

pn.pane.Vizzu(
    df, column_types={'index': 'dimension'}, config={'x': 'index', 'y': 'Y', 'geometry': 'line'},
    height=300, sizing_mode='stretch_width'
)

windturbines = pd.read_csv('https://datasets.holoviz.org/windturbines/v1/windturbines.csv')

agg = windturbines.groupby(['p_year', 't_manu'])[['p_cap']].sum().sort_index(level=0).reset_index()

chart_type = pn.widgets.RadioButtonGroup(options={'Stream': 'stream', 'Bar': 'stackedColumn'}, align='center')

preset_chart = pn.pane.Vizzu(
    agg,
    config=pn.bind(lambda preset: {'preset': preset, 'x': 'p_year', 'y': 'p_cap', 'stackedBy': 't_manu'}, chart_type),
    column_types={'p_year': 'dimension'},
    height=500,
    sizing_mode='stretch_width',
    style={
        'plot': {
            "xAxis": {
                "label": {
                    "angle": "-45deg"
                }
            }
        }
    }
)


pn.Column(chart_type, preset_chart).embed()

pn.Row(vizzu.controls(jslink=True), vizzu)