Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import re
 import streamlit as st
 from llm_groq import generate_post, transform_post, generate_hooks, DEFAULT_MODEL
 from prompts import build_quick_prompt, build_post_prompt, transform_instruction
@@ -8,7 +9,7 @@ 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 (Groq)")
 
-# Sidebar (unique keys not required here but safe to add if duplicated later)
+# Sidebar
 with st.sidebar:
     st.subheader("Groq & Decoding")
     model = st.selectbox("Model", [DEFAULT_MODEL, "llama-3.1-8b-instant", "mixtral-8x7b-32768"], index=0, key="sb_model")
@@ -21,6 +22,21 @@ tabs = st.tabs(["Quick Draft", "Pro Mode", "History"])
 if "history" not in st.session_state:
     st.session_state.history = []
 
+def quick_quality_fix(text, want_hashtags=True, allow_emoji=True):
+    lines = [l for l in text.strip().splitlines() if l.strip()]
+    if len(lines) < 4 or len(lines) > 7:
+        return None
+    if not allow_emoji:
+        text = re.sub(r"[^\w\s#.,:;%&()\-\+\[\]{}'\"/]", "", text)
+    tags = re.findall(r"#\w+", text)
+    if not want_hashtags and tags:
+        for t in tags:
+            text = text.replace(t, "")
+    if want_hashtags and len(tags) > 2:
+        for t in tags[2:]:
+            text = text.replace(t, "")
+    return text.strip()
+
 # Quick Draft
 with tabs[0]:
     idea, tone, words, variations, include_emoji, add_hashtags, language = quick_controls()
@@ -38,7 +54,18 @@ with tabs[0]:
                     for _ in range(variations):
                         raw = generate_post(prompt, model, temperature, top_p, max_tokens)
                         clean = dedupe_sentences(strip_labels(raw))
-                        posts.append(clean)
+                        fixed = quick_quality_fix(clean, want_hashtags=add_hashtags, allow_emoji=include_emoji)
+                        if fixed is None:
+                            corrective = (
+                                prompt + "\n\nRegenerate as 4–6 short lines, each under 18 words, "
+                                "include one concrete metric or date, "
+                                f"{'max 1 emoji' if include_emoji else 'no emojis'}, "
+                                f"{'1–2 niche hashtags at the end' if add_hashtags else 'no hashtags'}."
+                            )
+                            raw2 = generate_post(corrective, model, temperature, top_p, max_tokens)
+                            clean2 = dedupe_sentences(strip_labels(raw2))
+                            fixed = quick_quality_fix(clean2, want_hashtags=add_hashtags, allow_emoji=include_emoji) or clean2
+                        posts.append(fixed)
                 except Exception as e:
                     st.error(f"Generation failed: {e}")
                     posts = []
@@ -94,7 +121,7 @@ with tabs[1]:
 
     # Refinements
     col1,col2,col3,col4,col5 = st.columns(5)
-    def refine(kind, key_suffix):
+    def refine(kind):
         if st.session_state.get("history") and st.session_state.history[-1].get("post"):
             try:
                 instr = transform_instruction(kind)
@@ -103,11 +130,11 @@ with tabs[1]:
             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 col1.button("Shorter", key="pro_shorter"): refine("shorter")
+    if col2.button("Punchier hook", key="pro_punchy"): refine("punchier")
+    if col3.button("Add data point", key="pro_adddata"): refine("add_data")
+    if col4.button("No emojis", key="pro_noemoji"): refine("less_emoji")
+    if col5.button("Add hashtags", key="pro_addtags"): refine("add_tags")
 
     if st.session_state.get("history") and st.session_state.history[-1].get("post"):
         st.write(st.session_state.history[-1]["post"])