script.py

"""Script generation module — turns commit summaries into a 6-panel comic script via Gemini Flash."""

import json
import os
import re


def generate_comic_script(
    commits_summary: str,
    system_prompt: str,
    lang_addon: str,
    api_key: str = None,
) -> dict:
    """Use Gemini to turn commit summaries into a comic script.

    Args:
        commits_summary: Summarised git commits / standup notes.
        system_prompt: The base system prompt (e.g. contents of config/script_system.md).
        lang_addon: Language-specific addendum (e.g. contents of config/language_en.md).
        api_key: Google API key. Falls back to GOOGLE_API_KEY env var.

    Returns:
        Parsed JSON dict representing the 6-panel comic script.
    """
    from google import genai
    from google.genai import types

    client = genai.Client(api_key=api_key or os.environ.get("GOOGLE_API_KEY"))

    system = system_prompt + "\n\n" + lang_addon

    response = client.models.generate_content(
        model="gemini-2.5-flash",
        contents=f"Here are this week's git commits from our dev team:\n{commits_summary}\n\nWrite the 6-panel comic strip script!",
        config=types.GenerateContentConfig(
            system_instruction=system,
            temperature=1.0,
        ),
    )

    raw = response.text.strip()

    # Strip markdown fences if present
    raw = re.sub(r"^```(?:json)?\s*", "", raw)
    raw = re.sub(r"\s*```$", "", raw)

    return json.loads(raw)