Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,7 +19,6 @@ from design_generation import (
     OUTPUT_DIR,
     analyze_jewellery,
     create_zip_from_images,
-    generate_auto_design_image,
     generate_design_directions,
     generate_final_images,
     generate_six_campaign_images,
@@ -1162,9 +1161,9 @@ with tab_research:
         r_done.append(1)
     if "processed_product_refs" in st.session_state:
         r_done.append(2)
-    if "design_options_per_product" in st.session_state or "auto_design_results" in st.session_state:
+    if "design_options_per_product" in st.session_state:
         r_done.append(3)
-    if "all_final_images" in st.session_state or "auto_design_results" in st.session_state:
+    if "all_final_images" in st.session_state:
         r_done.append(4)
     r_current = (max(r_done) + 1) if r_done else 1
 
@@ -1230,8 +1229,7 @@ with tab_research:
                         st.session_state["researched_products"] = products
                         st.session_state["research_keywords_order"] = keywords
                         for key in ["processed_product_refs", "design_options_per_product",
-                                    "frozen_selected_campaigns", "all_final_images",
-                                    "auto_design_results"]:
+                                    "frozen_selected_campaigns", "all_final_images"]:
                             st.session_state.pop(key, None)
                         status_ui.update(label=f"Done — {len(products)} products found.", state="complete")
                     st.rerun()
@@ -1313,8 +1311,6 @@ with tab_research:
                                 if checked:
                                     selected_product_ids.add(pid)
 
-                    st.markdown("---")
-
                 st.markdown("---")
                 col_proc, col_cnt = st.columns([1, 3])
                 with col_proc:
@@ -1337,8 +1333,7 @@ with tab_research:
                         if p.get("local_image_path") and Path(p["local_image_path"]).exists()
                     ]
                     st.session_state["processed_product_refs"] = processed_refs
-                    for key in ["design_options_per_product", "frozen_selected_campaigns",
-                                "all_final_images", "auto_design_results"]:
+                    for key in ["design_options_per_product", "frozen_selected_campaigns", "all_final_images"]:
                         st.session_state.pop(key, None)
                     st.rerun()
 
@@ -1352,28 +1347,10 @@ with tab_research:
         ):
             st.markdown(
                 f"**{len(processed_refs)} product(s)** ready. "
-                "Generate 5 design direction previews per product, then pick one direction per product."
+                "Generate 6 design direction previews per product, then pick one direction per product."
             )
 
-            col_gen, col_auto = st.columns(2)
-            with col_gen:
-                gen_designs_btn = st.button(
-                    "Generate Design Directions",
-                    type="primary",
-                    key="btn_gen_designs",
-                    use_container_width=True,
-                    help="Generate 5 direction previews per product, then pick one per product.",
-                )
-            with col_auto:
-                auto_gen_btn = st.button(
-                    "Auto-Generate All Designs",
-                    type="primary",
-                    key="btn_auto_gen",
-                    use_container_width=True,
-                    help="Auto-generate one design per product and show results grouped by keyword.",
-                )
-
-            if gen_designs_btn:
+            if st.button("Generate Design Directions", type="primary", key="btn_gen_designs"):
                 design_options_per_product: Dict[str, Any] = {}
 
                 for item in processed_refs:
@@ -1407,43 +1384,10 @@ with tab_research:
                             st.error(f"{pname}: {e}")
 
                 st.session_state["design_options_per_product"] = design_options_per_product
-                st.session_state.pop("auto_design_results", None)
                 for key in ["frozen_selected_campaigns", "all_final_images"]:
                     st.session_state.pop(key, None)
                 st.rerun()
 
-            if auto_gen_btn:
-                auto_design_results: Dict[str, List[Dict[str, Any]]] = {}
-                total = len(processed_refs)
-                progress_bar = st.progress(0, text=f"Generating designs — 0 / {total} products…")
-
-                for idx, item in enumerate(processed_refs):
-                    product = item["product"]
-                    image_path = item["image_path"]
-                    keyword = product.get("source_keyword", "Other")
-                    pname = product.get("product_name", "Product")
-
-                    progress_bar.progress(
-                        idx / total if total else 0,
-                        text=f"Generating design {idx + 1}/{total}: {pname[:40]}…",
-                    )
-
-                    try:
-                        result = generate_auto_design_image(image_path)
-                        auto_design_results.setdefault(keyword, []).append({
-                            "product": product,
-                            "design_path": result["design_path"],
-                            "design_name": result["design_name"],
-                        })
-                    except Exception as e:
-                        st.warning(f"Skipped {pname}: {e}")
-
-                progress_bar.progress(1.0, text=f"Done — {total} designs generated.")
-                st.session_state["auto_design_results"] = auto_design_results
-                for key in ["design_options_per_product", "frozen_selected_campaigns", "all_final_images"]:
-                    st.session_state.pop(key, None)
-                st.rerun()
-
             if "design_options_per_product" in st.session_state:
                 design_opts = st.session_state["design_options_per_product"]
                 selected_campaigns: Dict[str, Any] = {}
@@ -1455,7 +1399,7 @@ with tab_research:
                     brand = product.get("brand", "")
 
                     st.markdown(f"#### {pname} — {brand}")
-                    st.caption("Review the 5 design directions, then choose one below.")
+                    st.caption("Review the 6 design directions, then choose one below.")
 
                     d_cols = st.columns(3)
                     for idx, design in enumerate(design_options):
@@ -1565,52 +1509,6 @@ with tab_research:
 
                 st.divider()
 
-    # ── Step 4 (Auto) · Designs grouped by keyword ───────────────────────────
-    if "auto_design_results" in st.session_state:
-        auto_results = st.session_state["auto_design_results"]
-        kw_order = st.session_state.get(
-            "research_keywords_order",
-            list(auto_results.keys()),
-        )
-        total_designs = sum(len(v) for v in auto_results.values())
-
-        with st.expander(
-            f"Step 4 · Auto-Generated Designs by Keyword  ({total_designs} designs across {len(auto_results)} keyword(s))",
-            expanded=True,
-        ):
-            for kw in kw_order:
-                entries = auto_results.get(kw, [])
-                if not entries:
-                    continue
-
-                st.markdown(f"### {kw}")
-                n_cols = min(len(entries), 5)
-                cols = st.columns(n_cols)
-                for i, entry in enumerate(entries):
-                    design_path = entry.get("design_path", "")
-                    with cols[i % n_cols]:
-                        if design_path and Path(design_path).exists():
-                            st.image(design_path, use_container_width=True)
-                            pname = entry["product"].get("product_name", "")
-                            label = entry.get("design_name", "")
-                            st.caption(
-                                f"{label}"
-                                + (f" · {pname[:22]}…" if len(pname) > 22 else (f" · {pname}" if pname else ""))
-                            )
-                            with open(design_path, "rb") as f:
-                                st.download_button(
-                                    "Download",
-                                    data=f,
-                                    file_name=Path(design_path).name,
-                                    mime="image/jpeg",
-                                    key=f"dl_auto_{kw}_{i}",
-                                    use_container_width=True,
-                                )
-                        else:
-                            st.caption("Image unavailable")
-
-                st.divider()
-
 
 # ══════════════════════════════════════════════════════════════════════════════
 # UPLOAD WORKFLOW
@@ -1697,7 +1595,7 @@ with tab_upload:
         campaign_options = st.session_state["upload_campaign_options"]
 
         with st.expander("Step 2 · Choose a Design Direction", expanded=(u_current == 2)):
-            st.markdown("Review the 5 design directions and select one to generate your final campaign images.")
+            st.markdown("Review the 6 design directions and select one to generate your final campaign images.")
 
             d_cols = st.columns(3)
             for idx, campaign in enumerate(campaign_options):