Update app.py

app.py

@@ -11,7 +11,6 @@ import gradio as gr
 from crewai import Agent, Task, Crew
 
 # --- Optional: ensure key exists (CrewAI will pick it up) ---
-# If the key is missing, we still allow template-mode to work.
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
 
 # --- DDS logo (raw GitHub URL conversion) ---
@@ -20,6 +19,12 @@ def to_raw_github(url: str) -> str:
 
 LOGO_URL = to_raw_github("https://github.com/Decoding-Data-Science/airesidency/blob/main/dds_logo.jpg")
 
+# --- Helper: safe filename slug ---
+def slugify(s: str) -> str:
+    # replace non-word chars with underscores, collapse repeats, strip edges
+    slug = re.sub(r"\W+", "_", (s or "").lower())
+    return slug.strip("_") or "post"
+
 # ----------------------------
 # Agents (same logic)
 # ----------------------------
@@ -236,6 +241,8 @@ def generate(product_brand, target_audience, objective,
     # 1) Run the main Crew once
     strategy = run_marketing_crew(product_brand, target_audience, objective)
 
+    product_slug = slugify(product_brand)
+
     # 2) Copy creation
     if use_llm and OPENAI_API_KEY:
         social = llm_copywriter(strategy, product_brand, target_audience, objective,
@@ -244,13 +251,13 @@ def generate(product_brand, target_audience, objective,
     else:
         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_{product_slug}.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_{product_slug}.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_{product_slug}.md"
 
         out_path = Path(fname).resolve()
         out_path.write_text(social, encoding="utf-8")
@@ -258,7 +265,7 @@ def generate(product_brand, target_audience, objective,
 
     # Save LLM result
     name_map = {"LinkedIn": "linkedin", "X (Twitter)": "tweet", "Article": "article"}
-    fname = f"{name_map.get(platform,'post')}_{re.sub(r'\\W+','_',product_brand.lower())}.md"
+    fname = f"{name_map.get(platform,'post')}_{product_slug}.md"
     out_path = Path(fname).resolve()
     out_path.write_text(social, encoding="utf-8")
     return strategy, social, str(out_path)