Update media_plan_templater.py

media_plan_templater.py

@@ -344,6 +344,219 @@ def render(get_conn):
 
     package_opts = opts["packages"]
     rate_opts = opts["rates"]
+    # ─────────────────────────────────────────────────────────────────────────────
+    # Guided 5-row entry (independent controls, then write into the table)
+    # ─────────────────────────────────────────────────────────────────────────────
+    st.subheader("Guided entry (5 rows)")
+    st.caption("Select Package → Placement (dependent), pick Targeting presets, add Budget/Rate, then click **Populate table**.")
+    
+    # one-time init to hold the guided rows' state
+    if "guided_rows" not in st.session_state:
+        st.session_state["guided_rows"] = [
+            {
+                "Media plan": "",
+                "Budget": None,
+                "Rate": rate_opts[0] if rate_opts else "",
+                "Product": "",           # Package
+                "Placement": "",
+                "Details": "",
+                "Targeting": [],         # list for multiselect, will stringify as "A OR B"
+                "Notes": "",
+            }
+            for _ in range(DEFAULT_INPUT_ROWS)
+        ]
+    
+    def _placements_for(pkg: str) -> list[str]:
+        pkg = (pkg or "").strip()
+        return opts["placementMap"].get(pkg, [])
+    
+    def _details_for(pkg: str, placement: str) -> str:
+        # PACKAGE_EXIST → bullets from extraction; else chosen placement
+        exists = opts["packageExistsMap"].get((pkg or "").strip(), False)
+        if exists:
+            bullets = opts["extractionMap"].get((pkg or "").strip(), [])
+            return "\n".join(f"• {b}" for b in bullets) if bullets else ""
+        return (placement or "").strip()
+    
+    with st.form("guided_form_5rows", clear_on_submit=False):
+        for i in range(DEFAULT_INPUT_ROWS):
+            row_state = st.session_state["guided_rows"][i]
+    
+            st.markdown(f"**Row {i+1}**")
+            c1, c2, c3, c4 = st.columns([1.1, 1.1, 1.2, 1.2])
+            with c1:
+                row_state["Media plan"] = st.text_input(
+                    "Media plan",
+                    key=f"g_media_{i}",
+                    value=row_state["Media plan"],
+                    placeholder="e.g., MP-01",
+                )
+            with c2:
+                row_state["Budget"] = st.number_input(
+                    "Budget",
+                    key=f"g_budget_{i}",
+                    value=0.0 if row_state["Budget"] in (None, "") else float(row_state["Budget"]),
+                    min_value=0.0,
+                    step=100.0,
+                    format="%0.2f",
+                )
+            with c3:
+                row_state["Rate"] = st.selectbox(
+                    "Rate",
+                    options=rate_opts,
+                    key=f"g_rate_{i}",
+                    index=max(0, rate_opts.index(row_state["Rate"])) if (row_state["Rate"] in rate_opts) else 0,
+                )
+            with c4:
+                row_state["Product"] = st.selectbox(
+                    "Package",
+                    options=[""] + package_opts,
+                    key=f"g_pkg_{i}",
+                    index=([""] + package_opts).index(row_state["Product"]) if row_state["Product"] in ([""] + package_opts) else 0,
+                    help="Choose a Package first.",
+                )
+    
+            # dependent Placement (disabled when PACKAGE_EXIST = yes)
+            pkg = row_state["Product"]
+            exists = opts["packageExistsMap"].get((pkg or "").strip(), False)
+            choices = _placements_for(pkg)
+    
+            cc1, cc2 = st.columns([1.2, 2.0])
+            with cc1:
+                if exists:
+                    # disable Placement when package is a bundle
+                    st.selectbox(
+                        "Placement",
+                        options=["(package-defined elements; no placement selection)"],
+                        index=0,
+                        key=f"g_plc_disabled_{i}",
+                        disabled=True,
+                        help="This package defines its own elements.",
+                    )
+                    row_state["Placement"] = ""
+                else:
+                    # allow selection when package is not a bundle
+                    if choices:
+                        default_ix = choices.index(row_state["Placement"]) if row_state["Placement"] in choices else 0
+                        row_state["Placement"] = st.selectbox(
+                            "Placement",
+                            options=choices,
+                            index=default_ix,
+                            key=f"g_plc_{i}",
+                            help="Placement options depend on the selected Package.",
+                        )
+                    else:
+                        st.selectbox(
+                            "Placement",
+                            options=["(no placements for this package)"],
+                            index=0,
+                            key=f"g_plc_empty_{i}",
+                            disabled=True,
+                        )
+                        row_state["Placement"] = ""
+    
+            with cc2:
+                # Targeting multi-select (stores a list; will stringify later)
+                row_state["Targeting"] = st.multiselect(
+                    "Targeting (presets)",
+                    options=TARGETING_OPTIONS,
+                    default=[t for t in (row_state["Targeting"] or []) if t in TARGETING_OPTIONS],
+                    key=f"g_tgt_{i}",
+                    help="Pick one or more. Will be saved as 'A OR B'.",
+                )
+    
+            # Notes
+            row_state["Notes"] = st.text_input(
+                "Notes",
+                key=f"g_notes_{i}",
+                value=row_state["Notes"],
+                placeholder="Optional",
+            )
+    
+            # Details preview (derived)
+            details_preview = _details_for(pkg, row_state["Placement"])
+            row_state["Details"] = details_preview
+            st.caption("**Details preview**")
+            st.text_area(
+                label=f"Details (auto)",
+                key=f"g_details_preview_{i}",
+                value=details_preview,
+                height=80,
+                disabled=True,
+            )
+            st.divider()
+    
+        cleft, cright = st.columns([1, 1])
+        with cleft:
+            reset_guided = st.form_submit_button("Reset guided rows")
+        with cright:
+            populate = st.form_submit_button("Populate table")
+    
+        if reset_guided:
+            st.session_state["guided_rows"] = [
+                {
+                    "Media plan": "",
+                    "Budget": None,
+                    "Rate": rate_opts[0] if rate_opts else "",
+                    "Product": "",
+                    "Placement": "",
+                    "Details": "",
+                    "Targeting": [],
+                    "Notes": "",
+                }
+                for _ in range(DEFAULT_INPUT_ROWS)
+            ]
+            st.experimental_rerun()
+    
+        if populate:
+            # Validate rows & build DataFrame
+            issues = []
+            out_rows = []
+            for i, r in enumerate(st.session_state["guided_rows"], start=1):
+                pkg = (r["Product"] or "").strip()
+                if not pkg:
+                    # allow entirely blank rows to be skipped
+                    if not any((r["Media plan"], r["Budget"], r["Rate"], r["Placement"], r["Notes"], r["Targeting"])):
+                        continue
+                    issues.append(f"Row {i}: Package is required.")
+                    continue
+    
+                exists = opts["packageExistsMap"].get(pkg, False)
+                placement = (r["Placement"] or "").strip()
+                if not exists:
+                    if not placement:
+                        issues.append(f"Row {i}: Placement required for non-bundle package '{pkg}'.")
+                        continue
+    
+                # stringify targeting
+                tgt_str = " OR ".join(r["Targeting"]) if r.get("Targeting") else ""
+    
+                out_rows.append({
+                    "Media plan": r["Media plan"],
+                    "Budget": r["Budget"],
+                    "Rate": r["Rate"],
+                    "Product": pkg,
+                    "Details": _details_for(pkg, placement),
+                    "Targeting": tgt_str,
+                    "Notes": r["Notes"],
+                    "Placement": placement,
+                })
+    
+            if issues:
+                st.error("Please fix the following before populating:\n- " + "\n- ".join(issues))
+            else:
+                if not out_rows:
+                    st.warning("No rows to populate.")
+                else:
+                    df_new = pd.DataFrame(out_rows, dtype="object")
+                    # Push into the main input table (normalize + keep Placement column)
+                    canonical = _normalize_input_table(df_new[DEFAULT_INPUT_COLUMNS])
+                    canonical["Rate"] = df_new["Rate"].fillna("").astype(str)
+                    if "Placement" in df_new.columns:
+                        canonical["Placement"] = df_new["Placement"].fillna("").astype(str)
+                    st.session_state["media_plan_inputs"] = canonical
+                    st.success(f"Populated {len(canonical)} row(s) into the input table.")
+                    st.dataframe(canonical, use_container_width=True)
 
     # Column config (Product column labeled as Package; Placement shown as text in grid)
     column_config = {