import plotly as py
import plotly.graph_objs as go
#from plotly.offline import iplot
import chart_studio.plotly as py
import numpy as np
import chart_studio

#For config setup only
#chart_studio.tools.set_credentials_file(username='ZubairVetter', api_key='1rVxncXXbOqD6XaMWE1i')

x = np.linspace(0,0.999,250)
Co = np.logspace(14,20,7)
ko = np.linspace(0.03,0.99,33)

#Add traces, one for each slider step
data = []
for C in Co:
    for k in ko:
        trace = go.Scatter(
            visible=False,
            line=dict(color="deepskyblue", width=4),
            name="Co = "+str("{:.2E}".format(C))+', k = '+str("{:.2f}".format(k)),
            x = x,
            y=C*k*(1-x)**(k-1))
        data.append(trace)

len(data)
dictionaries = []
i=0
for trace in data:
    visibility = [False]*len(data)
    visibility[i]=True
    dictionary = dict(label = str(trace.name),
                    method = 'restyle',
                    args = ['visible',visibility])
    dictionaries.append(dictionary)
    i+=1

updatemenus = list([dict(buttons=list(dictionaries),
                         direction="down",
                         showactive=True,
                         x=0.9,
                         xanchor="right",
                         y=1.1,
                         yanchor="top")])


layout = dict(title='Crystal Segregation',
              showlegend=True,
              updatemenus=updatemenus,
              yaxis_type="log",
              width=800, height=800,
              yaxis = dict(showexponent='all',
                            exponentformat='e',
                            automargin=True,
                            title_text="Concentration cm^-3"),
              xaxis = dict(title_text="f"))

figure = dict(data=data,
            layout=layout)

#plot_url = py.plot(figure)
py.iplot(figure, filename = 'Segmentation Coefficient')
#iplot(figure)