Update dashboard.py

dashboard.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-import pandas as pd
 import gradio as gr
 import plotly.express as px
 from data import read_events, aggregate
@@ -16,9 +15,8 @@ def ui_refresh_tables():
 def ui_recommend():
     agg = aggregate()
     if agg.empty:
-        return {"message": "No data yet. Upload or POST /api/ingest first."}
-    rec = BANDIT.recommend(agg)
-    return rec
+        return {"message": "No data yet. POST /api/ingest first."}
+    return BANDIT.recommend(agg)
 
 def ui_plot_posteriors(medium: str):
     agg = aggregate()
@@ -27,18 +25,16 @@ def ui_plot_posteriors(medium: str):
     g = agg[agg["medium"].astype(str) == str(medium)].copy()
     if g.empty:
         return gr.update(visible=False), f"No data for medium={medium}"
-
-    # 事後平均の棒グラフ（Laplace 平滑 CTR）
     g["ctr"] = (g["clicks"] + 1) / (g["impressions"] + 2)
-    fig = px.bar(g, x="creative", y="ctr", color="is_control", barmode="group", title=f"CTR (Laplace) by creative @ {medium}")
+    fig = px.bar(g, x="creative", y="ctr", color="is_control", barmode="group",
+                 title=f"CTR (Laplace) by creative @ {medium}")
     return gr.Plot(fig), ""
 
 def ui_fit_uplift():
     agg = aggregate()
     if agg.empty:
         return {"message": "No data"}
-    res = fit_uplift_binary(agg)
-    return res
+    return fit_uplift_binary(agg)
 
 def build_ui():
     with gr.Blocks(title="AdCopy MAB Optimizer Pro") as demo:
@@ -56,8 +52,7 @@ def build_ui():
                 medium = gr.Textbox(label="Medium for Plot", value="FB")
                 plot = gr.Plot(visible=False)
                 msg = gr.Markdown()
-            plot_btn = gr.Button("Plot CTR by Creative")
-            plot_btn.click(ui_plot_posteriors, inputs=[medium], outputs=[plot, msg])
+            gr.Button("Plot CTR by Creative").click(ui_plot_posteriors, inputs=[medium], outputs=[plot, msg])
         with gr.Tab("Uplift (Causal)"):
             cbtn = gr.Button("Fit Uplift Model")
             cout = gr.JSON()