Update Leaderboard.py

Leaderboard.py

@@ -8,13 +8,15 @@ def load_data():
     df_oblzn = pd.read_csv("oblzn.csv").set_index("model")
     df_vidvopr = pd.read_csv("vidvopr.csv").set_index("model")
     df_provokac = pd.read_csv("provokac.csv").set_index("model")
+    df_api = pd.read_csv("API.csv").set_index("model")
+    df_os = pd.read_csv("OS.csv").set_index("model")
 
-    return df_itog, df_oblzn, df_vidvopr, df_provokac
+    return df_itog, df_oblzn, df_vidvopr, df_provokac, df_api, df_os
 
 def select_table(tables):
     table_choice = st.selectbox(
         "Выберите таблицу для отображения:",
-        ["ITOG", "Область знаний", "Вид вопроса", "Провокационность"],
+        ["ITOG", "Область знаний", "Вид вопроса", "Провокационность", "API модели", "OS модели"],
         index=0
     )
     
@@ -118,13 +120,15 @@ def filter_provokac_columns(df):
 
 st.title("Leaderboard")
 
-df_itog, df_oblzn, df_vidvopr, df_provokac = load_data()
+df_itog, df_oblzn, df_vidvopr, df_provokac, df_api, df_os = load_data()
 
 tables = {
     "ITOG": df_itog,
     "Область знаний": df_oblzn,
     "Вид вопроса": df_vidvopr,
     "Провокационность": df_provokac
+    "API модели": df_api,
+    "OS модели": df_os
 }
 
 df_selected = select_table(tables)