Update app.py

app.py

@@ -257,6 +257,91 @@ with tab3:
             help="画像の構図テクニックを指定します"
         )
     
+    # MidJourney特有のパラメータセクションを追加
+    st.markdown("### MidJourney/nijiJourney特有のパラメータ")
+    
+    # タブでパラメータをカテゴリ分け
+    param_tab1, param_tab2, param_tab3 = st.tabs(["スタイル参照", "パーソナライズ", "その他"])
+    
+    with param_tab1:
+        # スタイルリファレンス (--sref) 設定
+        use_style_reference = st.checkbox("スタイルリファレンス (--sref) を使用", 
+                                         help="別画像のスタイルを参照し、色調や質感を新規生成に適用します")
+        
+        if use_style_reference:
+            sref_type = st.radio("参照方法", ["ランダム", "画像URL", "コード指定"], horizontal=True)
+            
+            if sref_type == "ランダム":
+                st.info("ランダムに選ばれたスタイルが適用されます (--sref random)")
+                sref_value = "random"
+            elif sref_type == "画像URL":
+                sref_value = st.text_input("画像URL", help="スタイル参照する画像のURLを入力してください")
+            else:
+                sref_value = st.text_input("スタイルコード", help="以前生成されたsrefコードを入力 (例: 1234567)")
+            
+            # スタイルウェイト
+            style_weight = st.slider("スタイルウェイト (--sw)", 0, 1000, 100, 
+                                    help="参照画像の影響度を調整します。高いほど強く反映されます")
+            
+            # スタイルバージョン
+            style_version = st.radio("スタイルバージョン (--sv)", [1, 2, 3, 4], index=3, horizontal=True,
+                                    help="Style Referenceアルゴリズムのバージョンを選択します")
+    
+    with param_tab2:
+        # パーソナライゼーション (--p) 設定
+        use_personalization = st.checkbox("パーソナライゼーション (--p) を使用", 
+                                         help="自身の好みに学習させたプロファイルを適用します")
+        
+        if use_personalization:
+            personalization_type = st.radio("適用方法", ["デフォルト", "プロファイルID", "コード指定"], horizontal=True)
+            
+            if personalization_type == "デフォルト":
+                st.info("設定済みのデフォルトプロファイルが適用されます (--p)")
+                personalization_value = ""
+            elif personalization_type == "プロファイルID":
+                personalization_value = st.text_input("プロファイルID", help="使用するプロファイルIDを入力")
+            else:
+                personalization_value = st.text_input("パーソナライズコード", help="以前生成されたコードを入力")
+            
+            # パーソナライズ強度は --stylize を共用
+            st.caption("パーソナライズの適用度はスタイル強度 (--s) パラメータで調整されます")
+    
+    with param_tab3:
+        # その他のパラメータ
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            use_character_reference = st.checkbox("キャラクター参照 (--cref)", 
+                                               help="複数シーンで同一キャラクターを維持します")
+            
+            if use_character_reference:
+                cref_value = st.text_input("キャラクター参照URL", help="キャラクターが写っている画像のURLを入力")
+            
+            use_repeat = st.checkbox("繰り返し生成 (--repeat / --r)", 
+                                   help="同一プロンプトで複数グリッドを生成します")
+            
+            if use_repeat:
+                repeat_count = st.number_input("繰り返し回数", min_value=1, max_value=10, value=2)
+        
+        with col2:
+            use_image_weight = st.checkbox("画像プロンプトの影響度 (--iw)", 
+                                         help="画像プロンプト（イメージURL）の影響度を調整します")
+            
+            if use_image_weight:
+                image_weight = st.slider("画像ウェイト", 0.0, 2.0, 1.0, 0.1)
+            
+            generation_mode = st.radio("生成モード", 
+                                      ["デフォルト", "fast", "relax", "turbo", "draft"], 
+                                      index=0, horizontal=True,
+                                      help="GPU速度やコストを最適化するモードを選択")
+            
+            visibility = st.radio("共有設定", 
+                                ["デフォルト", "public", "stealth"], 
+                                index=0, horizontal=True,
+                                help="MidJourneyサイトへの���開/非公開を制御")
+    
+    # カスタムパラメータセクション
+    st.markdown("### カスタムパラメータ")
     advanced_params = st.text_area(
         "カスタムパラメータ (JSONフォーマット)", 
         "{}",
@@ -318,7 +403,7 @@ if st.button("🚀 プロンプトを生成", type="primary", use_container_widt
                 "input": user_input,
                 "mode": mode.lower(),
                 "complexity": complexity,
-                "model": model,  # 選択されたモデルを追加
+                "model": model,
                 "aspect_ratio": aspect_ratio,
                 "quality": quality,
                 "style": style,
@@ -338,11 +423,59 @@ if st.button("🚀 プロンプトを生成", type="primary", use_container_widt
                 }
             }
             
+            # MidJourney特有のパラメータを追加
+            
+            # スタイルリファレンス
+            if 'use_style_reference' in locals() and use_style_reference:
+                params["advanced"]["style_reference"] = {
+                    "use": True,
+                    "value": sref_value,
+                    "style_weight": style_weight,
+                    "style_version": style_version
+                }
+            
+            # パーソナライゼーション
+            if 'use_personalization' in locals() and use_personalization:
+                params["advanced"]["personalization"] = {
+                    "use": True,
+                    "type": personalization_type,
+                    "value": personalization_value
+                }
+            
+            # キャラクター参照
+            if 'use_character_reference' in locals() and use_character_reference:
+                params["advanced"]["character_reference"] = {
+                    "use": True,
+                    "value": cref_value
+                }
+            
+            # 繰り返し生成
+            if 'use_repeat' in locals() and use_repeat:
+                params["advanced"]["repeat"] = {
+                    "use": True,
+                    "count": repeat_count
+                }
+            
+            # 画像プロンプトの影響度
+            if 'use_image_weight' in locals() and use_image_weight:
+                params["advanced"]["image_weight"] = {
+                    "use": True,
+                    "value": image_weight
+                }
+            
+            # 生成モード
+            if 'generation_mode' in locals() and generation_mode != "デフォルト":
+                params["advanced"]["generation_mode"] = generation_mode
+            
+            # 共有設定
+            if 'visibility' in locals() and visibility != "デフォルト":
+                params["advanced"]["visibility"] = visibility
+            
             # 実験的機能の詳細
-            if use_pattern_interrupt and pattern_concepts:
+            if use_pattern_interrupt and 'pattern_concepts' in locals() and pattern_concepts:
                 params["experimental"]["pattern_concepts"] = [c.strip() for c in pattern_concepts.split(",")]
             
-            if use_emotion_matrix:
+            if use_emotion_matrix and 'primary_emotion' in locals() and 'secondary_emotion' in locals():
                 params["experimental"]["emotions"] = {
                     "primary": primary_emotion,
                     "secondary": secondary_emotion
@@ -380,4 +513,4 @@ if st.button("🚀 プロンプトを生成", type="primary", use_container_widt
 
 # フッター
 st.markdown("---")
-st.markdown("© 2025 AI Art Prompt Generator - Powered by Groq & Hugging Face")
+st.markdown("© 2025 AI Art Prompt Generator")