Update app.py

app.py

@@ -6,26 +6,25 @@ from data_utils import load_posts, extract_keywords, dedupe_sentences, strip_lab
 from ui_components import quick_controls, pro_controls
 
 st.set_page_config(page_title="LinkedIn Post Generator — Groq", layout="centered")
-st.title("LinkedIn Post Generator — Quick & Pro")
+st.title("LinkedIn Post Generator — Quick & Pro (Groq)")
 
-# Sidebar
+# Sidebar (unique keys not required here but safe to add if duplicated later)
 with st.sidebar:
     st.subheader("Groq & Decoding")
-    model = st.selectbox("Model", [DEFAULT_MODEL, "llama-3.1-8b-instant", "mixtral-8x7b-32768"], index=0)
-    temperature = st.slider("Temperature", 0.1, 1.2, 0.6, 0.05)
-    top_p = st.slider("Top‑p", 0.1, 1.0, 0.9, 0.05)
+    model = st.selectbox("Model", [DEFAULT_MODEL, "llama-3.1-8b-instant", "mixtral-8x7b-32768"], index=0, key="sb_model")
+    temperature = st.slider("Temperature", 0.1, 1.2, 0.6, 0.05, key="sb_temp")
+    top_p = st.slider("Top‑p", 0.1, 1.0, 0.9, 0.05, key="sb_topp")
     st.markdown("Set GROQ_API_KEY in Space → Settings → Variables & Secrets.")
 
 tabs = st.tabs(["Quick Draft", "Pro Mode", "History"])
 
-# Session memory
 if "history" not in st.session_state:
     st.session_state.history = []
 
 # Quick Draft
 with tabs[0]:
     idea, tone, words, variations, include_emoji, add_hashtags, language = quick_controls()
-    if st.button("Generate"):
+    if st.button("Generate", key="qd_generate"):
         if not os.getenv("GROQ_API_KEY"):
             st.error("GROQ_API_KEY missing.")
         elif not idea.strip():
@@ -46,14 +45,14 @@ with tabs[0]:
             for i, p in enumerate(posts, start=1):
                 st.markdown(f"#### Post {i}")
                 st.write(p)
-                st.download_button(f"Download Post {i}", p, file_name=f"post_{i}.txt")
+                st.download_button(f"Download Post {i}", p, file_name=f"post_{i}.txt", key=f"qd_dl_{i}")
             if posts:
                 st.session_state.history.append({"mode":"quick","idea":idea,"tone":tone,"words":words,"posts":posts})
 
 # Pro Mode
 with tabs[1]:
     st.markdown("Upload CSV/JSON of past posts (must include 'text') to auto-extract keywords (optional).")
-    uploaded = st.file_uploader("Upload dataset", type=["csv","json"])
+    uploaded = st.file_uploader("Upload dataset", type=["csv","json"], key="pro_upload")
     defaults = {"topic":"AI agent playbooks for startup ops","audience":"SaaS founders in early stage"}
     topic, purpose, audience, tone2, language2, evidence, style_text = pro_controls(defaults)
     keywords = []
@@ -64,22 +63,21 @@ with tabs[1]:
             st.success(f"Loaded {len(df)} posts. Extracted keywords: {', '.join(keywords[:8]) or '—'}")
         except Exception as e:
             st.error(f"Dataset error: {e}")
-    chosen_hook = ""
-    if st.button("Suggest 5 hooks"):
+
+    if st.button("Suggest 5 hooks", key="pro_hooks_btn"):
         try:
             hooks = generate_hooks(topic, audience, tone2, 5, model, temperature, top_p, 200)
             st.code(hooks)
-            chosen_hook = st.text_input("Chosen opening line (optional)", value=chosen_hook)
         except Exception as e:
             st.error(f"Hook generation failed: {e}")
-    chosen_hook = st.text_input("Chosen opening line (optional)")  # visible even if not generated
 
-    outcome = st.text_input("Desired outcome (e.g., 10 demo requests this week)", value="")
-    extra_detail = st.text_input("One concrete detail (e.g., 'onboarding 14→3 days')", value="")
+    chosen_hook = st.text_input("Chosen opening line (optional)", key="pro_chosen_hook")
+    outcome = st.text_input("Desired outcome (e.g., 10 demo requests this week)", value="", key="pro_outcome")
+    extra_detail = st.text_input("One concrete detail (e.g., 'onboarding 14→3 days')", value="", key="pro_detail")
     clarifier_notes = "\n".join([f"Outcome: {outcome}" if outcome else "", f"Detail: {extra_detail}" if extra_detail else ""]).strip()
     style_cues = [s.strip() for s in style_text.splitlines() if s.strip()][:4]
 
-    if st.button("Generate Post (Pro)"):
+    if st.button("Generate Post (Pro)", key="pro_generate"):
         if not os.getenv("GROQ_API_KEY"):
             st.error("GROQ_API_KEY missing.")
         else:
@@ -89,22 +87,31 @@ with tabs[1]:
                 post = dedupe_sentences(strip_labels(raw))
                 st.success("Post")
                 st.write(post)
-                st.download_button("Download (.txt)", post, file_name="linkedin_post.txt")
+                st.download_button("Download (.txt)", post, file_name="linkedin_post.txt", key="pro_download")
                 st.session_state.history.append({"mode":"pro","topic":topic,"audience":audience,"post":post})
             except Exception as e:
                 st.error(f"Generation failed: {e}")
 
-    # Refinements (transformations)
-    if st.button("Shorter"):
+    # Refinements
+    col1,col2,col3,col4,col5 = st.columns(5)
+    def refine(kind, key_suffix):
         if st.session_state.get("history") and st.session_state.history[-1].get("post"):
             try:
-                instr = transform_instruction("shorter")
+                instr = transform_instruction(kind)
                 raw = transform_post(instr, st.session_state.history[-1]["post"], model, temperature, top_p, 500)
                 st.session_state.history[-1]["post"] = dedupe_sentences(strip_labels(raw))
-                st.write(st.session_state.history[-1]["post"])
             except Exception as e:
                 st.error(f"Refinement failed: {e}")
 
+    if col1.button("Shorter", key="pro_shorter"): refine("shorter","shorter")
+    if col2.button("Punchier hook", key="pro_punchy"): refine("punchier","punchy")
+    if col3.button("Add data point", key="pro_adddata"): refine("add_data","adddata")
+    if col4.button("No emojis", key="pro_noemoji"): refine("less_emoji","noemoji")
+    if col5.button("Add hashtags", key="pro_addtags"): refine("add_tags","addtags")
+
+    if st.session_state.get("history") and st.session_state.history[-1].get("post"):
+        st.write(st.session_state.history[-1]["post"])
+
 # History
 with tabs[2]:
     if not st.session_state.history: