Update app.py

app.py

@@ -14,13 +14,16 @@ g = Github(os.environ.get('github'))
 
 inference_repo = g.get_repo("bigcode-project/bigcode-inference-benchmark")
 df = list()
-
+all_status = 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=f"""<a href="https://www.github.com/bigcode-project/bigcode-inference-benchmark/milestone/{milestone.number}", target="_black">{milestone.title}</a>""", 
+    task_name = f"""<a href="https://www.github.com/bigcode-project/bigcode-inference-benchmark/milestone/{milestone.number}", target="_black">{milestone.title}</a>"""
+    if desc['status'] not in all_status:
+        all_status.append(desc['status'])
+    df.append(dict(Task=task_name, 
                    Start=desc['start date'], 
                    Finish=milestone.due_on.strftime('%Y-%m-%d'), 
                    Resource=desc['status'], 
@@ -37,6 +40,12 @@ colors = {'not started': 'rgb(217, 217, 217)',
           'high priority - help needed': 'rgb(255, 0, 0)',
           'completed': 'rgb(56, 118, 29)'}
 
+for key in colors.keys():
+    if key not in all_status:
+        df.append(dict(Task=task_name, 
+                   Start='2023-04-02', 
+                   Finish='2023-04-02', 
+                   Resource=key))
 
 fig = ff.create_gantt(df, colors=colors, 
                       index_col='Resource',