from github import Github
import os 
import streamlit as st

import plotly.figure_factory as ff

g = Github(os.environ.get('github'))

inference_repo = g.get_repo("bigcode-project/bigcode-inference-benchmark")
df = list()

for milestone in inference_repo.get_milestones():
    desc = dict()
    for line in milestone.description.split('\n'):
        tmp = line.split(":")
        desc[tmp[0]] = tmp[1]
    st.write(desc)
    df.append(dict(Task=milestone.title, Start=milestone.due_on, Finish=milestone.due_on, Resource="Not Started"))

colors = {'Not Started': 'rgb(220, 0, 0)',
          'Incomplete': (1, 0.9, 0.16),
          'Complete': 'rgb(0, 255, 100)'}

fig = ff.create_gantt(df, colors=colors, index_col='Resource', show_colorbar=True)
st.plotly_chart(fig)