diff --git a/scripts/badges.py b/scripts/badges.py index f49683a0b..d4ea8fb5b 100755 --- a/scripts/badges.py +++ b/scripts/badges.py @@ -178,6 +178,61 @@ def get_badges_md(): return generate_badges_md(total_downloads, stars, percentage) +def get_badges_html(): + """ + Get all statistics and return HTML-formatted badges + """ + # Load environment variables from .env file + load_dotenv() + + # Get API key from environment variable + api_key = os.environ.get("PEPY_API_KEY") + if not api_key: + print( + "API key not provided. Please set PEPY_API_KEY environment variable", + file=sys.stderr, + ) + sys.exit(1) + + # Get PyPI downloads for the default package + total_downloads = get_total_downloads(api_key, "aider-chat") + + # Get GitHub stars for the default repo + stars = get_github_stars("paul-gauthier/aider") + + # Get Aider contribution percentage in latest release + percentage, _ = get_latest_release_aider_percentage() + + # Format values + downloads_formatted = format_number(total_downloads) + stars_formatted = format_number(stars) if stars is not None else "0" + aider_percent_rounded = round(percentage) + + # Generate HTML badges + html = f''' + ⭐ GitHub Stars + {stars_formatted} + + + 📦 Installs + {downloads_formatted} + +
+ 📈 Tokens/week + {TOKENS_PER_WEEK} +
+ + 🏆 OpenRouter + Top 20 + + + 🔄 Singularity + {aider_percent_rounded}% +''' + + return html + + def main(): # Load environment variables from .env file load_dotenv()