Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import json
 import pandas as pd
 import random
+import os
 from groq import Groq
 from prompt_engine import PromptGenerator
 from utils import load_techniques, load_styles, parse_prompt_result
@@ -12,7 +13,8 @@ model_themes = {
     "llama-3.3-70b-versatile": {"primary": "#4b367c", "secondary": "#d1c4e9"},
     "meta-llama/llama-4-maverick-17b-128e-instruct": {"primary": "#00897b", "secondary": "#b2dfdb"},
     "mistral-saba-24b": {"primary": "#0277bd", "secondary": "#b3e5fc"},
-    "qwen-qwq-32b": {"primary": "#6a1b9a", "secondary": "#e1bee7"}
+    "qwen-qwq-32b": {"primary": "#6a1b9a", "secondary": "#e1bee7"},
+    "o4-mini-2025-04-16": {"primary": "#10a37f", "secondary": "#d2f4ea"}
 }
 
 # 選択されたモデル（初回読み込み時のデフォルト値）
@@ -52,6 +54,13 @@ st.markdown(f"""
 def get_groq_client():
     return Groq(api_key=st.secrets["GROQ_API_KEY"])
 
+# OpenAI APIキーの設定
+def setup_openai_api():
+    if "OPENAI_API_KEY" in st.secrets:
+        os.environ["OPENAI_API_KEY"] = st.secrets["OPENAI_API_KEY"]
+        return True
+    return False
+
 # データロード
 @st.cache_data
 def load_data():
@@ -61,6 +70,7 @@ def load_data():
 
 # 初期化
 client = get_groq_client()
+openai_available = setup_openai_api()
 techniques, styles = load_data()
 prompt_generator = PromptGenerator()
 
@@ -92,15 +102,22 @@ with st.sidebar:
     st.markdown("---")
     st.header("🧠 AIモデル")
 
+    # Claudeのsonnetをモデル選択に追加
+    model_options = [
+        "deepseek-r1-distill-llama-70b",
+        "llama-3.3-70b-versatile",
+        "meta-llama/llama-4-maverick-17b-128e-instruct",
+        "mistral-saba-24b", 
+        "qwen-qwq-32b"
+    ]
+    
+    # OpenAI APIキーが設定されている場合のみGPTモデルを選択肢に追加
+    if openai_available:
+        model_options.append("o4-mini-2025-04-16")
+    
     model = st.selectbox(
         "AIモデルを選択",
-        [
-            "deepseek-r1-distill-llama-70b",
-            "llama-3.3-70b-versatile",
-            "meta-llama/llama-4-maverick-17b-128e-instruct",
-            "mistral-saba-24b", 
-            "qwen-qwq-32b"
-        ],
+        model_options,
         index=4,  # デフォルトは qwen-qwq-32b
         help="使用するAIモデルを選択します。各モデルによって生成結果の特性が異なります"
     )
@@ -129,7 +146,8 @@ with st.sidebar:
         "llama-3.3-70b-versatile": "Meta AIの最新LLaMA 3.3。バランスの取れた高性能モデルで、多くのタスクに適しています。",
         "meta-llama/llama-4-maverick-17b-128e-instruct": "LLaMA 4 Maverickの指示チューニング版。小型ながら高性能です。",
         "mistral-saba-24b": "Mistral AIのSaba 24Bモデル。創造性と一貫性のバランスに優れています。",
-        "qwen-qwq-32b": "Qwen QWQモデル。創造的な表現力が高く、芸術的プロンプトに適しています。"
+        "qwen-qwq-32b": "Qwen QWQモデル。創造的な表現力が高く、芸術的プロンプトに適しています。",
+        "o4-mini-2025-04-16": "OpenAIの最新o4モデルのコンパクト版。高度なコンテキスト理解と創造的なプロンプト生成能力に優れています。"
     }
 
     st.caption(model_info.get(model, ""))
@@ -499,10 +517,16 @@ with tab4:
             ["hopeful", "melancholic", "anxious", "determined", "curious", "nostalgic", "dreamy", "tense"]
         )
 
+# OpenAI APIキーが設定されていない場合の警告
+if "o4-mini-2025-04-16" in model_options and not openai_available and model == "o4-mini-2025-04-16":
+    st.warning("OpenAI APIキーが設定されていないため、GPTモデルは使用できません。他のモデルを選択してください。")
+
 # 生成ボタン
 if st.button("🚀 プロンプトを生成", type="primary", use_container_width=True):
     if not user_input:
         st.error("描きたいイメージを入力してください")
+    elif model == "o4-mini-2025-04-16" and not openai_available:
+        st.error("OpenAI APIキーが設定されていないため、GPTモデルは使用できません。他のモデルを選択してください。")
     else:
         with st.spinner("AIがプロンプトを生成中..."):
             # パラメータの準備