Update app.py

app.py

@@ -5,6 +5,19 @@ from groq import Groq
 from prompt_engine import PromptGenerator
 from utils import load_techniques, load_styles, parse_prompt_result
 
+# モデル特性に基づくカラーテーマ設定
+model_themes = {
+    "deepseek-r1-distill-llama-70b": {"primary": "#3366cc", "secondary": "#b3c6ff"},
+    "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"}
+}
+
+# 選択されたモデル（初回読み込み時のデフォルト値）
+current_model = "llama-3.3-70b-versatile"
+theme = model_themes.get(current_model, {"primary": "#4b367c", "secondary": "#d1c4e9"})
+
 # ページ設定
 st.set_page_config(
     page_title="AI Art Prompt Generator",
@@ -13,6 +26,26 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
+# カスタムCSSの適用
+st.markdown(f"""
+<style>
+    .stApp {{
+        background-color: {theme["secondary"]}20;  /* 20%の透明度 */
+    }}
+    .stButton>button {{
+        background-color: {theme["primary"]};
+        color: white;
+    }}
+    .stTabs [data-baseweb="tab-list"] {{
+        border-bottom-color: {theme["primary"]}40;
+    }}
+    .stTabs [aria-selected="true"] {{
+        color: {theme["primary"]} !important;
+        border-bottom-color: {theme["primary"]} !important;
+    }}
+</style>
+""", unsafe_allow_html=True)
+
 # Groq APIクライアント設定
 @st.cache_resource
 def get_groq_client():
@@ -54,6 +87,52 @@ with st.sidebar:
         help="生成されるプロンプトの複雑さを調整します"
     )
     
+    # モデル選択セクションを追加
+    st.markdown("---")
+    st.header("🧠 AIモデル")
+    
+    model = st.selectbox(
+        "Groqモデルを選択",
+        [
+            "deepseek-r1-distill-llama-70b",
+            "llama-3.3-70b-versatile",
+            "meta-llama/llama-4-maverick-17b-128e-instruct",
+            "mistral-saba-24b", 
+            "qwen-qwq-32b"
+        ],
+        index=1,  # デフォルトは llama-3.3-70b-versatile
+        help="使用するGroqのAIモデルを選択します。各モデルによって生成結果の特性が異なります"
+    )
+    
+    # モデル選択時に変数を更新
+    if model != current_model:
+        current_model = model
+        theme = model_themes.get(current_model, {"primary": "#4b367c", "secondary": "#d1c4e9"})
+        
+        # テーマの更新（実際には再読み込みが必要）
+        st.markdown(f"""
+        <style>
+            .stApp {{
+                background-color: {theme["secondary"]}20;
+            }}
+            .stButton>button {{
+                background-color: {theme["primary"]};
+                color: white;
+            }}
+        </style>
+        """, unsafe_allow_html=True)
+    
+    # モデル情報の表示
+    model_info = {
+        "deepseek-r1-distill-llama-70b": "DeepSeekのLLaMA 70Bベースモデル。知識と推論能力に優れています。",
+        "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モデル。創造的な表現力が高く、芸術的プロンプトに適しています。"
+    }
+    
+    st.caption(model_info.get(model, ""))
+    
     st.markdown("---")
     
     st.header("🧙‍♂️ テクニック設定")
@@ -239,6 +318,7 @@ if st.button("🚀 プロンプトを生成", type="primary", use_container_widt
                 "input": user_input,
                 "mode": mode.lower(),
                 "complexity": complexity,
+                "model": model,  # 選択されたモデルを追加
                 "aspect_ratio": aspect_ratio,
                 "quality": quality,
                 "style": style,
@@ -291,6 +371,9 @@ if st.button("🚀 プロンプトを生成", type="primary", use_container_widt
             with col2:
                 st.caption("MidJourney/nijiJourneyにこのプロンプトを貼り付けて使用してください")
             
+            # 使用したモデルの表示を追加
+            st.caption(f"使用モデル: **{model}**")
+            
             if explanation:
                 st.markdown("### プロンプト解説")
                 st.markdown(explanation)