【Streamlit】DALL-E3で好みのスタイルの画像を生成する

import openai
import streamlit as st

# OpenAI APIキーを直接指定
OPENAI_API_KEY = "YOUR_API_KEY"
openai.api_key = OPENAI_API_KEY

def generate_image(prompt):
    """DALL-Eで画像を生成する関数"""
    try:
        response = openai.Image.create(
            prompt=prompt,
            model="dall-e-3",
            n=1,  # 生成する画像の数
            size="1024x1024"  # 画像サイズ
        )
        return response["data"][0]["url"]
    except Exception as e:
        st.error(f"エラーが発生しました: {e}")
        return None

# Streamlitアプリのレイアウト
st.title("DALL-E3 画像生成アプリ")

# ユーザーの入力
prompt = st.text_input("画像の説明を入力してください:")

# 画像のテイスト選択
taste = st.selectbox(
    "画像のテイストを選択してください:",
    ["絵画風", "写真風", "アニメ風", "手書きスケッチ風"]
)

# テイストごとのプロンプト修正
def modify_prompt(base_prompt, taste):
    styles = {
        "絵画風": "A beautiful oil painting of",
        "写真風": "A realistic photo of",
        "アニメ風": "An anime-style illustration of",
        "手書きスケッチ風": "A detailed pencil sketch of"
    }
    return f"{styles[taste]} {base_prompt}"

if st.button("画像を生成"):
    if prompt:
        modified_prompt = modify_prompt(prompt, taste)
        with st.spinner("画像を生成中..."):
            image_url = generate_image(modified_prompt)
            if image_url:
                st.image(image_url, caption="生成された画像", use_column_width=True)
    else:
        st.warning("画像の説明を入力してください。")

def generate_image(prompt):
    try:
        response = openai.Image.create(
            prompt=prompt,
            model="dall-e-3",
            n=1,
            size="1024x1024"
        )
        return response["data"][0]["url"]
    except Exception as e:
        st.error(f"エラーが発生しました: {e}")
        return None