Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import pandas as pd
 # from gradio_huggingfacehub_search import HuggingfaceHubSearch
 import requests
-from huggingface_hub import HfApi
+from urllib.parse import quote
 
 leaderboard_url = os.getenv("LEADERBOARD_URL", "https://leaderboard.nexa4ai.com")
 
@@ -14,29 +14,32 @@ submit_models_url = f"{leaderboard_url}/model/submit-models"
 
 def make_clickable_model(model_id):
     model_name_show = ' '.join(model_id.split('/')[1:])
-
     link = "https://huggingface.co/" + model_id
-    return f'<a target="_blank" href="{link}">{model_name_show}</a>'
-
-def update_table(category): 
-    url_with_params = f"{get_ranking_url}?category={category}"
-    
-    response = requests.get(url_with_params)
-
-    if response.status_code == 200:
-        ranking_data = response.json()["ranking"]
-        api_model_data = {
-            "Model": [item["model_id"] for item in ranking_data],
-            "Votes": [item["votes"] for item in ranking_data],
-            "Categories": [item["categories"] for item in ranking_data] if category == "all" else [category] * len(ranking_data)
-        }
-        api_df = pd.DataFrame(api_model_data)
-        api_df['Model'] = api_df['Model'].apply(make_clickable_model)
-        return api_df
-    else:
-        print(f"Failed to submit request: {response.text}")
-        api_df = pd.DataFrame()  
-        return api_df
+    return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name_show}</a>'
+
+def update_table(category):
+    retries = 3
+    while retries > 0:
+        try:
+            encoded_category = quote(category)
+            url_with_params = f"{get_ranking_url}?category={encoded_category}"
+            response = requests.get(url_with_params)
+            if response.status_code == 200:
+                ranking_data = response.json()["ranking"]
+                api_model_data = {
+                    "Model": [make_clickable_model(item["model_id"]) for item in ranking_data],
+                    "Votes": [item["votes"] for item in ranking_data],
+                    "Categories": [item["categories"] for item in ranking_data] if category == "all" else [category] * len(ranking_data)
+                }
+                api_df = pd.DataFrame(api_model_data)
+                return api_df
+            else:
+                print(f"Failed to submit request: {response.text}")
+                retries -= 1
+        except requests.exceptions.RequestException as e:
+            print(f"Error occurred while making request: {e}")
+            retries -= 1
+    return pd.DataFrame()
 
 
 def get_user(profile: gr.OAuthProfile | None) -> str:
@@ -112,9 +115,9 @@ theme = gr.themes.Soft()
 
 with gr.Blocks(theme=theme) as app:
     with gr.Tabs():
-        with gr.TabItem("Table"):
+        with gr.TabItem("Table"):       
             category = gr.Dropdown(
-                choices=["all", "Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"],
+                choices=["all", "Art & Design", "Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Education", "Health", "Law", "Engineering", "Coding", "Science", "Other"],
                 label="Select Category",
                 value="all"
             )
@@ -122,19 +125,20 @@ with gr.Blocks(theme=theme) as app:
             initial_data = update_table("all")
             table = gr.Dataframe(
                 headers=["Model", "Votes", "Categories"],
-                datatype=["str", "number", "str"],
+                datatype=["markdown", "number", "str"],
                 value=initial_data,
                 col_count=(3, "fixed"),
             )
+            app.load(lambda: update_table("all"), outputs=table)
             category.change(update_table, inputs=category, outputs=table)
-        
+   
         with gr.TabItem("Vote"):
             username_text = gr.State(value="")
             login_button = gr.LoginButton()
             app.load(get_user, inputs=None, outputs=username_text)
 
             category = gr.Dropdown(
-                choices=["Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"], 
+                choices=["Art & Design", "Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Education", "Health", "Law", "Engineering", "Coding", "Science", "Other"], 
                 label="Select Category"
             )
 
@@ -148,7 +152,7 @@ with gr.Blocks(theme=theme) as app:
 
         
         # with gr.TabItem("Submit Model"):
-        #     category = gr.Dropdown(choices=["Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"], label="Select Category")
+        #     category = gr.Dropdown(choices=["Art & Design", "Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Education", "Health", "Law", "Engineering", "Coding", "Science", "Other"], label="Select Category")
         #     customize_category = gr.Textbox(label="Customize category", placeholder="Can't find your category? Enter your own.")
         #     model_id = HuggingfaceHubSearch(label="Hub Model ID", placeholder="Search for model id on Huggingface", search_type="model")
         #     submit_model_button = gr.Button("Submit Model")