#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
get_from_ai_simple.py
get_from_ai.ini から PROMPT を読み込み、AI応答をファイルに書き出す簡易スクリプト。
"""

import os
import sys
import argparse
import re
from pathlib import Path
from openai import OpenAI
import google.generativeai as genai


def read_ini(path):
    """寛容な key=value 形式（multiライン対応）のINIを読む"""
    if not os.path.isfile(path):
        raise FileNotFoundError(f"INIファイルが見つかりません: {path}")

    result = {}
    buf = []
    current_key = None
    multi = None
    first_valid_line = True

    with open(path, encoding="utf-8") as f:
        for line in f:
            line = line.rstrip()
            if not line or line.startswith(("#", ";")):
                continue

            # multi-line mode
            if multi:
                if line.strip() == multi:
                    result[current_key] = "\n".join(buf)
                    buf = []
                    multi = None
                else:
                    buf.append(line)
                continue

            # key=value line
            if "=" in line:
                k, v = map(str.strip, line.split("=", 1))
                if v in ('"""', "'''"):
                    multi = v
                    current_key = k
                    buf = []
                else:
                    result[k] = v
                first_valid_line = False
            else:
                # 最初の有効行で = がない → multiライン開始と仮定
                if first_valid_line:
                    if line in ('"""', "'''"):
                        multi = line
                        current_key = "PROMPT"
                        buf = []
                    else:
                        # 開始記号がない場合でも、multiラインとして扱う
                        multi = '"""'  # 仮の閉じ記号（どちらでもよい）
                        current_key = "PROMPT"
                        buf = [line]
                    first_valid_line = False
                else:
                    # それ以降の =なし行は無視またはログ出力
                    pass

    # ファイル末尾まで multi-line が閉じられなかった場合
    if multi and buf:
        result[current_key] = "\n".join(buf)

    return result

def call_ai(prompt, api="gemini", model=None):
    """AI呼び出し：GeminiまたはOpenAI"""
    if api == "gemini":
        key = os.getenv("GOOGLE_API_KEY")
        if not key:
            raise ValueError("環境変数 GOOGLE_API_KEY が未設定です。")
        genai.configure(api_key=key)
        model = model or os.getenv("GOOGLE_MODEL", "gemini-2.5-flash")
        print(f"🚀 Gemini [{model}] 実行中...")
        m = genai.GenerativeModel(model)
        resp = m.generate_content(prompt)
        return resp.text
    elif api in ("openai", "openai5"):
        key = os.getenv("OPENAI_API_KEY")
        if not key:
            raise ValueError("環境変数 OPENAI_API_KEY が未設定です。")
        model = model or os.getenv("OPENAI_MODEL", "gpt-4o")
        print(f"🚀 OpenAI [{model}] 実行中...")
        client = OpenAI(api_key=key)
        if api == "openai5":
            r = client.responses.create(model=model, input=prompt)
            return r.output_text
        else:
            r = client.chat.completions.create(model=model,
                                               messages=[{"role": "user", "content": prompt}])
            return r.choices[0].message.content
    else:
        raise ValueError(f"未対応API: {api}")

def main():
    p = argparse.ArgumentParser(description="get_from_ai.iniからプロンプトを読み込み、AI応答をファイルに出力")
    p.add_argument("--inifile",     "-i", default="get_from_ai.ini", help="プロンプトを含むINIファイル (default: get_from_ai.ini)")
    p.add_argument("--output_path", "-o", required=True, help="AI応答を書き込む出力ファイルパス")
    p.add_argument("--api",         "-a", choices=["gemini", "openai", "openai5"], default="gemini", help="使用するAPI")
    p.add_argument("--model",       "-m", default=None, help="モデル名を明示指定（任意）")
    args = p.parse_args()

    print()

    data = read_ini(args.inifile)
    print(f"keys in [{args.inifile}]:")
    for key in data.keys():
        print(f"  {key}")

    prompt = data.get("PROMPT") or data.get("PROMPT_TEMPLATE_JA") or data.get("PROMPT_TEMPLATE_EN")
    if not prompt:
        print("❌ INIにPROMPTが定義されていません", file=sys.stderr)
        sys.exit(1)

    print("Prompt:", prompt)

    print("Requesting to {args.api}/{args.model}...")
    response = call_ai(prompt, api=args.api, model=args.model)

    Path(args.output_path).write_text(response.strip(), encoding="utf-8")
    print(f"✅ 出力ファイルを生成しました: {args.output_path}")

if __name__ == "__main__":
    main()