Survey Insight MCP Server

"""WordCloud画像生成モジュール."""

import logging
import random
from pathlib import Path
from typing import Optional

import matplotlib
import matplotlib.pyplot as plt
from wordcloud import WordCloud

# matplotlibのバックエンドをAggに設定（GUI不要）
matplotlib.use("Agg")

logger = logging.getLogger(__name__)


class WordCloudGenerator:
    """WordCloud画像生成を行うクラス.

    日本語対応のWordCloudを生成し、美しいカラースキームを適用します。
    """

    def __init__(self):
        """初期化."""
        self.default_font = self._get_japanese_font()
        self.color_schemes = {"accenture": self._accenture_color_func, "default": None}

    def generate(
        self,
        text_or_frequencies: str | dict[str, int],
        output_path: Optional[str] = None,
        color_scheme: str = "accenture",
        width: int = 1200,
        height: int = 600,
    ) -> str:
        """WordCloudを生成.

        Args:
            text_or_frequencies: WordCloud生成元のテキストまたは頻度辞書
            output_path: 画像出力パス（省略時は一時ファイル）
            color_scheme: カラースキーム（'accenture' or 'default'）
            width: 画像幅
            height: 画像高さ

        Returns:
            str: 生成された画像のパス

        Raises:
            ValueError: データが空の場合
        """
        # 入力チェック
        if isinstance(text_or_frequencies, dict):
            if not text_or_frequencies:
                raise ValueError("頻度辞書が空です")
            logger.info(f"WordCloud生成開始: {len(text_or_frequencies)}語の頻度辞書")
        elif isinstance(text_or_frequencies, str):
            if not text_or_frequencies or not text_or_frequencies.strip():
                raise ValueError("テキストが空です")
            logger.info(f"WordCloud生成開始: {len(text_or_frequencies)}文字")
        else:
            raise ValueError("テキストまたは頻度辞書を指定してください")

        # カラー関数取得
        color_func = self.color_schemes.get(color_scheme, None)

        # WordCloud生成
        wordcloud = WordCloud(
            width=width,
            height=height,
            background_color="#F8F8F8" if color_scheme == "accenture" else "white",
            font_path=self.default_font,
            color_func=color_func,
            prefer_horizontal=0.7,
            relative_scaling=0.5,
            min_font_size=10,
        )

        # 頻度辞書またはテキストから生成
        if isinstance(text_or_frequencies, dict):
            wordcloud.generate_from_frequencies(text_or_frequencies)
        else:
            wordcloud.generate(text_or_frequencies)

        # 画像保存
        if output_path is None:
            output_path = "wordcloud.png"

        output_file = Path(output_path)
        output_file.parent.mkdir(parents=True, exist_ok=True)

        plt.figure(figsize=(width / 100, height / 100), facecolor="#F8F8F8" if color_scheme == "accenture" else "white")
        plt.imshow(wordcloud, interpolation="bilinear")
        plt.axis("off")
        plt.tight_layout(pad=0)
        plt.savefig(
            output_file, format="png", dpi=100, bbox_inches="tight", facecolor="#F8F8F8" if color_scheme == "accenture" else "white"
        )
        plt.close()

        logger.info(f"WordCloud生成完了: {output_file}")
        return str(output_file)

    def _get_japanese_font(self) -> Optional[str]:
        """日本語フォントパスを取得.

        Returns:
            Optional[str]: フォントパス（見つからない場合はNone）
        """
        # Windows環境での日本語フォント候補
        font_candidates = [
            "C:\\Windows\\Fonts\\msgothic.ttc",  # MSゴシック
            "C:\\Windows\\Fonts\\meiryo.ttc",  # メイリオ
            "C:\\Windows\\Fonts\\yugothm.ttc",  # 游ゴシック
            # Linux環境
            "/usr/share/fonts/truetype/fonts-japanese-gothic.ttf",
            "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
            # macOS環境
            "/System/Library/Fonts/ヒラギノ角ゴシック W3.ttc",
            "/Library/Fonts/Hiragino Sans GB.ttc",
        ]

        for font_path in font_candidates:
            if Path(font_path).exists():
                logger.info(f"日本語フォント検出: {font_path}")
                return font_path

        logger.warning("日本語フォントが見つかりません。デフォルトフォントを使用します。")
        return None

    def _accenture_color_func(self, word=None, font_size=None, position=None, orientation=None, font_path=None, random_state=None):
        """パープルグラデーションカラー関数.

        パープルグラデーション（#A100FF → #6B00B6）のカラースキームを生成します。

        Returns:
            str: RGB文字列
        """
        # パープルグラデーションの色リスト
        purple_colors = [
            (161, 0, 255),  # #A100FF
            (139, 0, 230),
            (107, 0, 182),  # #6B00B6
            (201, 102, 255),  # #C966FF (lighter)
            (180, 50, 255),
        ]

        # random_stateを使用して決定的に色を選択
        # 単語のハッシュ値から色を決定（再現性を持たせる）
        if word:
            idx = hash(word) % len(purple_colors)
        elif random_state is not None:
            # numpy.random.RandomStateオブジェクトの場合
            if hasattr(random_state, 'randint'):
                idx = random_state.randint(0, len(purple_colors))
                # 念のため範囲チェック
                idx = min(idx, len(purple_colors) - 1)
            else:
                # 整数などの場合
                idx = hash(random_state) % len(purple_colors)
        else:
            # random_stateがない場合はランダムに選択
            idx = random.randint(0, len(purple_colors) - 1)

        color = purple_colors[idx]
        return f"rgb({color[0]}, {color[1]}, {color[2]})"