|
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 |
|
class Contributions: |
|
additions: int |
|
deletions: int |
|
descriptions: list[str] |
|
|
|
|
|
def get_release_notes( |
|
tag: str, |
|
branch: str, |
|
contributor_treshhold: int, |
|
ignore_dependabot: bool, |
|
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: |
|
return str(e) |
|
|
|
result = "" |
|
contributors = {} |
|
|
|
for commit in commits[::-1]: |
|
if "Hugging Face" not in commit.user.organizations: |
|
if commit.user.name not in contributors: |
|
contributors[commit.user.name] = Contributions( |
|
additions=commit.additions, |
|
deletions=commit.deletions, |
|
descriptions=[commit.message], |
|
) |
|
else: |
|
contributors[commit.user.name].additions += commit.additions |
|
contributors[commit.user.name].deletions += commit.deletions |
|
contributors[commit.user.name].descriptions += [commit.message] |
|
|
|
if "(#" in commit.message: |
|
if ignore_dependabot and commit.user.name == "dependabot[bot]": |
|
continue |
|
|
|
split = commit.message.split("(#") |
|
message = split[0] |
|
number = split[1].strip(")") |
|
|
|
result += f"* {message} by @{commit.user.name} in #{number}\n" |
|
elif not ignore_direct: |
|
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>" |
|
} |
|
|
|
if len(significant_contributors): |
|
result += ( |
|
"\n## Significant community contributions\n" |
|
"\nThe following contributors have made significant " |
|
"changes to the library over the last release:\n\n" |
|
) |
|
|
|
for significant_contributor, contributions in significant_contributors.items(): |
|
result += f"* @{significant_contributor}\n" |
|
for description in contributions.descriptions: |
|
result += f" * {description}\n" |
|
|
|
return result |
|
|
|
|
|
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 for 🤗 Diffusers", |
|
article=article, |
|
description="**See instructions below the form.**", |
|
fn=get_release_notes, |
|
inputs=[ |
|
gr.components.Textbox( |
|
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 linear branch on which the new version tag will be added", |
|
value=github_info["latest_release_branch"] if github_info else None, |
|
), |
|
gr.components.Slider( |
|
minimum=0, |
|
maximum=2000, |
|
value=500, |
|
label="Threshold for significant contributors", |
|
), |
|
gr.components.Checkbox(label="Ignore dependabot commits"), |
|
gr.components.Checkbox(label="Ignore direct commits"), |
|
], |
|
outputs="text", |
|
allow_flagging="never" |
|
) |
|
|
|
|
|
if __name__ == "__main__": |
|
demo.launch(show_error=True) |
|
|
