Update ui.py

ui.py

@@ -22,15 +22,30 @@ from app.storage import (
 from app.bandit import ThompsonBandit
 from app.forecast import SeasonalityModel
 from app.compliance import rule_based_check, llm_check_and_fix
-from app.openai_client import openai_chat
+from app.openai_client import openai_chat_json  # ← ここを変更
+
 
 # 初期化
 init_db()
 _seasonality_cache: Dict[str, SeasonalityModel] = {}
 
+# JSONモード前提の厳格プロンプト
 GEN_SYSTEM = """
 あなたは日本語広告コピーのプロフェッショナルコピーライターです。
-出力はJSON配列（各要素は{\"headline\":..., \"body\":...}）のみで返してください。句読点や記号は自然に。誇大・断定は避け、事実ベースで魅力を伝えます。
+出力は**次のJSONオブジェクトのみ**で厳密に返してください。余計な文章・説明・前置きは禁止です。
+
+形式:
+{
+  "variants": [
+    {"headline": "全角15-25字程度", "body": "全角40-90字程度"},
+    ...
+  ]
+}
+
+ルール:
+- 医薬効能の断定、100%、永久、即効、根拠のない数値などの誇大表現は禁止
+- CTAは自然に
+- 日本語で、句読点や記号は自然に
 """
 
 GEN_USER_TEMPLATE = """
@@ -40,10 +55,10 @@ GEN_USER_TEMPLATE = """
 トーン: {tone}
 制約: {constraints}
 生成本数: {k}
-条件:
-- 1本あたり見出し（全角15-25字目安）+ 本文（全角40-90字目安）
-- 禁止: 医薬効能の断定、100%、永久、即効、根拠のない数値
-- CTAは自然に
+
+要件:
+- "variants" 配列の要素数は **ちょうど {k}** 件にしてください
+- 各要素は {"headline": "...", "body": "..."} のみ
 """
 
 def _seasonal(campaign_id: str) -> SeasonalityModel:
@@ -73,16 +88,20 @@ async def ui_generate(campaign_id: str, brand: str, product: str, target: str, t
         k=k_variants,
     )
 
-    raw = await openai_chat([
-        {"role": "system", "content": GEN_SYSTEM},
-        {"role": "user", "content": user}
-    ], temperature=0.8, max_tokens=800)
+    # ← JSONモードを使用
+    data = await openai_chat_json(
+        [
+            {"role": "system", "content": GEN_SYSTEM},
+            {"role": "user", "content": user},
+        ],
+        temperature=0.3,
+        max_tokens=1200,
+    )
 
-    try:
-        items = json.loads(raw)
-        assert isinstance(items, list)
-    except Exception:
-        raise gr.Error("LLM出力のJSONパースに失敗しました。プロンプトを短くするか、再実行してください。")
+    # 許容: {"variants":[...] } もしくはリストそのもの（保険）
+    items = data.get("variants", data if isinstance(data, list) else [])
+    if not isinstance(items, list) or not items:
+        raise gr.Error("JSON応答の 'variants' が取得できませんでした。もう一度お試しください。")
 
     rows = []
     for it in items[:k_variants]: