Update app.py

app.py

@@ -42,7 +42,27 @@ WRITING_STYLE_COLS = ['#P', 'Model', 'Reg+MyScore 🏆', 'Reg+Int 🏆', 'MyScor
 ANIME_RATING_COLS = ['#P', 'Model', 'Score 🏆', 'Dif', 'Cor', 'Std']
 
 # Load the leaderboard data from a CSV file
-def update_table(df: pd.DataFrame, query: str, param_ranges: list, w10_range: list, columns: list) -> pd.DataFrame:
+def load_leaderboard_data(csv_file_path):
+    try:
+        df = pd.read_csv(csv_file_path)
+        df['Model'] = df.apply(lambda row: f'<a href="{row["Link"]}" target="_blank" style="color: blue; text-decoration: none;">{row["Model"]}</a>' if pd.notna(row["Link"]) else row["Model"], axis=1)
+        df.drop(columns=['Link'], inplace=True)
+        
+        # Round numeric columns to 3 decimal places
+        numeric_columns = df.select_dtypes(include=[np.number]).columns
+        df[numeric_columns] = df[numeric_columns].round(3)
+        
+        # Round the W/10 column to 1 decimal place
+        if 'W/10 👍' in df.columns:
+            df['W/10 👍'] = df['W/10 👍'].round(1)
+        
+        return df
+    except Exception as e:
+        print(f"Error loading CSV file: {e}")
+        return pd.DataFrame(columns=UGI_COLS + WRITING_STYLE_COLS + ANIME_RATING_COLS)
+
+# Update the leaderboard table based on the search query and parameter range filters
+def update_table(df: pd.DataFrame, query: str, param_ranges: list, columns: list) -> pd.DataFrame:
     filtered_df = df.copy()
     if param_ranges:
         param_mask = pd.Series(False, index=filtered_df.index)
@@ -65,9 +85,6 @@ def update_table(df: pd.DataFrame, query: str, param_ranges: list, w10_range: li
                 param_mask |= (filtered_df['Params'] >= 65)
         filtered_df = filtered_df[param_mask]
     
-    # Apply W/10 filter
-    filtered_df = filtered_df[(filtered_df['W/10 👍'] >= w10_range[0]) & (filtered_df['W/10 👍'] <= w10_range[1])]
-    
     if query:
         filtered_df = filtered_df[filtered_df['Model'].str.contains(query, case=False, na=False)]
     
@@ -103,15 +120,6 @@ with GraInter:
                 interactive=True,
                 elem_id="filter-columns-size",
             )
-        with gr.Row():
-            w10_slider = gr.Slider(
-                minimum=0,
-                maximum=10,
-                value=[0, 10],
-                step=0.1,
-                label="W/10 Range",
-                elem_id="w10-slider"
-            )
     
     # Load the initial leaderboard data
     leaderboard_df = load_leaderboard_data("ugi-leaderboard-data.csv")
@@ -231,36 +239,30 @@ with GraInter:
             **NA:** When models either reply with one number for every anime, give ratings not between 1 and 10, or don't give every anime in the list a rating.
             """)
 
-    def update_all_tables(query, param_ranges, w10_range):
-        ugi_table = update_table(leaderboard_df, query, param_ranges, w10_range, UGI_COLS)
+    def update_all_tables(query, param_ranges):
+        ugi_table = update_table(leaderboard_df, query, param_ranges, UGI_COLS)
         
         ws_df = leaderboard_df.sort_values(by='Reg+MyScore 🏆', ascending=False)
-        ws_table = update_table(ws_df, query, param_ranges, w10_range, WRITING_STYLE_COLS)
+        ws_table = update_table(ws_df, query, param_ranges, WRITING_STYLE_COLS)
         
         arp_df = leaderboard_df.sort_values(by='Score 🏆', ascending=False)
         arp_df_na = arp_df[arp_df[['Dif', 'Cor']].isna().any(axis=1)]
         arp_df = arp_df[~arp_df[['Dif', 'Cor']].isna().any(axis=1)]
         
-        arp_table = update_table(arp_df, query, param_ranges, w10_range, ANIME_RATING_COLS)
-        arp_na_table = update_table(arp_df_na, query, param_ranges, w10_range, ANIME_RATING_COLS).fillna('NA')
+        arp_table = update_table(arp_df, query, param_ranges, ANIME_RATING_COLS)
+        arp_na_table = update_table(arp_df_na, query, param_ranges, ANIME_RATING_COLS).fillna('NA')
         
         return ugi_table, ws_table, arp_table, arp_na_table
 
     search_bar.change(
         fn=update_all_tables,
-        inputs=[search_bar, filter_columns_size, w10_slider],
+        inputs=[search_bar, filter_columns_size],
         outputs=[leaderboard_table_ugi, leaderboard_table_ws, leaderboard_table_arp, leaderboard_table_arp_na]
     )
     
     filter_columns_size.change(
         fn=update_all_tables,
-        inputs=[search_bar, filter_columns_size, w10_slider],
-        outputs=[leaderboard_table_ugi, leaderboard_table_ws, leaderboard_table_arp, leaderboard_table_arp_na]
-    )
-    
-    w10_slider.change(
-        fn=update_all_tables,
-        inputs=[search_bar, filter_columns_size, w10_slider],
+        inputs=[search_bar, filter_columns_size],
         outputs=[leaderboard_table_ugi, leaderboard_table_ws, leaderboard_table_arp, leaderboard_table_arp_na]
     )