Update ui.py

ui.py

@@ -22,13 +22,16 @@ 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_json  # ← ここを変更
-
+from app.openai_client import openai_chat_json
 
 # 初期化
 init_db()
 _seasonality_cache: Dict[str, SeasonalityModel] = {}
 
+# 固定カラム（常にこの形で返す）
+GENERATE_COLUMNS = ["variant_id", "status", "rejection_reason", "text"]
+REPORT_COLUMNS   = ["variant_id","impressions","clicks","conversions","ctr","cvr","expected_value"]
+
 # JSONモード前提の厳格プロンプト
 GEN_SYSTEM = """
 あなたは日本語広告コピーのプロフェッショナルコピーライターです。
@@ -88,7 +91,6 @@ async def ui_generate(campaign_id: str, brand: str, product: str, target: str, t
         k=k_variants,
     )
 
-    # ← JSONモードを使用
     data = await openai_chat_json(
         [
             {"role": "system", "content": GEN_SYSTEM},
@@ -98,13 +100,9 @@ async def ui_generate(campaign_id: str, brand: str, product: str, target: str, t
         max_tokens=1200,
     )
 
-    # 許容: {"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]:
+    for it in (items[:k_variants] if isinstance(items, list) else []):
         headline = (it.get("headline") or "").strip()
         body = (it.get("body") or "").strip()
         text = f"{headline}\n{body}".strip()
@@ -134,7 +132,8 @@ async def ui_generate(campaign_id: str, brand: str, product: str, target: str, t
             "text": text,
         })
 
-    df = pd.DataFrame(rows, columns=["variant_id", "status", "rejection_reason", "text"]) if rows else pd.DataFrame()
+    # 常に固定カラムで返す（空でもカラムを持つDataFrame）
+    df = pd.DataFrame(rows, columns=GENERATE_COLUMNS)
     return df
 
 def ui_serve(campaign_id: str, hour: int, segment: str):
@@ -180,7 +179,8 @@ def ui_report(campaign_id: str):
             "cvr": round(cvr, 4),
             "expected_value": round(ev, 6),
         })
-    df = pd.DataFrame(rows, columns=["variant_id","impressions","clicks","conversions","ctr","cvr","expected_value"]) if rows else pd.DataFrame()
+    # 常に固定カラムで返す（空でもカラムを持つDataFrame）
+    df = pd.DataFrame(rows, columns=REPORT_COLUMNS)
     return df
 
 def ui_check(text: str):
@@ -211,7 +211,7 @@ with gr.Blocks(title="AdCopy MAB Optimizer", fill_height=True) as demo:
         tone = gr.Textbox(label="トーン", value="エビデンス重視で安心感")
         ng_words = gr.Textbox(label="NGワード（改行区切り）", value="治る\n奇跡")
         btn_gen = gr.Button("広告案を生成＆審査＆保存")
-        table_gen = gr.Dataframe(headers=["variant_id","status","rejection_reason","text"], interactive=False)
+        table_gen = gr.Dataframe(headers=GENERATE_COLUMNS, interactive=False)
         btn_gen.click(ui_generate, [campaign_id, brand, product, target, tone, k_variants, ng_words, value_per_conv], [table_gen])
 
     with gr.Tab("2) Serve & Feedback"):
@@ -234,7 +234,7 @@ with gr.Blocks(title="AdCopy MAB Optimizer", fill_height=True) as demo:
     with gr.Tab("3) Report"):
         campaign_id3 = gr.Textbox(label="campaign_id", value="cmp-demo")
         btn_rep = gr.Button("更新")
-        table_rep = gr.Dataframe(headers=["variant_id","impressions","clicks","conversions","ctr","cvr","expected_value"], interactive=False)
+        table_rep = gr.Dataframe(headers=REPORT_COLUMNS, interactive=False)
         btn_rep.click(ui_report, [campaign_id3], [table_rep])
 
     with gr.Tab("4) Compliance Check"):