nataliaElv commited on
Commit
8d9fbc8
1 Parent(s): 3036563

Improve user contributions chart

Browse files
Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -5,7 +5,6 @@ from typing import List
5
  import argilla as rg
6
  import gradio as gr
7
  import pandas as pd
8
- import plotly.colors as colors
9
  import plotly.graph_objects as go
10
 
11
  client = rg.Argilla(
@@ -33,10 +32,7 @@ def get_progress(dataset: rg.Dataset) -> dict:
33
  def create_progress_bar(progress):
34
  top_labels = ['Completed', 'Pending']
35
 
36
- colors = ['rgba(38, 24, 74, 0.8)',
37
- # 'rgba(71, 58, 131, 0.8)',
38
- # 'rgba(122, 120, 168, 0.8)', 'rgba(164, 163, 204, 0.85)',
39
- 'rgba(190, 192, 213, 1)']
40
 
41
  x_data = [[progress["annotated"], progress["total"] - progress["annotated"]]]
42
 
@@ -118,28 +114,27 @@ def create_progress_bar(progress):
118
 
119
  def create_piechart(user_annotations):
120
  sorted_users = sorted(user_annotations.items(), key=lambda x: x[1], reverse=True)
121
- color_scale = colors.qualitative.Pastel + colors.qualitative.Set3
122
 
123
- labels, values, user_colors = [], [], []
124
 
125
- for i, (user, contribution) in enumerate(sorted_users):
126
  labels.append(user)
127
  values.append(contribution)
128
- user_colors.append(color_scale[i % len(color_scale)])
129
 
130
  fig = go.Figure(
131
  go.Pie(
132
  labels=labels,
133
- values=values,
134
  )
135
  )
136
 
137
  fig.update_layout(
138
- title_text="User contributions to the total end dataset",
139
  height=500,
140
  margin=dict(l=10, r=10, t=50, b=10),
141
- template="ggplot2",
142
  )
 
143
 
144
  return fig
145
 
 
5
  import argilla as rg
6
  import gradio as gr
7
  import pandas as pd
 
8
  import plotly.graph_objects as go
9
 
10
  client = rg.Argilla(
 
32
  def create_progress_bar(progress):
33
  top_labels = ['Completed', 'Pending']
34
 
35
+ colors = ['rgba(38, 24, 74, 0.8)', 'rgba(190, 192, 213, 1)']
 
 
 
36
 
37
  x_data = [[progress["annotated"], progress["total"] - progress["annotated"]]]
38
 
 
114
 
115
  def create_piechart(user_annotations):
116
  sorted_users = sorted(user_annotations.items(), key=lambda x: x[1], reverse=True)
 
117
 
118
+ labels, values = [], []
119
 
120
+ for user, contribution in sorted_users:
121
  labels.append(user)
122
  values.append(contribution)
 
123
 
124
  fig = go.Figure(
125
  go.Pie(
126
  labels=labels,
127
+ values=values
128
  )
129
  )
130
 
131
  fig.update_layout(
132
+ title_text="User contributions",
133
  height=500,
134
  margin=dict(l=10, r=10, t=50, b=10),
135
+ template="ggplot2"
136
  )
137
+ fig.update_traces(textposition='inside', textinfo='percent+label')
138
 
139
  return fig
140