Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import pandas as pd
 from gradio_huggingfacehub_search import HuggingfaceHubSearch
 import requests
+from huggingface_hub import HfApi
 
 leaderboard_url = os.getenv("LEADERBOARD_URL", "https://leaderboard.nexa4ai.com")
 
@@ -11,6 +12,12 @@ get_models_url = f"{leaderboard_url}/model/get-models-by-category"
 vote_url = f"{leaderboard_url}/model/vote"
 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}"
     
@@ -21,13 +28,15 @@ def update_table(category):
         api_model_data = {
             "Model": [item["model_id"] for item in ranking_data],
             "Votes": [item["votes"] for item in ranking_data],
-            "Categories": [category] * len(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 api_df
 
 
 def get_user(profile: gr.OAuthProfile | None) -> str:
@@ -74,13 +83,17 @@ def submit_vote(username, category, models):
         return f"Failed to vote: {response.text}"
 
 
-def submit_model(category, model_id):
-    if not category or not model_id:
+def submit_model(category, customize_category, model_id):
+    if not model_id:
         return "All fields are required!"
+    if not category or not customize_category:
+        return "Please choose a cateogry or customize your own category!"
     
+    selected_category = customize_category if customize_category else category
+
     data = {
         "model_id": model_id,
-        "categories": [category]  
+        "categories": [selected_category]
     }
     
     response = requests.post(submit_models_url, json=data)
@@ -90,6 +103,10 @@ def submit_model(category, model_id):
     else:
         return f"Failed to submit request: {response.text}"
 
+def update_dropdown(customize_category, category):
+    if customize_category:  
+        return "", gr.update(value="") 
+    return category, gr.update() 
 
 theme = gr.themes.Soft()
 
@@ -132,13 +149,11 @@ 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")
-            model_id = HuggingfaceHubSearch(
-            label="Hub Model ID",
-            placeholder="Search for model id on Huggingface",
-            search_type="model",
-        )
+            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")
             submit_model_result = gr.Markdown()
-            submit_model_button.click(fn=submit_model, inputs=[category, model_id], outputs=submit_model_result)
+            customize_category.change(fn=update_dropdown, inputs=[customize_category, category], outputs=[category])
+            submit_model_button.click(fn=submit_model, inputs=[category, customize_category, model_id], outputs=submit_model_result)
 
 app.launch(share=True)