Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import time
 os.system("wget https://raw.githubusercontent.com/Weyaxi/scrape-open-llm-leaderboard/main/openllm.py")
 from huggingface_hub import CommitOperationAdd, create_commit, HfApi, HfFileSystem, login
 from huggingface_hub import ModelCardData, EvalResult, ModelCard
@@ -12,12 +13,6 @@ import pandas as pd
 from pytablewriter import MarkdownTableWriter
 import gradio as gr
 
-from gradio_space_ci import enable_space_ci
-
-enable_space_ci()
-
-
-
 api = HfApi()
 fs = HfFileSystem()
 
@@ -139,8 +134,8 @@ Detailed results can be found [here]({get_details_url(repo)})
   return text
 
 
-def get_edited_yaml_readme(repo):
-  card = ModelCard.load(repo)
+def get_edited_yaml_readme(repo, token: str | None):
+  card = ModelCard.load(repo, token=token)
   results = search(df, repo)
 
   common = {"task_type": 'text-generation', "task_name": 'Text Generation', "source_name": "Open LLM Leaderboard", "source_url": f"https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard?query={repo}"}
@@ -159,21 +154,26 @@ def get_edited_yaml_readme(repo):
   return str(card)
     
 
-def commit(hf_token, repo, pr_number=None, message="Adding Evaluation Results"): # specify pr number if you want to edit it, don't if you don't want
-  login(hf_token)
+def commit(repo, pr_number=None, message="Adding Evaluation Results", oauth_token: gr.OAuthToken | None = None): # specify pr number if you want to edit it, don't if you don't want
+  if oauth_token is None:
+    raise gr.Error("You must be logged in to open a PR. Click on 'Sign in with Huggingface' first.")
+  if oauth_token.expires_at < time.time():
+    raise gr.Error("Token expired. Logout and try again.")
+  token = oauth_token.token
+    
   edited = {"revision": f"refs/pr/{pr_number}"} if pr_number else {"create_pr": True}
 
   try:
     try: # check if there is a readme already
-      readme_text = get_edited_yaml_readme(repo) + get_eval_results(repo)
+      readme_text = get_edited_yaml_readme(repo, token=token) + get_eval_results(repo)
     except Exception as e:
       if "Repo card metadata block was not found." in str(e): # There is no readme
-        readme_text = get_edited_yaml_readme(repo)
+        readme_text = get_edited_yaml_readme(repo, token=token)
       else:
         print(f"Something went wrong: {e}")
 
     liste = [CommitOperationAdd(path_in_repo="README.md", path_or_fileobj=readme_text.encode())]
-    commit = (create_commit(repo_id=repo, operations=liste, commit_message=message, commit_description=desc, repo_type="model", **edited).pr_url)
+    commit = (create_commit(repo_id=repo, token=token, operations=liste, commit_message=message, commit_description=desc, repo_type="model", **edited).pr_url)
 
     return commit
 
@@ -209,6 +209,25 @@ The leaderboard's backend mainly runs on the [Hugging Face Hub API](https://hugg
 - Special thanks to [Lucain Pouget (Wauplin)](https://huggingface.co/Wauplin) for assisting with the [Hugging Face Hub API](https://huggingface.co/docs/huggingface_hub/v0.5.1/en/package_reference/hf_api).
 """
 
-demo = gr.Interface(title=gradio_title, description=gradio_desc, fn=commit, inputs=["text", "text"], outputs="text")
+with gr.Blocks() as demo:
+    gr.Markdown(f"""<h1 align="center" id="space-title">{gradio_title}</h1>""")
+    gr.Markdown(gradio_desc)
+
+    with gr.Row(equal_height=False):
+        with gr.Column():
+            space_id = gr.Textbox(label="Model ID or URL", lines=1)
+            gr.LoginButton()
+
+        with gr.Column():
+            output = gr.Textbox(label="Output", lines=1)
+            gr.LogoutButton()
+
+    submit_btn = gr.Button("Submit", variant="primary")
+    try:
+        space_id = "/".join(space_id.split('/')[-2::]) if space_id.startswith("https://huggingface.co/") else space_id
+    except Exception as e:
+        gr.Error(f"There is an error with your model ID, check it again: {e}")
+        
+    submit_btn.click(commit, space_id, output)
 
 demo.launch()