from github import Github
import os 
import streamlit as st
st.set_page_config(layout="wide")

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].lower()] = tmp[1].lower().strip()
    df.append(dict(Task=milestone.title, Start=desc['start date'], Finish=milestone.due_on.strftime('%Y-%m-%d'), Resource=desc['status']))

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, title="BigCode planning")
fig.update_xaxes(ticks= "outside",
                 ticklabelmode= "period", 
                 tickcolor= "black", 
                 ticklen=10, 
                 minor=dict(
                     ticklen=4,  
                     dtick=7*24*60*60*1000,  
                     tick0="2023-01-16", 
                     griddash='dot', 
                     gridcolor='white')
                )

st.plotly_chart(fig, use_container_width=True)