Add user info

yourbench_space/app.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, whoami
 import requests
 import pathlib
 import subprocess
@@ -8,7 +8,6 @@ import shutil
 import io
 import yaml
 import gradio as gr
-from huggingface_hub import whoami
 from loguru import logger
 from yourbench.pipeline import run_pipeline
 
@@ -142,9 +141,13 @@ def hello(profile: gr.OAuthProfile | None) -> str:
 def update_hf_org_dropdown(oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
         print("Please deploy this on Spaces and log in to list organizations.")
-        return []
-    org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]]
-    return gr.Dropdown(org_names, label="Organization")
+        return list()
+    user_info = whoami(oauth_token.token) 
+
+    org_names = [org["name"] for org in user_info["orgs"]]
+    user_name = user_info["name"]
+    all_orgs = [user_name].extend(org_names)
+    return gr.Dropdown(all_orgs, label="Organization")
 
 config_output = gr.Code(label="Generated Config", language="yaml")
 
@@ -158,10 +161,10 @@ with gr.Blocks() as app:
     with gr.Tab("Configuration"):
         model_name = gr.Textbox(label="Model Name")
 
-        hf_org_dropdown = gr.Dropdown(list(), label="Organization")
+        hf_org_dropdown = gr.Dropdown(list(), label="Organization", allow_custom_value=True)
         app.load(update_hf_org_dropdown, inputs=None, outputs=hf_org_dropdown)
 
-        provider = gr.Dropdown(["openrouter", "openai", "huggingface"], value="huggingface", label="Provider")
+        provider = gr.Dropdown(["openrouter", "openai", "huggingface"], value="huggingface", label="Provider", allow_custom_value=True)
         base_url = gr.Textbox(label="Base URL")
         api_key = gr.Textbox(label="API Key")
         max_concurrent_requests = gr.Dropdown([8, 16, 32], value=16, label="Max Concurrent Requests")