WIP commit. Troubleshoot chart display. Add behavior of filter

app.py

@@ -48,6 +48,14 @@ class MultiURLData:
 
 data_provider = MultiURLData()
 
+st.title('Leaderboard')
+
+# TODO actually use these checkboxes as filters
+## Desired behavior
+## model and column selection is hidden by default
+## when the user clicks the checkbox, the model and column selection appears
+filters = st.checkbox('Add filters')
+
 # Create checkboxes for each column
 selected_columns = st.multiselect(
     'Select Columns',
@@ -65,3 +73,30 @@ selected_models = st.multiselect(
 # Get the filtered data and display it in a table
 filtered_data = data_provider.get_data(selected_models)
 st.dataframe(filtered_data)
+
+
+
+#TODO fix this plot.  currently has an error
+# Create a plot with new data
+df = pd.DataFrame({
+    'Model': list(filtered_data['Model Name']),
+    'harness|arc:challenge|25_rank': list(filtered_data['harness|arc:challenge|25_rank']),
+    'harness|moral_scenarios|5_rank': list(filtered_data['harness|moral_scenarios|5_rank']),
+})
+
+# Calculate color column
+df['color'] = 'purple'
+df.loc[df['harness|moral_scenarios|5_rank'] < df['harness|arc:challenge|25_rank'], 'color'] = 'red'
+df.loc[df['harness|moral_scenarios|5_rank'] > df['harness|arc:challenge|25_rank'], 'color'] = 'blue'
+
+# Create the scatter plot
+fig = px.scatter(df, x='harness|arc:challenge|25_rank', y='harness|moral_scenarios|5_rank', color='color', hover_data=['Model'])
+fig.update_layout(showlegend=False,  # hide legend
+                  xaxis = dict(autorange="reversed"),  # reverse X-axis
+                  yaxis = dict(autorange="reversed"))  # reverse Y-axis
+
+# Show the plot in Streamlit
+st.plotly_chart(fig)
+
+
+