feat: add AI coding stats from WakaTime (badge, weekly breakdown, insights) (#671)

WakaTime's existing stats endpoints already expose AI-coding data
(lines by AI vs. human, cost, tokens, sessions/prompts, per-model
breakdown), so no new API integration was needed.

- Add SHOW_AI_CODE_TIME (all-time AI Code Time badge, sourced from
  stats/all_time since all_time_since_today lacks AI fields) and
  SHOW_AI_CODING (weekly breakdown: lines, tokens, cost, sessions,
  per-model split) flags, both default True.
- Deduce a few insights purely from the raw ratios (AI reliance,
  prompt style, session style, review style) rather than just
  dumping numbers, mirroring the existing "I'm an Early"/"Most
  Productive on" style.
- Render a "No AI Coding Activity" fallback instead of hiding the
  section when an account has no AI data that week.
- Update mocks to match the real schema and propagate new
  translation keys across all 22 locales.

Author: anmol098

.env.example

@@ -14,6 +14,8 @@ INPUT_SHOW_LINES_OF_CODE=True
 INPUT_SHOW_LOC_CHART=True
 INPUT_SHOW_PROFILE_VIEWS=True
 INPUT_SHOW_TOTAL_CODE_TIME=True
+INPUT_SHOW_AI_CODE_TIME=True
+INPUT_SHOW_AI_CODING=True
 INPUT_SHOW_SHORT_INFO=True
 INPUT_SHOW_COMMIT=True
 INPUT_SHOW_DAYS_OF_WEEK=True

README.md

@@ -157,6 +157,50 @@ The `SHOW_TOTAL_CODE_TIME` flag can be set to `False` to hide *Code Time*.
 
 ![Code Time](http://img.shields.io/badge/Code%20Time-1%2C438%20hrs%2054%20mins-blue)
 
+> [!NOTE]
+> The `SHOW_AI_CODE_TIME` and `SHOW_AI_CODING` flags below require WakaTime's AI coding tracking to be recording activity on your account. If your account has no all-time AI data the **AI Code Time** badge is hidden entirely; if it has no AI data for the current week, the weekly block still shows with a "No AI Coding Activity Tracked This Week" message instead of numbers.
+
+The `SHOW_AI_CODE_TIME` flag can be set to `False` to hide the all-time **AI Code Time** badge.
+
+![AI Code Time](http://img.shields.io/badge/AI%20Code%20Time-77%20hrs%2022%20mins-blue)
+
+The `SHOW_AI_CODING` flag can be set to `False` to hide the weekly AI coding breakdown: AI coding time, AI vs. human written lines, token usage, estimated AI cost, sessions/prompts, a per-model breakdown, and a few deduced insights.
+
+**🤖 AI Coding This Week**
+
+```text
+⏱ AI Coding Time: 1 hr 53 mins (3.59%)
+
+✍️ 1,245 lines written by AI, 3,120 lines written by hand (28.52% AI-written)
+
+🔤 845,000 Input Tokens, 21,000 Output Tokens
+
+💵 $12.48 Estimated AI Cost This Week
+
+🧠 5 AI Sessions, 20 AI Prompts
+
+Sonnet                   1,200 lines         ██████████████████████░░░   89.96 %
+GPT-4                    134 lines           ███░░░░░░░░░░░░░░░░░░░░░░   10.04 %
+
+🔎 AI Coding Insights:
+🧑‍💻 Mostly Hands-On — 28.52% of written lines came from AI
+📄 Detailed Prompter — average 925 characters per prompt
+🔁 Iterative Prompter — average 4 prompts per session
+🔍 Hands-On Reviewer — 73.29% of changed lines were hand-edited
+```
+
+The insight lines are all deduced from the raw numbers above, not extra API data:
+- **AI Reliance** (`🤖 AI-Driven` / `⚖️ Balanced with AI` / `🧑‍💻 Mostly Hands-On`) — from the share of added lines written by AI.
+- **Prompt Style** (`📝 Concise` / `📄 Detailed` / `📚 Verbose`) — from the average prompt length.
+- **Session Style** (`🎯 One-Shot` / `🔁 Iterative`) — from the average number of prompts per AI session, i.e. whether you tend to get it right in one prompt or rely on follow-ups.
+- **Review Style** (`🔍 Hands-On Reviewer` / `🚀 High AI Trust`) — from the share of all changed lines that were still hand-edited, as a proxy for how much AI output gets manually reviewed/reworked.
+
+If there was no AI coding activity that week, the block still renders with a fallback message instead of disappearing:
+
+```text
+No AI Coding Activity Tracked This Week
+```
+
 The `SHOW_PROFILE_VIEWS` flag can be set to `False` to hide **Profile Views**
 
 ![Profile Views](http://img.shields.io/badge/Profile%20Views-2189-blue)

action.yml

@@ -102,6 +102,16 @@ inputs:
     description: "Show Total Time you have coded"
     default: "True"
 
+  SHOW_AI_CODE_TIME:
+    required: false
+    description: "Show the all-time AI Code Time badge"
+    default: "True"
+
+  SHOW_AI_CODING:
+    required: false
+    description: "Show the weekly AI coding breakdown (AI vs human lines, cost, model breakdown, sessions)"
+    default: "True"
+
   COMMIT_BY_ME:
     required: false
     description: "Git commit with your own name and email"

sources/main.py

@@ -4,7 +4,7 @@
 
 from asyncio import run
 from datetime import datetime
-from typing import Dict
+from typing import Dict, List, Optional
 from urllib.parse import quote
 
 from humanize import intword, naturalsize, intcomma
@@ -23,6 +23,106 @@
 )
 
 
+def find_category(categories: List[Dict], name: str) -> Optional[Dict]:
+    """
+    Find a WakaTime category entry (e.g. "AI Coding") by name in a stats response's `categories` list.
+
+    :param categories: List of category dictionaries from a WakaTime stats response.
+    :param name: Category name to look for.
+    :returns: The matching category dictionary, or None if not found.
+    """
+    return next((category for category in categories if category["name"] == name), None)
+
+
+def make_ai_coding_insights(ai_written_percent: float, prompt_length_avg: float, prompts_per_session: float, manual_touch_percent: float) -> str:
+    """
+    Deduce a few human-readable insight lines from the raw AI coding numbers:
+    how AI-reliant the week was, prompting style (length), session style (one-shot vs. follow-ups),
+    and how much of the changed code was still touched by hand (a proxy for manual review).
+    All tiers are computed purely from ratios already present in the WakaTime response, no extra API calls.
+
+    :param ai_written_percent: Share of added lines written by AI (0-100).
+    :param prompt_length_avg: Average prompt length in characters.
+    :param prompts_per_session: Average number of prompts per AI session.
+    :param manual_touch_percent: Share of all changed lines (additions + deletions) that were human-made (0-100).
+    :returns: String representation of the insight lines.
+    """
+    # Thresholds are heuristic tiers over continuous ratios, not WakaTime-defined categories.
+    reliance_label = (
+        FM.t("AI Reliance: AI-Driven")
+        if ai_written_percent >= 66
+        else FM.t("AI Reliance: Balanced") if ai_written_percent >= 33 else FM.t("AI Reliance: Hands-On")
+    )
+    prompt_style_label = (
+        FM.t("Prompt Style: Verbose")
+        if prompt_length_avg > 1500
+        else FM.t("Prompt Style: Detailed") if prompt_length_avg >= 500 else FM.t("Prompt Style: Concise")
+    )
+    session_style_label = FM.t("Session Style: Iterative") if prompts_per_session > 1.5 else FM.t("Session Style: One-Shot")
+    review_label = FM.t("Review Style: Hands-On Reviewer") if manual_touch_percent >= 50 else FM.t("Review Style: High AI Trust")
+
+    insights = f"🔎 {FM.t('AI Coding Insights')}:\n"
+    insights += f"{FM.t('AI Reliance Detail') % (reliance_label, round(ai_written_percent, 2))}\n"
+    insights += f"{FM.t('Prompt Style Detail') % (prompt_style_label, intcomma(round(prompt_length_avg)))}\n"
+    insights += f"{FM.t('Session Style Detail') % (session_style_label, round(prompts_per_session, 1))}\n"
+    insights += f"{FM.t('Review Style Detail') % (review_label, round(manual_touch_percent, 2))}\n"
+    return insights
+
+
+def make_ai_coding_stats(data: Dict) -> str:
+    """
+    Build the weekly AI coding stats block: AI coding time, AI vs human written lines,
+    token usage, estimated AI cost, sessions/prompts, per-model breakdown and deduced insights.
+    Renders a "no activity" fallback (instead of hiding the section) if the account has no AI coding data this week.
+
+    :param data: WakaTime weekly stats response (`waka_latest`).
+    :returns: String representation of the AI coding stats.
+    """
+    ai_category = find_category(data["data"].get("categories", []), "AI Coding")
+    ai_sessions = data["data"].get("ai_sessions", 0)
+
+    stats = f"🤖 **{FM.t('AI Coding This Week')}** \n\n```text\n"
+
+    if ai_category is None or not ai_sessions:
+        stats += f"{FM.t('No AI Coding Activity Tracked This Week')}\n\n"
+        return f"{stats[:-1]}```\n\n"
+
+    ai_additions = data["data"].get("ai_additions", 0)
+    ai_deletions = data["data"].get("ai_deletions", 0)
+    human_additions = data["data"].get("human_additions", 0)
+    human_deletions = data["data"].get("human_deletions", 0)
+    ai_input_tokens = data["data"].get("ai_input_tokens", 0)
+    ai_output_tokens = data["data"].get("ai_output_tokens", 0)
+    ai_cost = data["data"].get("ai_model_total_cost", 0)
+    ai_prompts = data["data"].get("ai_prompt_events_total", 0)
+    prompt_length_avg = data["data"].get("ai_prompt_length_avg", 0)
+    prompts_per_session = data["data"].get("ai_prompt_events_avg_per_session", 0)
+
+    total_additions = ai_additions + human_additions
+    ai_written_percent = (ai_additions / total_additions * 100) if total_additions else 0
+
+    total_changes = ai_additions + ai_deletions + human_additions + human_deletions
+    manual_touch_percent = ((human_additions + human_deletions) / total_changes * 100) if total_changes else 0
+
+    stats += f"⏱ {FM.t('AI Coding Time')}: {ai_category['text']} ({ai_category['percent']}%)\n\n"
+    stats += f"✍️ {FM.t('AI vs Human Lines') % (intcomma(ai_additions), intcomma(human_additions), round(ai_written_percent, 2))}\n\n"
+    stats += f"🔤 {FM.t('AI Token Usage') % (intcomma(ai_input_tokens), intcomma(ai_output_tokens))}\n\n"
+    stats += f"💵 {FM.t('Estimated AI Cost') % f'{ai_cost:.2f}'}\n\n"
+    stats += f"🧠 {FM.t('AI Sessions and Prompts') % (ai_sessions, ai_prompts)}\n\n"
+
+    ai_model_breakdown = data["data"].get("ai_model_breakdown", [])
+    if ai_model_breakdown:
+        total_lines = sum(model["lines"] for model in ai_model_breakdown) or 1
+        names = [model["name"] for model in ai_model_breakdown]
+        texts = [f"{intcomma(model['lines'])} lines" for model in ai_model_breakdown]
+        percents = [round(model["lines"] / total_lines * 100, 2) for model in ai_model_breakdown]
+        stats += f"{make_list(names=names, texts=texts, percents=percents)}\n\n"
+
+    stats += f"{make_ai_coding_insights(ai_written_percent, prompt_length_avg, prompts_per_session, manual_touch_percent)}\n"
+
+    return f"{stats[:-1]}```\n\n"
+
+
 async def get_waka_time_stats(repositories: Dict, commit_dates: Dict) -> str:
     """
     Collects user info from wakatime.
@@ -74,6 +174,10 @@ async def get_waka_time_stats(repositories: Dict, commit_dates: Dict) -> str:
 
         stats = f"{stats[:-1]}```\n\n"
 
+    if EM.SHOW_AI_CODING:
+        DBM.i("Adding AI coding stats...")
+        stats += make_ai_coding_stats(data)
+
     DBM.g("WakaTime stats added!")
     return stats
 
@@ -193,13 +297,28 @@ async def get_stats() -> str:
         yearly_data, commit_data = dict(), dict()
         DBM.w("User yearly data not needed, skipped.")
 
-    if EM.SHOW_TOTAL_CODE_TIME:
+    if EM.SHOW_TOTAL_CODE_TIME or EM.SHOW_AI_CODE_TIME:
         DBM.i("Adding total code time info...")
         data = await DM.get_remote_json("waka_all")
         if data is None:
             DBM.p("WakaTime data unavailable!")
         else:
-            stats += f"![Code Time](http://img.shields.io/badge/{quote('Code Time')}-{quote(str(data['data']['text']))}-blue?style={quote(EM.BADGE_STYLE)})\n\n"
+            if EM.SHOW_TOTAL_CODE_TIME:
+                stats += (
+                    f"![Code Time](http://img.shields.io/badge/{quote('Code Time')}-"
+                    f"{quote(str(data['data']['human_readable_total']))}-blue?style={quote(EM.BADGE_STYLE)})\n\n"
+                )
+
+            if EM.SHOW_AI_CODE_TIME:
+                DBM.i("Adding AI code time info...")
+                ai_category = find_category(data["data"].get("categories", []), "AI Coding")
+                if ai_category is None:
+                    DBM.w("No all-time AI coding data available, skipping AI Code Time badge.")
+                else:
+                    stats += (
+                        f"![AI Code Time](http://img.shields.io/badge/{quote('AI Code Time')}-"
+                        f"{quote(str(ai_category['text']))}-blue?style={quote(EM.BADGE_STYLE)})\n\n"
+                    )
 
     if EM.SHOW_PROFILE_VIEWS:
         if EM.DEBUG_RUN or GHM.REMOTE is None:

sources/manager_download.py

@@ -130,7 +130,7 @@ async def init_download_manager(user_login: str):
     await DownloadManager.load_remote_resources(
         linguist="https://cdn.jsdelivr.net/gh/github/linguist@master/lib/linguist/languages.yml",
         waka_latest=f"{EM.WAKATIME_API_URL}users/current/stats/last_7_days?api_key={EM.WAKATIME_API_KEY}",
-        waka_all=f"{EM.WAKATIME_API_URL}users/current/all_time_since_today?api_key={EM.WAKATIME_API_KEY}",
+        waka_all=f"{EM.WAKATIME_API_URL}users/current/stats/all_time?api_key={EM.WAKATIME_API_KEY}",
         github_stats=f"https://github-contributions.vercel.app/api/v1/{user_login}",
     )
 

sources/manager_environment.py

@@ -49,6 +49,8 @@ class EnvironmentManager:
     SHOW_SHORT_INFO = getenv("INPUT_SHOW_SHORT_INFO", "True").lower() in _TRUTHY
     SHOW_UPDATED_DATE = getenv("INPUT_SHOW_UPDATED_DATE", "True").lower() in _TRUTHY
     SHOW_TOTAL_CODE_TIME = getenv("INPUT_SHOW_TOTAL_CODE_TIME", "True").lower() in _TRUTHY
+    SHOW_AI_CODE_TIME = getenv("INPUT_SHOW_AI_CODE_TIME", "True").lower() in _TRUTHY
+    SHOW_AI_CODING = getenv("INPUT_SHOW_AI_CODING", "True").lower() in _TRUTHY
 
     COMMIT_BY_ME = getenv("INPUT_COMMIT_BY_ME", "False").lower() in _TRUTHY
     COMMIT_MESSAGE = getenv("INPUT_COMMIT_MESSAGE", "Updated with Dev Metrics")

sources/mock_data/mock_wakatime_all_time.json

@@ -1,5 +1,47 @@
 {
-  "data": {
-    "text": "test"
-  }
+    "data": {
+        "id": "00000000-0000-0000-0000-000000000000",
+        "user_id": "11111111-1111-1111-1111-111111111111",
+        "range": "all_time",
+        "start": "2020-07-14T00:00:00Z",
+        "end": "2026-07-27T23:59:59Z",
+        "timezone": "UTC",
+        "status": "ok",
+        "total_seconds": 19356005.587,
+        "human_readable_total": "5,376 hrs 40 mins",
+        "is_up_to_date": true,
+        "percent_calculated": 100,
+        "categories": [
+            {
+                "name": "Coding",
+                "total_seconds": 19021442.488,
+                "percent": 98.27,
+                "digital": "5283:44",
+                "decimal": "5283.73",
+                "text": "5,283 hrs 44 mins",
+                "hours": 5283,
+                "minutes": 44
+            },
+            {
+                "name": "AI Coding",
+                "total_seconds": 278541.827,
+                "percent": 1.44,
+                "digital": "77:22",
+                "decimal": "77.37",
+                "text": "77 hrs 22 mins",
+                "hours": 77,
+                "minutes": 22
+            },
+            {
+                "name": "Writing Docs",
+                "total_seconds": 49815.63,
+                "percent": 0.26,
+                "digital": "13:50",
+                "decimal": "13.83",
+                "text": "13 hrs 50 mins",
+                "hours": 13,
+                "minutes": 50
+            }
+        ]
+    }
 }

sources/mock_data/mock_wakatime_stats.json

@@ -8,6 +8,42 @@
         "timeout": 15,
         "writes_only": false,
         "timezone": "UTC",
+        "ai_additions": 1245,
+        "ai_deletions": 89,
+        "human_additions": 3120,
+        "human_deletions": 540,
+        "ai_line_changes_total": 1334,
+        "ai_model_total_cost": 12.4821,
+        "ai_input_tokens": 845000,
+        "ai_output_tokens": 21000,
+        "ai_prompt_length_sum": 18500,
+        "ai_prompt_length_avg": 925,
+        "ai_prompt_events_total": 20,
+        "ai_prompt_events_avg_per_session": 4,
+        "ai_prompt_events_median_per_session": 3,
+        "ai_prompt_length_avg_per_session": 1233,
+        "ai_prompt_length_median_per_session": 1100,
+        "ai_sessions": 5,
+        "ai_model_line_changes": {
+            "Sonnet": 1200,
+            "GPT-4": 134
+        },
+        "ai_model_costs": {
+            "Sonnet": 9.5,
+            "GPT-4": 2.98
+        },
+        "ai_model_breakdown": [
+            {
+                "name": "Sonnet",
+                "lines": 1200,
+                "cost": 9.5
+            },
+            {
+                "name": "GPT-4",
+                "lines": 134,
+                "cost": 2.98
+            }
+        ],
         "holidays": 0,
         "status": "ok",
         "created_at": "2025-12-31T23:02:40Z",
@@ -627,4 +663,4 @@
         "is_category_usage_visible": false,
         "is_os_usage_visible": false
     }
-}
+}