Mの備忘録 - Carnet de M -

import os
import openai
import datetime
from slack_bolt import App
from slack_bolt.adapter.socket_mode import SocketModeHandler

# 環境変数の設定
SLACK_BOT_TOKEN = os.environ.get("SLACK_BOT_TOKEN")
SLACK_APP_TOKEN = os.environ["SLACK_APP_TOKEN"]
OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]

# OpenAI APIキーの設定
openai.api_key = OPENAI_API_KEY

# Slackアプリの初期化
app = App(token=SLACK_BOT_TOKEN)

# 対話履歴とモデルの設定
channel_conversation_history = {}
last_interaction_time = {}
model = "gpt-4-1106-preview"  # デフォルトモデルを設定

# GPTによる会話解析と画像生成の推奨
def analyze_and_recommend_image(message):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4-1106-preview",  # 分析用のモデルを指定
            messages=[
                {"role": "system", "content": "Analyze the message provided. If the message explicitly requests the creation of a photograph, drawing, illustration, or any form of visual artwork, respond with 'Yes'.This includes requests for specific scenes, objects, concepts, or designs to be visualized. However, if the message does not explicitly ask for an image or is seeking information, explanations, or any non-visual content, respond with 'No'. This applies to general questions, requests for written content, or any query where a visual representation is not directly requested."},
                {"role": "user", "content": message}
            ]
        )
        return "Yes" in response['choices'][0]['message']['content']
    except Exception as e:
        print(f"Error in analyze_and_recommend_image: {e}")
        return False

# DALL-E APIを呼び出す関数
def generate_dalle_image(prompt):
    try:
        response = openai.Image.create(
            model="dall-e-3",
            prompt=prompt,
            n=1,
            size="1024x1024"
        )
        image_url = response['data'][0]['url']
        return image_url
    except openai.error.InvalidRequestError as e:
        return "error"

# Slackに画像を投稿する関数
def post_image_to_slack(channel_id, image_url, ts):
    try:
        if image_url == "error":
            app.client.chat_update(
                channel=channel_id,
                ts=ts,
                text="画像を生成できませんでした。"
            )
        else:
            app.client.chat_update(
                channel=channel_id,
                ts=ts,
                text="",
                blocks=[
                    {
                        "type": "image",
                        "title": {
                            "type": "plain_text",
                            "text": "Generated Image"
                        },
                        "image_url": image_url,
                        "alt_text": "Generated Image"
                    }
                ]
            )
    except Exception as e:
        logger.error(f"Error posting image to Slack: {e}")

# Slackイベントハンドラ
@app.event("message")
def handle_message(body, logger):
    global channel_conversation_history
    global last_interaction_time

    user_id = body["event"]["user"]
    channel_id = body["event"]["channel"]
    message = body["event"]["text"]

    initial_message_response = app.client.chat_postMessage(
        channel=channel_id,
        text="..."
    )
    ts = initial_message_response['ts']

    if analyze_and_recommend_image(message):
        image_url = generate_dalle_image(message)
        post_image_to_slack(channel_id, image_url, ts)
    else:
        if channel_id not in channel_conversation_history:
            channel_conversation_history[channel_id] = []

        if channel_id not in last_interaction_time or (datetime.datetime.now() - last_interaction_time[channel_id]).seconds > 3600:
            channel_conversation_history[channel_id] = []

        last_interaction_time[channel_id] = datetime.datetime.now()

        gpt_response = get_gpt_response(channel_conversation_history[channel_id], message, model)

        try:
            app.client.chat_update(
                channel=channel_id,
                ts=ts,

if __name__ == "__main__":
    handler = SocketModeHandler(app, SLACK_APP_TOKEN)
    handler.start()

@app.event("message")
def handle_message(body, logger):
    user_id = body["event"]["user"]
    channel_id = body["event"]["channel"]
    message = body["event"]["text"]

    initial_message_response = app.client.chat_postMessage(
        channel=channel_id,
        text="..."
    )
    ts = initial_message_response['ts']

    if analyze_and_recommend_image(message):
        # 画像生成の処理
    else:
        # テキスト応答の生成

def analyze_and_recommend_image(message):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4-1106-preview",
            messages=[
                {"role": "system", "content": "Analyze the message provided. If the message explicitly requests the creation of a photograph, drawing, illustration, or any form of visual artwork, respond with 'Yes'.This includes requests for specific scenes, objects, concepts, or designs to be visualized. However, if the message does not explicitly ask for an image or is seeking information, explanations, or any non-visual content, respond with 'No'. This applies to general questions, requests for written content, or any query where a visual representation is not directly requested."},
                {"role": "user", "content": message}
            ]
        )
        return "Yes" in response['choices'][0]['message']['content']
    except Exception as e:
        print(f"Error in analyze_and_recommend_image: {e}")
        return False

def get_gpt_response(conversation_history, message, model_name):
    conversation_history.append({"role": "user", "content": message})
    # トークン数のチェックと会話履歴の調整
    response = openai.ChatCompletion.create(
        model=model_name,
        messages=[{"role": "system", "content": "You are a helpful assistant."}] + conversation_history
    )
    assistant_message = response.choices[0]['message']['content'].strip()
    conversation_history.append({"role": "assistant", "content": assistant_message})
    return assistant_message

def generate_dalle_image(prompt):
    try:
        response = openai.Image.create(
            model="dall-e-3",
            prompt=prompt,
            n=1,
            size="1024x1024"
        )
        image_url = response['data'][0]['url']
        return image_url
    except openai.error.InvalidRequestError as e:
        return "error"