Update app.py

app.py

@@ -13,11 +13,36 @@ api = HfApi()
 
 HF_TOKEN = os.getenv('HF_TOKEN')
 
+
+headers_models = ["🔢 Serial Number", "👤 Author Name", "📥 Total Downloads", "👍 Total Likes", "🤖 Number of Models",
+            "🏆 Best Model On Open LLM Leaderboard", "🥇 Best Rank On Open LLM Leaderboard",
+            "📊 Average Downloads per Model", "📈 Average Likes per Model", "🚀 Most Downloaded Model",
+            "📈 Most Download Count", "❤️ Most Liked Model", "👍 Most Like Count", "🔥 Trending Model",
+            "👑 Best Rank at Trending Models", "🏷️ Type"]
+
+headers_datasets = ["🔢 Serial Number", "👤 Author Name", "📥 Total Downloads", "👍 Total Likes", "📊 Number of Datasets",
+            "📊 Average Downloads per Dataset", "📈 Average Likes per Dataset", "🚀 Most Downloaded Dataset",
+            "📈 Most Download Count", "❤️ Most Liked Dataset", "👍 Most Like Count", "🔥 Trending Dataset",
+            "👑 Best Rank at Trending Datasets", "🏷️ Type"]
+
+headers_spaces = ["🔢 Serial Number", "👤 Author Name", "👍 Total Likes", "🚀 Number of Spaces", "📈 Average Likes per Space",
+            "❤️ Most Liked Space", "👍 Most Like Count", "🔥 Trending Space", "👑 Best Rank at Trending Spaces",
+            "🏷️ Type"]
+
+
+def apply_headers(df, headers):
+    tmp = df.copy()
+    tmp.columns = headers
+
+    return tmp
+
+
 def get_time():
     return datetime.datetime.now().strftime("%d-%m-%Y %H-%M")
 
 
 def upload_datasets(dfs):
+
   time = get_time()
 
   operations = [CommitOperationAdd(path_in_repo=f"{time}/models_df.csv", path_or_fileobj=(dfs[0].to_csv()).encode()),
@@ -320,12 +345,12 @@ def update_table(orgs, users, how_much=400, return_all=False):
         filtered_df = dataFrame[(dataFrame['Type'] == 'Organization') | (dataFrame['Type'] == 'User')]
 
     else:
-        return dataFrame.head(0)
+        return apply_headers(dataFrame.head(0), headers_models)
 
     if return_all:
-        return filtered_df
+        return apply_headers(filtered_df, headers_models)
     else:
-        return filtered_df.head(how_much)
+        return apply_headers(filtered_df, headers_models).head(how_much)
 
 
 def update_table_datasets(orgs, users, how_much=250, return_all=False):
@@ -341,12 +366,12 @@ def update_table_datasets(orgs, users, how_much=250, return_all=False):
         filtered_df = dataFrame[(dataFrame['Type'] == 'Organization') | (dataFrame['Type'] == 'User')]
 
     else:
-        return dataFrame.head(0)
+        return apply_headers(dataFrame, headers_datasets).head(0)
 
     if return_all:
-        return filtered_df
+        return apply_headers(filtered_df, headers_datasets)
     else:
-        return filtered_df.head(how_much)
+        return apply_headers(filtered_df, headers_datasets).head(how_much)
 
 
 def update_table_spaces(orgs, users, how_much=200, return_all=False):
@@ -362,12 +387,12 @@ def update_table_spaces(orgs, users, how_much=200, return_all=False):
         filtered_df = dataFrame[(dataFrame['Type'] == 'Organization') | (dataFrame['Type'] == 'User')]
 
     else:
-        return dataFrame.head(0)
+        return apply_headers(dataFrame, headers_spaces).head(0)
 
     if return_all:
-        return filtered_df
+        return apply_headers(filtered_df, headers_spaces)
     else:
-        return filtered_df.head(how_much)
+        return apply_headers(filtered_df, headers_spaces).head(how_much)
 
 
 
@@ -477,7 +502,6 @@ def search_df(author):
   return markdown_text
 
 
-
 with gr.Blocks() as demo:
     gr.Markdown("""<h1 align="center" id="space-title">🤗 Huggingface Leaderboard</h1>""")
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
@@ -497,13 +521,7 @@ with gr.Blocks() as demo:
         models_df = make_leaderboard(org_names_in_list, user_names_in_list, "models", group_models_by_author(all_models))
         models_df = models_df_to_clickable(models_df, columns_to_convert, "models")
 
-        headers = ["🔢 Serial Number", "👤 Author Name", "📥 Total Downloads", "👍 Total Likes", "🤖 Number of Models",
-                   "🏆 Best Model On Open LLM Leaderboard", "🥇 Best Rank On Open LLM Leaderboard",
-                   "📊 Average Downloads per Model", "📈 Average Likes per Model", "🚀 Most Downloaded Model",
-                   "📈 Most Download Count", "❤️ Most Liked Model", "👍 Most Like Count", "🔥 Trending Model",
-                   "👑 Best Rank at Trending Models", "🏷️ Type"]
-
-        gr_models = gr.Dataframe(models_df.head(400), headers=headers, interactive=True,
+        gr_models = gr.Dataframe(apply_headers(models_df, headers_models).head(400), headers=headers_models, interactive=True,
                                  datatype=["str", "markdown", "str", "str", "str", "markdown", "str", "str", "str",
                                            "markdown", "str", "markdown", "str", "markdown", "str", "str"])
 
@@ -512,12 +530,7 @@ with gr.Blocks() as demo:
         dataset_df = make_leaderboard(org_names_in_list, user_names_in_list, "datasets", group_models_by_author(all_datasets))
         dataset_df = models_df_to_clickable(dataset_df, columns_to_convert, "datasets")
 
-        headers = ["🔢 Serial Number", "👤 Author Name", "📥 Total Downloads", "👍 Total Likes", "📊 Number of Datasets",
-                   "📊 Average Downloads per Dataset", "📈 Average Likes per Dataset", "🚀 Most Downloaded Dataset",
-                   "📈 Most Download Count", "❤️ Most Liked Dataset", "👍 Most Like Count", "🔥 Trending Dataset",
-                   "👑 Best Rank at Trending Datasets", "🏷️ Type"]
-
-        gr_datasets = gr.Dataframe(dataset_df.head(250), headers=headers, interactive=False,
+        gr_datasets = gr.Dataframe(apply_headers(dataset_df, headers_datasets).head(250), headers=headers_datasets, interactive=False,
                                    datatype=["str", "markdown", "str", "str", "str", "str", "str", "markdown", "str",
                                              "markdown", "str", "markdown", "str", "str"])
 
@@ -527,11 +540,7 @@ with gr.Blocks() as demo:
         spaces_df = make_leaderboard(org_names_in_list, user_names_in_list, "spaces", group_models_by_author(all_spaces))
         spaces_df = models_df_to_clickable(spaces_df, columns_to_convert, "spaces")
 
-        headers = ["🔢 Serial Number", "👤 Author Name", "👍 Total Likes", "🚀 Number of Spaces", "📈 Average Likes per Space",
-                   "❤️ Most Liked Space", "👍 Most Like Count", "🔥 Trending Space", "👑 Best Rank at Trending Spaces",
-                   "🏷️ Type"]
-
-        gr_spaces = gr.Dataframe(spaces_df.head(200), headers=headers, interactive=False,
+        gr_spaces = gr.Dataframe(apply_headers(spaces_df, headers_spaces).head(200), headers=headers_spaces, interactive=False,
                                  datatype=["str", "markdown", "str", "str", "str", "markdown", "str", "markdown", "str",
                                            "str"])
 
@@ -563,13 +572,14 @@ with gr.Blocks() as demo:
     users.change(fn=update_table_spaces, inputs=[orgs, users], outputs=gr_spaces)
 
 
-filtered_model_users = update_table(orgs=False, users=True, return_all=True)['Author Name'].tolist()
-filtered_model_orgs = update_table(orgs=True, users=False, return_all=True)['Author Name'].tolist()
+filtered_model_users = update_table(orgs=False, users=True, return_all=True)['👤 Author Name'].tolist()
+filtered_model_orgs = update_table(orgs=True, users=False, return_all=True)['👤 Author Name'].tolist()
+
+filtered_datasets_users = update_table_datasets(orgs=False, users=True, return_all=True)['👤 Author Name'].tolist()
+filtered_datasets_orgs = update_table_datasets(orgs=True, users=False, return_all=True)['👤 Author Name'].tolist()
 
-filtered_datasets_users = update_table_datasets(orgs=False, users=True, return_all=True)['Author Name'].tolist()
-filtered_datasets_orgs = update_table_datasets(orgs=True, users=False, return_all=True)['Author Name'].tolist()
+filtered_spaces_users = update_table_spaces(orgs=False, users=True, return_all=True)['👤 Author Name'].tolist()
+filtered_spaces_orgs = update_table_spaces(orgs=True, users=False, return_all=True)['👤 Author Name'].tolist()
 
-filtered_spaces_users = update_table_spaces(orgs=False, users=True, return_all=True)['Author Name'].tolist()
-filtered_spaces_orgs = update_table_spaces(orgs=True, users=False, return_all=True)['Author Name'].tolist()
+demo.launch(debug=True)
 
-demo.launch(debug=True)