Update app.py

app.py

@@ -5,20 +5,28 @@ import re
 from datetime import datetime
 from pathlib import Path
 import os
+from openai import OpenAI  # make sure `openai` is in requirements.txt
 
+# ---- OpenAI setup (explicit) -------------------------------------------------
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OPENAI_API_KEY:
+    raise RuntimeError(
+        "OPENAI_API_KEY is not set. In your Hugging Face Space, go to "
+        "Settings → Secrets and add OPENAI_API_KEY."
+    )
 
+# Optional: allow swapping models via secret/env
+OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-mini")
+openai_client = OpenAI(api_key=OPENAI_API_KEY)  # CrewAI also reads env automatically
 
-# --- Logo: convert GitHub page URL to raw if needed ---
+# ---- Logo helper --------------------------------------------------------------
 def to_raw_github(url: str) -> str:
     # Accepts both blob and raw URLs; converts blob → raw
     return url.replace("https://github.com/", "https://raw.githubusercontent.com/").replace("/blob/", "/")
 
 LOGO_URL = to_raw_github("https://github.com/Decoding-Data-Science/airesidency/blob/main/dds_logo.jpg")
 
-# ----------------------------
-# Agents
-# ----------------------------
+# ---- Agents ------------------------------------------------------------------
 lead_market_analyst = Agent(
     role="Lead Market Analyst",
     goal="Deliver sharp, data-driven market insights for the brand/product.",
@@ -46,7 +54,7 @@ creative_content_creator = Agent(
     verbose=True,
 )
 
-# Optional: focused social copywriter (only if toggled)
+# Optional social copywriter (only used if toggled)
 social_copywriter = Agent(
     role="Social Copywriter",
     goal="Turn strategy into platform-appropriate copy with strong hooks and clear CTAs.",
@@ -55,9 +63,7 @@ social_copywriter = Agent(
     verbose=False,
 )
 
-# ----------------------------
-# Core crew
-# ----------------------------
+# ---- Tasks & Crew -------------------------------------------------------------
 def run_marketing_crew(product_brand: str, target_audience: str, objective: str) -> str:
     topic = f"{product_brand} | Audience: {target_audience} | Objective: {objective}"
 
@@ -109,9 +115,7 @@ def run_marketing_crew(product_brand: str, target_audience: str, objective: str)
     )
     return crew.kickoff()
 
-# ----------------------------
-# Helpers
-# ----------------------------
+# ---- Helpers -----------------------------------------------------------------
 def _first_n_points(text: str, n: int = 5):
     lines = [l.strip() for l in text.splitlines() if l.strip()]
     bullets = []
@@ -132,9 +136,11 @@ def _hashtags(csv_tags: str) -> str:
 def _truncate_chars(s: str, max_chars: int) -> str:
     return s if len(s) <= max_chars else s[:max_chars - 1] + "…"
 
-# ----------------------------
-# Lightweight templates (no extra LLM call)
-# ----------------------------
+def _slugify(name: str) -> str:
+    # Replace non-word chars with underscores and trim
+    return re.sub(r"\W+", "_", (name or "").lower()).strip("_")
+
+# ---- Lightweight templates (no extra LLM call) --------------------------------
 def tpl_linkedin(strategy, brand, audience, objective, hashtags, max_words=180):
     hook = f"{brand}: a sharper path to {objective} for {audience}."
     pts = "\n".join([f"- {p}" for p in _first_n_points(strategy, 5)])
@@ -185,9 +191,7 @@ def tpl_article(strategy, brand, audience, objective, hashtags, max_words=800):
     words = text.split()
     return (" ".join(words[:max_words]) + "…") if len(words) > max_words else text
 
-# ----------------------------
-# Optional LLM copywriter
-# ----------------------------
+# ---- Optional LLM copywriter (uses the Social Copywriter agent) ---------------
 def llm_copywriter(strategy_text, brand, audience, objective, tone, platform,
                    hashtags, li_words, tweet_chars, article_words):
     limits = {
@@ -217,9 +221,7 @@ Requirements:
     crew = Crew(agents=[social_copywriter], tasks=[task], verbose=False)
     return crew.kickoff()
 
-# ----------------------------
-# Generation wrapper
-# ----------------------------
+# ---- Generation wrapper -------------------------------------------------------
 PLATFORMS = ["LinkedIn", "X (Twitter)", "Article"]  # Reduced for better UX
 
 def generate(product_brand, target_audience, objective,
@@ -237,31 +239,31 @@ def generate(product_brand, target_audience, objective,
         social = llm_copywriter(strategy, product_brand, target_audience, objective,
                                 tone, platform, hashtags,
                                 li_max_words, tweet_max_chars, article_max_words)
+
+        name_map = {"LinkedIn": "linkedin", "X (Twitter)": "tweet", "Article": "article"}
+        safe = _slugify(product_brand)
+        fname = f"{name_map.get(platform, 'post')}_{safe}.md"
+        out_path = Path(fname).resolve()
+        out_path.write_text(social, encoding="utf-8")
+        return strategy, social, str(out_path)
+
     else:
+        safe = _slugify(product_brand)
         if platform == "LinkedIn":
             social = tpl_linkedin(strategy, product_brand, target_audience, objective, hashtags, li_max_words)
-            fname = f"linkedin_{re.sub(r'\\W+','_',product_brand.lower())}.md"
+            fname = f"linkedin_{safe}.md"
         elif platform == "X (Twitter)":
             social = tpl_tweet(strategy, product_brand, target_audience, objective, hashtags, tweet_max_chars)
-            fname = f"tweet_{re.sub(r'\\W+','_',product_brand.lower())}.txt"
+            fname = f"tweet_{safe}.txt"
         else:  # Article
             social = tpl_article(strategy, product_brand, target_audience, objective, hashtags, article_max_words)
-            fname = f"article_{re.sub(r'\\W+','_',product_brand.lower())}.md"
+            fname = f"article_{safe}.md"
 
         out_path = Path(fname).resolve()
         out_path.write_text(social, encoding="utf-8")
         return strategy, social, str(out_path)
 
-    # Save LLM result with reasonable extension
-    name_map = {"LinkedIn": "linkedin", "X (Twitter)": "tweet", "Article": "article"}
-    fname = f"{name_map.get(platform,'post')}_{re.sub(r'\\W+','_',product_brand.lower())}.md"
-    out_path = Path(fname).resolve()
-    out_path.write_text(social, encoding="utf-8")
-    return strategy, social, str(out_path)
-
-# ----------------------------
-# Theming & Layout (2 columns with header)
-# ----------------------------
+# ---- Theming & Layout (2 columns with header) --------------------------------
 theme = gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")
 
 CUSTOM_CSS = """
@@ -295,8 +297,10 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
 
             with gr.Group(elem_classes="card"):
                 platform = gr.Dropdown(choices=PLATFORMS, value="LinkedIn", label="Platform")
-                tone = gr.Dropdown(choices=["Professional", "Friendly", "Bold", "Educational", "Conversational"],
-                                   value="Professional", label="Tone")
+                tone = gr.Dropdown(
+                    choices=["Professional", "Friendly", "Bold", "Educational", "Conversational"],
+                    value="Professional", label="Tone"
+                )
                 hashtags = gr.Textbox(label="Hashtags (comma separated)", placeholder="ai, generativeai, datascience")
                 use_llm = gr.Checkbox(value=False, label="Use LLM Copywriter (higher fidelity)")
 
@@ -325,9 +329,8 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
         outputs=[strategy_md, social_tb, download_file]
     )
 
-# Space-friendly launch
+# ---- Space-friendly launch ----------------------------------------------------
 if __name__ == "__main__":
-    # Space runners set HOST/PORT; default to 0.0.0.0:7860 for local
     host = os.getenv("HOST", "0.0.0.0")
     port = int(os.getenv("PORT", "7860"))
     demo.launch(server_name=host, server_port=port)