1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# Geminiくんの召喚とフィルター関係
import google.generativeai as genai
from google.generativeai.types import HarmCategory, HarmBlockThreshold

# discord.pyを呼び出す
import discord

token = "discordのトークンを貼る"

# ~~discordの設定~~

# intents(おまじない)を唱える
intents = discord.Intents.default()
intents.message_content = True

# discordのクライアントを作成
client = discord.client(intents=intents)


# ~~Geminiの設定~~

# APIキーを貼り付けましょう
api_key = "ここにさっき出てきたAPIキーを貼り付ける"

# うし，じゃあ(APIキーを)ぶち込んでやるぜ！
genai.configure(api_key=api_key)

# モデルをジュンピ
model = genai.GenerativeModel("gemini-1.5-pro-latest")


# 起動した時の設定
async def on_ready():
    print(f"Logged in as {client.user.name}")

# メッセージが送られたら反応する関数
async def on_message(message):
    # bot自身には反応しない
    if message.author == client.user:
        return

    if message.content.startswith("!t"):
        # "!t"の文字を消す
        prompt = message.content[2:].strip()

        # チャット形式に変える！
        chat = model.start_chat(history=[])

        #新たにsafety_settingsを追加
        response = chat.send_message(
            prompt, 
            safety_settings={
                HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
                HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
                HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
                HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT:HarmBlockThreshold.BLOCK_NONE
                })

        # Geminiくんの返信をdiscordに送信
        message.reply(response.text)

if __name__ == "__main__":
    # botを起動!
    client.run(token)