Update app.py

app.py

@@ -1,8 +1,9 @@
-import gradio as gr
 import altair as alt
-from datasets import load_dataset
+import gradio as gr
 import pandas as pd
 
+from datasets import load_dataset
+
 model_id = "ybelkada/model_cards_correct_tag"
 dataset = load_dataset(model_id, split="train").to_pandas()
 
@@ -10,29 +11,70 @@ dataset = load_dataset(model_id, split="train").to_pandas()
 df = pd.DataFrame(dataset)
 df["commit_dates"] = pd.to_datetime(df["commit_dates"])  # Convert commit_dates to datetime format
 df = df.sort_values(by="commit_dates")
+melted_df = pd.melt(df, id_vars=['commit_dates'], value_vars=['total_transformers_model', 'missing_library_name'], var_name='type')
+
+df['ratio'] = (1 - df['missing_library_name'] / df['total_transformers_model']) * 100
+ratio_df = df = df[['commit_dates', 'ratio']].copy()
+
+def make_plot(plot_type):
+    if plot_type == "Total models with missing 'transformers' tag":
+        highlight = alt.selection(type='single', on='mouseover',
+                                fields=['type'], nearest=True)
+
+
+        base = alt.Chart(melted_df).encode(
+            x=alt.X('commit_dates:T', title='Date'),
+            y=alt.Y('value:Q', scale=alt.Scale(domain=(melted_df['value'].min(), melted_df['value'].max())), title="Count"),
+            color='type:N',
+        )
+
+        points = base.mark_circle().encode(
+            opacity=alt.value(1),
+        ).add_selection(
+            highlight
+        ).properties(
+            width=1200,
+            height=800,
+        )
+
+        lines = base.mark_line().encode(
+            size=alt.condition(~highlight, alt.value(1), alt.value(3))
+        )
+
+        return points + lines
+    else:
+        highlight = alt.selection(type='single', on='mouseover',
+                                fields=['ratio'], nearest=True)
+
+        base = alt.Chart(ratio_df).encode(
+            x=alt.X('commit_dates:T', title='Date'),
+            y=alt.Y('ratio:Q', scale=alt.Scale(domain=(ratio_df['ratio'].min(), ratio_df['ratio'].max())), title="(1 - missing_library_name / total_transformers_model) * 100 - Higher is better"),
+        )
+
+        points = base.mark_circle().encode(
+            opacity=alt.value(1)
+        ).add_selection(
+            highlight
+        ).properties(
+            width=1200,
+            height=800,
+        )
 
-def plot_fn():
-    line_chart = alt.Chart(df).mark_line().encode(
-        x=alt.X('commit_dates:T', axis=alt.Axis(title='Date')),
-        y=alt.Y('total_transformers_model:Q', axis=alt.Axis(title='Count'), scale=alt.Scale(zero=False)),
-        color=alt.value('blue'),
-        tooltip=['commit_dates:T', 'total_transformers_model:Q'],
-    ).properties(width=600, height=400)
-
-    line_chart_missing_library = alt.Chart(df).mark_line().encode(
-        x=alt.X('commit_dates:T', axis=alt.Axis(title='Date')),
-        y=alt.Y('missing_library_name:Q', axis=alt.Axis(title='Count'), scale=alt.Scale(zero=False)),
-        color=alt.value('orange'),
-        tooltip=['commit_dates:T', 'missing_library_name:Q'],
-    ).properties(width=600, height=400)
-    
-    chart = (line_chart + line_chart_missing_library).properties(width=600, height=400)
-
-    return chart
+        lines = base.mark_line().encode(
+            size=alt.condition(~highlight, alt.value(1), alt.value(3))
+        )
+        
+        return points + lines
 
 with gr.Blocks() as demo:
+    button = gr.Radio(
+        label="Plot type",
+        choices=["Total models with missing 'transformers' tag", "Proportion of models correctly tagged with 'transformers' tag"], 
+        value="Total models with missing 'transformers' tag"
+    )
     plot = gr.Plot(label="Plot")
-    demo.load(plot_fn, inputs=[], outputs=[plot])
+    button.change(make_plot, inputs=button, outputs=[plot])
+    demo.load(make_plot, inputs=[button], outputs=[plot])
 
 if __name__ == "__main__":
     demo.launch()