add

app.py

@@ -1,7 +1,8 @@
 from dataclasses import dataclass
-
+import os
 import gradio as gr
 from graphql_calls import get_tag_commit_date, get_commits
+from github import fetch_github_info
 
 
 @dataclass
@@ -12,8 +13,6 @@ class Contributions:
 
 
 def get_release_notes(
-    token: str,
-    repo: str,
     tag: str,
     branch: str,
     contributor_treshhold: int,
@@ -21,6 +20,8 @@ def get_release_notes(
     ignore_direct: bool,
 ):
     try:
+        token = os.getenv("GITHUB_TOKEN")
+        repo = "huggingface/diffusers"
         date = get_tag_commit_date(token, repo, tag)
         commits = get_commits(token, repo, branch, date)
     except ValueError as e:
@@ -43,7 +44,7 @@ def get_release_notes(
                 contributors[commit.user.name].descriptions += [commit.message]
 
         if "(#" in commit.message:
-            if ignore_dependabot and commit.user.name == 'dependabot[bot]':
+            if ignore_dependabot and commit.user.name == "dependabot[bot]":
                 continue
 
             split = commit.message.split("(#")
@@ -55,7 +56,9 @@ def get_release_notes(
             result += f"* {commit.message} by @{commit.user.name} (direct commit on {branch})\n"
 
     significant_contributors = {
-        k: v for k, v in contributors.items() if (v.additions + v.deletions) > contributor_treshhold and k != '<NOT FOUND>'
+        k: v
+        for k, v in contributors.items()
+        if (v.additions + v.deletions) > contributor_treshhold and k != "<NOT FOUND>"
     }
 
     if len(significant_contributors):
@@ -72,54 +75,28 @@ def get_release_notes(
 
     return result
 
-article = """
-## How to use the interface?
-
-⚠️ Most errors are due to:
-- A wrong `tag` -> the tag must exist on the repository!
-- A wrong `branch` -> The branch must exist on the repository!
-- A wrong `token` -> Obtaining a token is detailed below.
-
-### token
-
-This is a personal access token generated by GitHub. You can obtain one with the following guide:
-https://docs.github.com/en/authentication/keeping-your-account-and-data-secure/creating-a-personal-access-token.
-
-### Repository
-
-The repository for which to create the release notes. Should be in the format `organization/repository`.
-
-### Tag
 
-The tag from which all commits were made. The app will fetch the date at which this tag was created, and will
-return all commits of the branch defined below that have a timestamp following that date.
-
-### Branch
-
-The branch on which all commits lie. Usually `master` or `main`.
-
-### Threshold
-
-This threshold allows highlighting specific community contributors according to the size of their contributions.
-It currently adds all their additions/deletions across all PRs included in this release. It is then compared
-to the defined threshold: if above, that user will get a special note mentioning what they have worked on!
+article = """
+Mostly copied from [lysandre/github-release](https://huggingface.co/spaces/lysandre/github-release) but suited for
+Diffusers.
 """
+github_info = fetch_github_info("https://github.com/huggingface/diffusers")
 
 demo = gr.Interface(
-    title='Automatic release notes 🤗',
+    title="Automatic release notes for 🤗 Diffusers",
     article=article,
     description="**See instructions below the form.**",
     fn=get_release_notes,
     inputs=[
-        gr.components.Textbox(lines=1, placeholder="Your GitHub token"),
         gr.components.Textbox(
-            lines=1, placeholder="Repository", value="huggingface/transformers"
+            lines=1,
+            value=github_info["second_last_tag"] if github_info else None,
+            placeholder="The tag from which to get commit",
         ),
-        gr.components.Textbox(lines=1, placeholder="The tag from which to get commit"),
         gr.components.Textbox(
             lines=1,
             placeholder="The linear branch on which the new version tag will be added",
-            value="main",
+            value=github_info["latest_release_branch"] if github_info else None,
         ),
         gr.components.Slider(
             minimum=0,
@@ -131,13 +108,6 @@ demo = gr.Interface(
         gr.components.Checkbox(label="Ignore direct commits"),
     ],
     outputs="text",
-    examples=[
-        ['ghp_XXX', 'huggingface/datasets', '2.1.0', 'master', 60, True, True],
-        ['ghp_XXX', 'huggingface/accelerate', 'v0.7.1', 'main', 500, True, True],
-        ['ghp_XXX', 'huggingface/transformers', 'v4.18.0', 'main', 500, True, True],
-        ['ghp_XXX', 'huggingface/diffusers', 'v0.17.0', 'main', 500, True, True],
-        ['ghp_XXX', 'gradio-app/gradio', 'v2.9.0', 'main', 500, True, True],
-    ],
 )
 
 

github.py

@@ -0,0 +1,89 @@
+import requests
+from functools import lru_cache
+from packaging.version import parse
+import os
+
+TOKEN = os.getenv("GITHUB_TOKEN", None)
+HEADERS = {"Authorization": f"token {TOKEN}"} if TOKEN else {}
+
+
+@lru_cache()
+def fetch_all_branches(user, repo):
+    branches = []  # List to store all branches
+    page = 1  # Start from first page
+    while True:
+        # Make a request to the GitHub API for the branches
+        response = requests.get(
+            f"https://api.github.com/repos/{user}/{repo}/branches", params={"page": page}, headers=HEADERS
+        )
+
+        # Check if the request was successful
+        if response.status_code == 200:
+            # Add the branches from the current page to the list
+            branches.extend([branch["name"] for branch in response.json()])
+
+            # Check if there is a 'next' link for pagination
+            if "next" in response.links:
+                page += 1  # Move to the next page
+            else:
+                break  # Exit loop if there is no next page
+        else:
+            print("Failed to retrieve branches:", response.status_code)
+            break
+
+    return branches
+
+
+def fetch_github_info(repo_url):
+    # Extract owner and repo name from the URL
+    repo_parts = repo_url.rstrip("/").split("/")
+    owner, repo = repo_parts[-2], repo_parts[-1]
+    print(f"{owner=}, {repo=}")
+
+    # GitHub API base URL
+    api_url = f"https://api.github.com/repos/{owner}/{repo}"
+
+    try:
+        # Fetch tags
+        tags_response = requests.get(f"{api_url}/tags", headers=HEADERS)
+        tags_response.raise_for_status()
+        tags = tags_response.json()
+
+        if len(tags) < 2:
+            raise ValueError("Not enough tags to fetch the second last tag.")
+
+        second_last_tag = tags[1]["name"]
+
+        # Fetch branches
+        branches = fetch_all_branches(user=owner, repo=repo)
+        if not branches:
+            raise ValueError("No branches found.")
+
+        filtered_branches = []
+        for branch in branches:
+            if branch.startswith("v") and ("-release" in branch or "-patch" in branch):
+                filtered_branches.append(branch)
+        if not filtered_branches:
+            raise ValueError("No release branches found.")
+
+        sorted_branches = sorted(filtered_branches, key=lambda x: parse(x.split("-")[0][1:]), reverse=True)
+        latest_release_branch = sorted_branches[0]
+
+        return {"second_last_tag": second_last_tag, "latest_release_branch": latest_release_branch}
+
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching data from GitHub: {e}")
+        return None
+    except ValueError as e:
+        print(e)
+        return None
+
+
+# Example usage
+if __name__ == "__main__":
+    repo_url = "https://github.com/huggingface/diffusers"
+    info = fetch_github_info(repo_url)
+
+    if info:
+        print("Second Last Tag:", info["second_last_tag"])
+        print("Latest Release Branch:", info["latest_release_branch"])

graphql_calls.py

@@ -18,13 +18,13 @@ class Commit:
 
 
 def call_with_query(query, token):
-    url = 'https://api.github.com/graphql'
-    r = requests.post(url, json={'query': query}, headers={'Authorization': f'Bearer {token}'})
+    url = "https://api.github.com/graphql"
+    r = requests.post(url, json={"query": query}, headers={"Authorization": f"Bearer {token}"})
     return r.json()
 
 
 def get_tag_commit_date(token, repository, tag):
-    owner, name = repository.split('/')
+    owner, name = repository.split("/")
     query = f"""
     query GetTagCommit {{
         repository(owner: "{owner}", name: "{name}"){{
@@ -47,24 +47,24 @@ def get_tag_commit_date(token, repository, tag):
     response = call_with_query(query, token)
 
     try:
-        repository = response['data']['repository']['object']
+        repository = response["data"]["repository"]["object"]
 
         if repository is None:
-            if 'errors' in response:
-                raise ValueError(response['errors'][0]['message'])
-            raise ValueError('Invalid tag. Does this tag exist?')
+            if "errors" in response:
+                raise ValueError(response["errors"][0]["message"])
+            raise ValueError("Invalid tag. Does this tag exist?")
 
-        committed_date = repository['committedDate']
+        committed_date = repository["committedDate"]
     except (KeyError, TypeError):
-        raise ValueError('Invalid token. Does your token have the valid permissions?')
+        raise ValueError("Invalid token. Does your token have the valid permissions?")
 
     return committed_date
 
 
 def get_commits(token, repository, branch, since):
-    owner, name = repository.split('/')
+    owner, name = repository.split("/")
 
-    def get_page_result(next_page=''):
+    def get_page_result(next_page=""):
         query = f"""
         query GetCommits {{
             repository(owner: "{owner}", name: "{name}"){{
@@ -100,14 +100,14 @@ def get_commits(token, repository, branch, since):
         """
         result = call_with_query(query, token)
 
-        if 'data' not in result:
-            raise ValueError(result['errors'][0]['message'])
+        if "data" not in result:
+            raise ValueError(result["errors"][0]["message"])
 
-        if result['data']['repository']['object'] is None:
+        if result["data"]["repository"]["object"] is None:
             raise ValueError("Either the tag or the branch were incorrect.")
 
-        nodes = result['data']['repository']['object']['history']['nodes']
-        cursor = result['data']['repository']['object']['history']['pageInfo']['endCursor']
+        nodes = result["data"]["repository"]["object"]["history"]["nodes"]
+        cursor = result["data"]["repository"]["object"]["history"]["pageInfo"]["endCursor"]
         return nodes, cursor
 
     nodes, cursor = get_page_result()
@@ -116,22 +116,28 @@ def get_commits(token, repository, branch, since):
         _nodes, cursor = get_page_result(f' after:"{cursor}"')
         nodes.extend(_nodes)
 
-
     commits = []
     for node in nodes:
-        if node['author']['user'] is None:
-            commits.append(Commit(
-                message=node['message'].split('\n')[0],
-                user=User(name='<NOT FOUND>', organizations=[]),
-                additions=node.get('additions'),
-                deletions=node.get('deletions')
-            ))
+        if node["author"]["user"] is None:
+            commits.append(
+                Commit(
+                    message=node["message"].split("\n")[0],
+                    user=User(name="<NOT FOUND>", organizations=[]),
+                    additions=node.get("additions"),
+                    deletions=node.get("deletions"),
+                )
+            )
         else:
-            commits.append(Commit(
-                message=node['message'].split('\n')[0],
-                user=User(name=node['author']['user']['login'], organizations=[n['name'] for n in node['author']['user']['organizations']['nodes']]),
-                additions=node.get('additions'),
-                deletions=node.get('deletions')
-            ))
+            commits.append(
+                Commit(
+                    message=node["message"].split("\n")[0],
+                    user=User(
+                        name=node["author"]["user"]["login"],
+                        organizations=[n["name"] for n in node["author"]["user"]["organizations"]["nodes"]],
+                    ),
+                    additions=node.get("additions"),
+                    deletions=node.get("deletions"),
+                )
+            )
 
     return commits