sudo apt update && sudo apt install -y python3.11 python3.11-venv python3-pip

mkdir ~/discord-ai-bot && cd ~/discord-ai-bot
python3.11 -m venv venv
source venv/bin/activate

pip install discord.py requests python-dotenv aiohttp

# Ollamaインストール
curl -fsSL https://ollama.ai/install.sh | sh

# テキスト生成用モデルを取得（約4.7GB）
ollama pull llama3

# バックグラウンドで起動
sudo systemctl enable ollama
sudo systemctl start ollama

# bot.py
import discord
from discord.ext import commands
from discord import app_commands
import aiohttp
import os
from dotenv import load_dotenv

load_dotenv()

DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
OLLAMA_URL = "http://localhost:11434/api/generate"

intents = discord.Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix="!", intents=intents)
tree = bot.tree

@bot.event
async def on_ready():
    await tree.sync()
    print(f"ボット起動: {bot.user}")

async def ask_ollama(prompt: str, model: str = "llama3") -> str:
    """Ollamaに問い合わせて回答を取得"""
    payload = {
        "model": model,
        "prompt": prompt,
        "stream": False,
    }
    async with aiohttp.ClientSession() as session:
        async with session.post(OLLAMA_URL, json=payload) as resp:
            data = await resp.json()
            return data.get("response", "回答を生成できませんでした")

@tree.command(name="ask", description="AIに質問する")
async def ask_command(interaction: discord.Interaction, question: str):
    """スラッシュコマンド /ask で質問を受け付ける"""
    await interaction.response.defer()  # 処理中を示す

    response = await ask_ollama(question)

    # 2000文字制限を考慮して分割
    if len(response) > 1900:
        response = response[:1900] + "...(省略)"

    await interaction.followup.send(
        f"**質問:** {question}\n\n**回答:**\n{response}"
    )

@tree.command(name="chat", description="AIとフリートークする")
async def chat_command(interaction: discord.Interaction, message: str):
    """日本語でフリートーク"""
    await interaction.response.defer()

    system_prompt = f"あなたは親切なアシスタントです。日本語で簡潔に回答してください。\n\nユーザー: {message}"
    response = await ask_ollama(system_prompt)

    embed = discord.Embed(
        title="AI の回答",
        description=response[:4000],
        color=discord.Color.blue()
    )
    embed.set_footer(text="Powered by Ollama + Llama 3")
    await interaction.followup.send(embed=embed)

@bot.event
async def on_message(message: discord.Message):
    """ボットがメンションされたときに反応"""
    if message.author == bot.user:
        return
    if bot.user.mentioned_in(message):
        question = message.content.replace(f"<@{bot.user.id}>", "").strip()
        if question:
            async with message.channel.typing():
                response = await ask_ollama(question)
                await message.reply(response[:2000])

    await bot.process_commands(message)

bot.run(DISCORD_TOKEN)

nano .env

python bot.py

sudo nano /etc/systemd/system/discord-bot.service

[Unit]
Description=Discord AI Bot
After=network.target ollama.service

[Service]
Type=simple
User=ubuntu
WorkingDirectory=/home/ubuntu/discord-ai-bot
EnvironmentFile=/home/ubuntu/discord-ai-bot/.env
ExecStart=/home/ubuntu/discord-ai-bot/venv/bin/python bot.py
Restart=always
RestartSec=10

[Install]
WantedBy=multi-user.target

sudo systemctl daemon-reload
sudo systemctl enable discord-bot
sudo systemctl start discord-bot
sudo systemctl status discord-bot

# ComfyUIインストール（GPUプランのVPS推奨）
git clone https://github.com/comfyanonymous/ComfyUI.git
cd ComfyUI
pip install -r requirements.txt
python main.py --listen 127.0.0.1 --port 8188

# bot.py に追加
import aiofiles
import uuid

@tree.command(name="image", description="AIに画像を生成させる")
async def image_command(interaction: discord.Interaction, prompt: str):
    await interaction.response.defer()

    # ComfyUI API呼び出し（簡略版）
    async with aiohttp.ClientSession() as session:
        payload = {
            "prompt": {
                "3": {
                    "inputs": {"text": prompt, "clip": ["4", 1]},
                    "class_type": "CLIPTextEncode"
                }
                # 実際はComfyUIのワークフローJSON全体を指定
            }
        }
        async with session.post(
            "http://127.0.0.1:8188/prompt", json=payload
        ) as resp:
            result = await resp.json()
            await interaction.followup.send(f"画像生成中... プロンプト: {prompt}")