Login

app.py

@@ -2,20 +2,55 @@ import gradio as gr
 from df.PaperCentral import PaperCentral
 from gradio_calendar import Calendar
 from datetime import datetime, timedelta
-from typing import Union, List, Optional
+from typing import Union, List, Optional, Tuple
 from author_leaderboard_tab import author_leaderboard_tab
+from pr_paper_central_tab import pr_paper_central_tab
+from huggingface_hub import whoami
+import json
+
 from author_leaderboard_contrib_tab import author_resource_leaderboard_tab
 from zoneinfo import ZoneInfo  # Available in Python 3.9 and later
 
 # Initialize the PaperCentral class instance
 paper_central_df = PaperCentral()
 
+
+def hide_if_logged_in(oauth_token: Optional[gr.OAuthToken]):
+    if oauth_token is None:
+        return gr.update(visible=True)
+    return gr.update(visible=False)
+
+
+def display_user_info(oauth_token: Optional[gr.OAuthToken]):
+    if oauth_token is None:
+        return ""
+    user_info = whoami(oauth_token.token)
+    avatar_url = user_info.get("avatarUrl", "")
+    fullname = user_info.get("fullname", "")
+    name = user_info.get("name", "")
+
+    # Prepare the custom HTML for the avatar
+    avatar_html = \
+        f'''
+        <div style="display: flex; align-items: center;">   
+                <img alt="Avatar" src="{avatar_url}" 
+                     style="border: 2px solid rgb(245, 158, 11); border-radius: 50%; width: 1.38rem; height: 1.38rem; margin-right: 0.5rem;">
+                <span style="font-size: 1rem;">{name}</span>
+            </div>
+        Head to "PR paper-central" tab to modify your paper!
+        '''
+
+    return gr.update(value=avatar_html, label="User")
+
+
 # Create the Gradio Blocks app with custom CSS
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown("# Paper Central")
-
-    with gr.Tabs() as tabs:
-        with gr.Tab("Paper-central", id="tab-paper-central"):
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.LoginButton(value="PR this space").attach_load_event(hide_if_logged_in, None)
+            user_info_md = gr.HTML().attach_load_event(display_user_info, None)
+        with gr.Column(scale=1):
             with gr.Accordion(label="⭐Release notes", open=False):
                 gr.Markdown("""
                 - **October 16, 2024** – Added functionality to filter papers by date ranges.
@@ -25,6 +60,8 @@ with gr.Blocks(css="style.css") as demo:
                 - **October 4, 2024** – Added functionality to filter papers by title.
                 """)
 
+    with gr.Tabs() as tabs:
+        with gr.Tab("Paper-central", id="tab-paper-central"):
             # Create a row for navigation buttons and calendar
             with gr.Row():
                 with gr.Column(scale=1):
@@ -119,6 +156,9 @@ with gr.Blocks(css="style.css") as demo:
                 wrap=True,
             )
 
+        with gr.Tab("PR paper-central", id="tab-pr"):
+            pr_paper_central_tab(paper_central_df.df_raw)
+
         with gr.Tab("Leaderboards", id="tab-leaderboards"):
             with gr.Tab("Authors"):
                 author_leaderboard_tab()

df/PaperCentral.py

@@ -425,12 +425,12 @@ class PaperCentral:
             if 'id' not in columns_to_show:
                 columns_to_show.append('id')
 
-            if "In proceedings" in conference_options:
+            if "ALL" in conference_options:
                 filtered_df = filtered_df[
                     filtered_df['conference_name'].notna() & (filtered_df['conference_name'] != "")
                     ]
 
-            other_conferences = [conf for conf in conference_options if conf != "In proceedings"]
+            other_conferences = [conf for conf in conference_options if conf != "ALL"]
             if other_conferences:
                 conference_filter = pd.Series(False, index=filtered_df.index)
                 for conference in other_conferences:

pr_paper_central_tab.py

@@ -0,0 +1,165 @@
+import gradio as gr
+from typing import Optional
+import pandas as pd
+from huggingface_hub import HfApi, hf_hub_download, CommitOperationAdd
+import json
+import os
+
+
+# PR function as before
+def create_pr_in_hf_dataset(new_entry, oauth_token: gr.OAuthToken):
+    # Dataset and filename
+    REPO_ID = 'IAMJB/paper-central-pr'
+    FILENAME = 'data.json'
+
+    # Initialize HfApi
+    api = HfApi()
+    token = oauth_token.token
+
+    # Ensure the repository exists and has an initial empty data.json if not present
+    try:
+        # Create the repository if it doesn't exist
+        api.create_repo(repo_id=REPO_ID, token=token, repo_type='dataset', exist_ok=True)
+
+        # Check if data.json exists; if not, create it with empty list
+        files = api.list_repo_files(REPO_ID, repo_type='dataset', token=token)
+        if FILENAME not in files:
+            # Initialize with empty list
+            empty_data = []
+            temp_filename = 'temp_data.json'
+            with open(temp_filename, 'w') as f:
+                json.dump(empty_data, f)
+            commit = CommitOperationAdd(path_in_repo=FILENAME, path_or_fileobj=temp_filename)
+            api.create_commit(
+                repo_id=REPO_ID,
+                operations=[commit],
+                commit_message="Initialize data.json",
+                repo_type="dataset",
+                token=token,
+            )
+            os.remove(temp_filename)
+    except Exception as e:
+        return f"Error creating or accessing repository: {e}"
+
+    # Download existing data from the dataset
+    try:
+        # Download the existing data.json file
+        local_filepath = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type='dataset', token=token)
+        with open(local_filepath, 'r') as f:
+            data = json.load(f)
+    except Exception as e:
+        print(f"Error downloading existing data: {e}")
+        data = []
+
+    # Add the new entry
+    data.append(new_entry)
+
+    # Save to temporary file
+    temp_filename = 'temp_data.json'
+    with open(temp_filename, 'w') as f:
+        json.dump(data, f, indent=2)
+
+    # Create commit operation
+    commit = CommitOperationAdd(path_in_repo=FILENAME, path_or_fileobj=temp_filename)
+
+    # Create PR
+    try:
+        res = api.create_commit(
+            repo_id=REPO_ID,
+            operations=[commit],
+            commit_message=f"Add new entry for arXiv ID {new_entry['arxiv_id']}",
+            repo_type="dataset",
+            create_pr=True,
+            token=token,
+        )
+        pr_url = res.pr_url
+        os.remove(temp_filename)
+    except Exception as e:
+        print(f"Error creating PR: {e}")
+        pr_url = "Error creating PR."
+
+    return pr_url
+
+
+def pr_paper_central_tab(paper_central_df):
+    with gr.Column():
+        gr.Markdown("## PR Paper-central")
+
+        # Message to prompt user to log in
+        login_prompt = gr.Markdown("Please log in to proceed.", visible=False)
+
+        # Input for arXiv ID
+        arxiv_id_input = gr.Textbox(label="Enter arXiv ID")
+        arxiv_id_button = gr.Button("Submit")
+
+        # Message to display errors or information
+        message = gr.Markdown("", visible=False)
+
+        # Define the fields dynamically
+        fields = [
+            {'name': 'paper_page', 'label': 'Paper Page'},
+            {'name': 'github', 'label': 'GitHub URL'},
+            {'name': 'conference_name', 'label': 'Conference Name'},
+            {'name': 'type_', 'label': 'Type'},  # Renamed from 'type' to 'type_'
+            {'name': 'proceedings', 'label': 'Proceedings'},
+            # Add or remove fields here as needed
+        ]
+
+        input_fields = {}
+        for field in fields:
+            input_fields[field['name']] = gr.Textbox(label=field['label'], visible=False)
+
+        # Button to create PR
+        create_pr_button = gr.Button("Create PR", visible=False)
+
+        # Output message
+        pr_message = gr.Markdown("", visible=False)
+
+        # Function to handle arxiv_id submission and check login
+        def check_login_and_handle_arxiv_id(arxiv_id, oauth_token: Optional[gr.OAuthToken]):
+            if oauth_token is None:
+                return [gr.update(value="Please log in to proceed.", visible=True)] + \
+                    [gr.update(visible=False) for _ in fields] + \
+                    [gr.update(visible=False)]  # create_pr_button
+            else:
+                if arxiv_id not in paper_central_df['arxiv_id'].values:
+                    return [gr.update(value="arXiv ID not found. Please try again.", visible=True)] + \
+                        [gr.update(visible=False) for _ in fields] + [gr.update(visible=False)]  # create_pr_button
+                else:
+                    row = paper_central_df[paper_central_df['arxiv_id'] == arxiv_id].iloc[0]
+                    updates = [gr.update(value="", visible=False)]  # message
+                    for field in fields:
+                        value = row.get(field['name'], "")
+                        updates.append(gr.update(value=value, visible=True))
+                    updates.append(gr.update(visible=True))  # create_pr_button
+                    return updates
+
+        arxiv_id_button.click(
+            fn=check_login_and_handle_arxiv_id,
+            inputs=[arxiv_id_input],
+            outputs=[message] + [input_fields[field['name']] for field in fields] + [create_pr_button]
+        )
+
+        # Function to create PR
+        def create_pr(message, arxiv_id, paper_page, github, conference_name, type_, proceedings,
+                      oauth_token: Optional[gr.OAuthToken] = None):
+            if oauth_token is None:
+                return gr.update(value="Please log in first.", visible=True)
+            else:
+                new_entry = {
+                    'arxiv_id': arxiv_id,
+                    'paper_page': paper_page,
+                    'github': github,
+                    'conference_name': conference_name,
+                    'type': type_,
+                    'proceedings': proceedings
+                }
+                # Now add this to the dataset and create a PR
+                pr_url = create_pr_in_hf_dataset(new_entry, oauth_token)
+                return gr.update(value=f"PR created: {pr_url}", visible=True)
+
+        create_pr_button.click(
+            fn=create_pr,
+            inputs=[pr_message, arxiv_id_input] + [input_fields[field['name']] for field in fields],
+            outputs=[pr_message]
+        )

style.css

@@ -21,15 +21,40 @@ body a:hover {
 }
 
 
+#login-row {
+    justify-content: center;
+    display: flex;
+}
+
+
+
+
+.avatar-container {
+    position: relative;
+    margin-left: 0.5rem;
+    width: 1.38rem;
+    height: 1.38rem;
+}
+
+.avatar-button {
+    margin-left: auto;
+    border-radius: 9999px;
+    ring: 2px solid rgb(245 158 11);
+    outline: 2px solid transparent;
+    outline-offset: 2px;
+    background: none;
+    border: none;
+    padding: 0;
+    cursor: pointer;
+}
 
-#hf_options label[for='github']::before {
-    content: "";
-    background-image: url('https://github.githubassets.com/favicons/favicon.png');
-    background-size: contain;
-    display: inline-block;
-    width: 16px;
-    height: 16px;
-    vertical-align: middle;
-    margin-right: 5px;
+.avatar-button:hover {
+    ring-offset: 1px;
 }
 
+.avatar-image {
+    width: 1.38rem;
+    height: 1.38rem;
+    overflow: hidden;
+    border-radius: 9999px;
+}