Clean up

app.py

@@ -1,5 +1,3 @@
-import os
-import json
 import gradio as gr
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
@@ -34,9 +32,11 @@ from src.envs import API, EVAL_REQUESTS_PATH, EVAL_RESULTS_PATH, QUEUE_REPO, REP
 from src.populate import get_evaluation_queue_df, get_leaderboard_df
 from src.submission.submit import add_new_eval
 
+
 def restart_space():
     API.restart_space(repo_id=REPO_ID)
 
+
 ### Space initialisation
 try:
     print(EVAL_REQUESTS_PATH)
@@ -63,6 +63,7 @@ leaderboard_df = original_df.copy()
     failed_eval_queue_df,
 ) = get_evaluation_queue_df(EVAL_REQUESTS_PATH, EVAL_COLS)
 
+
 # Searching and filtering
 def update_table(
     hidden_df: pd.DataFrame,
@@ -79,13 +80,13 @@ def update_table(
 ):
     print(f"Update table called with: type_query={type_query}, precision_query={precision_query}, size_query={size_query}")
     print(f"hidden_df shape before filtering: {hidden_df.shape}")
-    
+
     filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, add_special_tokens_query, num_few_shots_query, show_deleted, show_merges, show_flagged)
     print(f"filtered_df shape after filter_models: {filtered_df.shape}")
-    
+
     filtered_df = filter_queries(query, filtered_df)
     print(f"filtered_df shape after filter_queries: {filtered_df.shape}")
-    
+
     print(f"Filter applied: query={query}, columns={columns}, type_query={type_query}, precision_query={precision_query}")
     print("Filtered dataframe head:")
     print(filtered_df.head())
@@ -122,11 +123,11 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
         AutoEvalColumn.model_type_symbol.name,  # 'T'
         AutoEvalColumn.model.name,              # 'Model'
     ]
-    
+
     # 'always_here_cols' を 'columns' から除外して重複を避ける
     columns = [c for c in columns if c not in always_here_cols]
     new_columns = always_here_cols + [c for c in COLS if c in df.columns and c in columns]
-    
+
     # 重複を排除しつつ順序を維持
     seen = set()
     unique_columns = []
@@ -207,6 +208,7 @@ def filter_models(
     print(filtered_df.head())
     return filtered_df
 
+
 leaderboard_df = filter_models(leaderboard_df, [t.to_str(" : ") for t in ModelType], list(NUMERIC_INTERVALS.keys()), [i.value.name for i in Precision], [i.value.name for i in AddSpecialTokens], [i.value.name for i in NumFewShots], False, False, False)
 
 demo = gr.Blocks(css=custom_css)
@@ -304,17 +306,17 @@ with demo:
             # DataFrameの初期化部分のみを修正
             initial_columns = ['T'] + [c.name for c in fields(AutoEvalColumn) if (c.never_hidden or c.displayed_by_default) and c.name != 'T']
             leaderboard_df_filtered = select_columns(leaderboard_df, initial_columns)
-            
+
             # Model列のリンク形式を修正
             leaderboard_df_filtered['Model'] = leaderboard_df_filtered['Model'].apply(
                 lambda x: f'[{x.split(">")[-2].split("<")[0]}]({x.split("href=")[1].split(chr(34))[1]})' if isinstance(x, str) and 'href=' in x else x
             )
-            
+
             # 数値データを文字列に変換
             for col in leaderboard_df_filtered.columns:
                 if col not in ['T', 'Model']:
                     leaderboard_df_filtered[col] = leaderboard_df_filtered[col].astype(str)
-            
+
             # DataFrameコンポーネントの初期化
             leaderboard_table = gr.components.Dataframe(
                 value=leaderboard_df_filtered,
@@ -324,7 +326,7 @@ with demo:
                 interactive=False,
                 visible=True
             )
-            
+
             # Dummy leaderboard for handling the case when the user uses backspace key
             hidden_leaderboard_table_for_search = gr.components.Dataframe(
                 value=original_df[COLS],
@@ -371,7 +373,7 @@ with demo:
             )
             # Check query parameter once at startup and update search bar + hidden component
             demo.load(load_query, inputs=[], outputs=[search_bar, hidden_search_bar])
-            
+
             for selector in [shown_columns, filter_columns_type, filter_columns_precision, filter_columns_size, filter_columns_add_special_tokens, filter_columns_num_few_shots, deleted_models_visibility, merged_models_visibility, flagged_models_visibility]:
                 selector.change(
                     update_table,
@@ -515,4 +517,4 @@ with demo:
 scheduler = BackgroundScheduler()
 scheduler.add_job(restart_space, "interval", seconds=1800)
 scheduler.start()
-demo.queue(default_concurrency_limit=40).launch()
+demo.queue(default_concurrency_limit=40).launch()