Update src/streamlit_app.py

src/streamlit_app.py

@@ -151,23 +151,74 @@ def create_bar_chart(df, view_type):
     """Create interactive bar chart based on view type"""
     
     if view_type == "Total Score":
-        fig = go.Figure(data=[
-            go.Bar(
-                x=df['Model'],
-                y=df['Total Score'],
-                orientation='v',
-                marker_color=px.colors.sequential.Blues,
-                text=df['Total Score'].round(1),
-                textposition='outside',
-            )
-        ])
+        
+        # Format df
+        df_fig = df.copy()
+        df_fig["Models"] = df_fig["model"].str.replace('_', ' ')
+        df_fig = df_fig[df_fig["score"] != np.inf]
+
+        # Calculate mean score per model
+        df_fig = df_fig.groupby("Models")[["score"]].mean().reset_index()
+        
+        # Sort the results from best to worst
+        df_fig = df_fig.sort_values(by="score", ascending=True)
+        
+        # Create the Plotly figure using Plotly Express, now plotting only the 'frechet' score.
+        fig = px.bar(
+            model_scores,
+            x="Models",
+            y="score",
+            color="score",
+            color_continuous_scale=px.colors.diverging.Fall,
+            template="plotly_white",
+            orientation="v",
+        )
+        
         fig.update_layout(
-            title="Model Performance - Total Score",
-            xaxis_title="Model",
-            yaxis_title="Score",
-            yaxis_range=[0, 100],
-            height=500,
+            xaxis_title_text="Model",
+            yaxis_title_text="Score (Lower is better)",
+            title_text="",
+            font=dict(size=15, color="black"),
+            xaxis_tickangle=-45,
+            bargap=0.2, # Increase gap for slimmer bars
+            height=500, # Set the height of the plot
+            margin=dict(
+                l=0,   # Left
+                r=0,   # Right
+                b=0,   # Bottom
+                t=5,   # Top
+                pad=0  # Padding
+            ),
         )
+        
+        # Remove the color legend from the chart.
+        fig.update_coloraxes(showscale=False)
+        
+        # Add annotations to show the exact score on each bar.
+        fig.update_traces(
+            texttemplate=""%{y:.2f}",
+            textposition="outside"
+        )
+
+
+        # TODO
+        # fig = go.Figure(data=[
+        #     go.Bar(
+        #         x=df['Model'],
+        #         y=df['Total Score'],
+        #         orientation='v',
+        #         marker_color=px.colors.sequential.Blues,
+        #         text=df['Total Score'].round(1),
+        #         textposition='outside',
+        #     )
+        # ])
+        # fig.update_layout(
+        #     title="Model Performance - Total Score",
+        #     xaxis_title="Model",
+        #     yaxis_title="Score",
+        #     yaxis_range=[0, 100],
+        #     height=500,
+        # )
     
     elif view_type == "Per Embodiment":
         embodiment_cols = [col for col in df.columns if col.startswith('Embodiment-')]