Update src/streamlit_app.py

src/streamlit_app.py

@@ -172,6 +172,7 @@ def create_bar_chart(df, view_type):
             color_continuous_scale=px.colors.diverging.Fall,
             orientation="v",
         )
+        max_score = df_fig["score"].max()
         fig.update_layout(
             xaxis_title_text="Model",
             yaxis_title_text="Score (Lower is better)",
@@ -182,12 +183,13 @@ def create_bar_chart(df, view_type):
             height=600,
             showlegend=False,
             margin=dict(
-                l=80,   # Left
-                r=0,    # Right
-                b=80,   # Bottom
-                t=70,   # Top
-                pad=0   # Padding
+                l=80,  # Left
+                r=0,   # Right
+                b=80,  # Bottom
+                t=70,  # Top
+                pad=0  # Padding
             ),
+            yaxis_range=[0, max_score * 1.25],
         )
         
         # Remove the color legend from the chart.
@@ -227,7 +229,7 @@ def create_bar_chart(df, view_type):
             x="model",
             y="score",
             color="embodiment",
-            color_continuous_scale=px.colors.qualitative.Plotly,
+            color_discrete_sequence=px.colors.qualitative.Plotly,
             orientation="v",
         )
         max_score = df_fig["score"].max()
@@ -241,33 +243,28 @@ def create_bar_chart(df, view_type):
             barmode="group",
             height=600,
             margin=dict(
-                l=80,   # Left
-                r=0,    # Right
-                b=80,   # Bottom
-                t=70,   # Top
-                pad=0   # Padding
+                l=80,  # Left
+                r=0,   # Right
+                b=80,  # Bottom
+                t=70,  # Top
+                pad=0  # Padding
             ),
             showlegend=True,
             legend=dict(
+                orientation="h",
                 x=0,
                 y=1,
                 xanchor="left",
                 yanchor="top",
-                bgcolor="rgba(255,255,255,0)",
-                bordercolor="rgba(0,0,0,0)",
                 borderwidth=1,
                 itemclick="toggle",
                 itemdoubleclick="toggleothers",
-                title=dict(
-                    text=f"<b>Embodiments</b>",
-                    font=dict(size=12),
-                    side="top center",
-                ),
-                font=dict(size=10)
+                title_text="<b>Embodiments</b>",
+                font=dict(size=15)
             ),
             uniformtext_minsize=10,
             uniformtext_mode="show",
-            yaxis_range=[0, max_score * 1.25]
+            yaxis_range=[0, max_score * 1.25],
         )
         
         # Remove the color legend from the chart.