import os
from google import genai

def run_gemini_test():
    """
    環境変数からAPIキーを取得し、Gemini APIで簡単な動作確認を行う関数。
    """
    # 1. 環境変数 'GOOGLE_API_KEY' からAPIキーを読み込む
    # クライアント初期化時にAPIキーが見つからない場合、自動で環境変数 'GEMINI_API_KEY' を探します。
    # したがって、直接 'os.environ.get()' を使う必要はありませんが、確認のために記載します。
    if not os.getenv("GOOGLE_API_KEY"):
        print("エラー: 環境変数 'GOOGLE_API_KEY' が設定されていません。")
        print("APIキーを設定してから再度実行してください。")
        return

    try:
        # 2. Gemini クライアントを初期化 (APIキーは環境変数から自動取得)
        client = genai.Client()
        
        # 3. 動作確認のための簡単なプロンプト
        prompt = "最も簡単な四字熟語を一つだけ教えてください。"
        
        print(f"--- Gemini API 動作確認 ---")
        print(f"プロンプト: {prompt}")

        # 4. gemini-2.5-flash モデルにリクエストを送信
        response = client.models.generate_content(
            model='gemini-2.5-flash',
            contents=prompt,
        )

        # 5. 結果を出力
        print("\n✅ APIからの応答に成功しました。")
        print("--- 結果 ---")
        print(response.text)
        print("---------------------------")

    except Exception as e:
        print(f"\n❌ エラーが発生しました: {e}")
        print("APIキーが正しくない、またはモデル名が間違っている可能性があります。")

if __name__ == "__main__":
    run_gemini_test()