Style

app.py

@@ -179,9 +179,10 @@ org_info = {
     "AI21 Labs": ("#1E90FF", "🇮🇱"),  # Dodger Blue,
     "Reka AI": ("#FFC300", "🇺🇸"),
     "Zhipu AI": ("#FFC300", "🇨🇳"),
+    "Nvidia": ("#76B900", "🇺🇸"),
 }
 
-def make_figure(original_df, start_time_gradio):
+def make_figure(original_df, start_time_gradio, speak_french):
     fig = go.Figure()
 
     start_date = pd.to_datetime(start_time_gradio, unit='s')
@@ -267,7 +268,6 @@ def make_figure(original_df, start_time_gradio):
 
 
     # Update layout
-    speak_french = False
     if speak_french:
         fig.update_layout(
             title="La course au classement",
@@ -286,9 +286,9 @@ def make_figure(original_df, start_time_gradio):
         xaxis_title="Date",
         hovermode="closest",
         xaxis_range=[start_date, current_date],  # Extend x-axis for labels
-        yaxis_range=[df.loc[df["Release Date"] >= start_date]["rating"].min() - margin, df["rating"].max() + margin],
+        yaxis_range=[df.loc[df["Release Date"] >= start_date]["rating"].min(), df["rating"].max() + margin],
     )
-    apply_template(fig, annotation_text="Aymeric Roucher", height=500)
+    apply_template(fig, annotation_text="Aymeric Roucher", height=600)
 
     fig.update_xaxes(
         tickformat="%m-%Y",
@@ -318,7 +318,8 @@ with gr.Blocks(
     with gr.Row():
         top_n_orgs = gr.Slider(minimum=1, maximum=30, value=10, step=1, label="View top N companies")
         # minimum_rating = gr.Slider(minimum=800, maximum=1300, value=1000, step=1, label="Restrict to ELO scores above N")
-        start_time = gr.DateTime(value="2024-01-01 00:00:00")
+        start_time = gr.DateTime(value="2024-01-01 00:00:00", label="Start time")
+        speak_french = gr.Checkbox(value=False, label="Parler français")
     with gr.Group():
         with gr.Tab("Plot"):
             plot = gr.Plot(show_label=False)
@@ -344,24 +345,29 @@ with gr.Blocks(
         outputs=filtered_df,
     ).then(
         fn=make_figure,
-        inputs=[filtered_df, start_time],
+        inputs=[filtered_df, start_time, speak_french],
         outputs=[plot, display_df],
     )
 
-    # minimum_rating.change(
-    #     fn=filter_df,
-    #     inputs=[top_n_orgs, minimum_rating],
-    #     outputs=filtered_df,
-    # ).then(
-    #     fn=make_figure,
-    #     inputs=[filtered_df],
-    #     outputs=[plot, display_df],
-    # )
+    top_n_orgs.change(
+        fn=filter_df,
+        inputs=[top_n_orgs],
+        outputs=filtered_df,
+    ).then(
+        fn=make_figure,
+        inputs=[filtered_df, start_time, speak_french],
+        outputs=[plot, display_df],
+    )
 
     start_time.change(
         fn=make_figure,
-        inputs=[filtered_df, start_time],
+        inputs=[filtered_df, start_time, speak_french],
         outputs=[plot, display_df],
     )
-        
+       
+    speak_french.change(
+        fn=make_figure,
+        inputs=[filtered_df, start_time, speak_french],
+        outputs=[plot, display_df],
+    ) 
 demo.launch()