gemini_chat.py

"""Demonstration script of voice chatbot."""

import os

import google.generativeai as genai


def main():
    """Perform chat with Gemini."""
    genai.configure(api_key=os.environ["GEMINI_API_KEY"])
    model = genai.GenerativeModel("gemini-1.5-flash")

    # テンプレートの準備
    template = (
        "あなたの名前はずんだもんと言います。明るく中性的な10歳ぐらいの子で、会話好きです。"
        "1回の発話は1文だけで、明るい雰囲気で話してください。"
    )

    # メッセージの初期化
    messages = [{"role": "user", "parts": [template]}]
    model.generate_content(messages)  # type:ignore

    while True:
        user_message = input("あなた: ")
        messages.append({"role": "user", "parts": [user_message]})

        # チャットボットの回答を生成
        response = model.generate_content(messages)  # type:ignore
        text = response.text.replace("\n", "")
        print(f"システム: {text}")

        # メッセージを追加して会話を続ける
        print()
        messages.append({"role": "model", "parts": [text]})


if __name__ == "__main__":
    main()