Update app.py

app.py

@@ -1,32 +1,23 @@
-# app.py — Hugging Face Spaces (Gradio) for DDS Strategy → Social Generator
-# Two-column UI, DDS logo, reduced dropdown (LinkedIn / X / Article)
-# Secrets: set OPENAI_API_KEY in Spaces -> Settings -> Secrets
-
-import os
+# app.py
+from crewai import Agent, Task, Crew
+import gradio as gr
 import re
 from datetime import datetime
 from pathlib import Path
+import os
 
-import gradio as gr
-from crewai import Agent, Task, Crew
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
-# --- Optional: ensure key exists (CrewAI will pick it up) ---
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
 
-# --- DDS logo (raw GitHub URL conversion) ---
+# --- Logo: convert GitHub page URL to raw if needed ---
 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")
 
-# --- 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)
+# Agents
 # ----------------------------
 lead_market_analyst = Agent(
     role="Lead Market Analyst",
@@ -55,7 +46,7 @@ creative_content_creator = Agent(
     verbose=True,
 )
 
-# Optional: focused social copywriter (used only if toggled)
+# Optional: focused social copywriter (only if toggled)
 social_copywriter = Agent(
     role="Social Copywriter",
     goal="Turn strategy into platform-appropriate copy with strong hooks and clear CTAs.",
@@ -65,7 +56,7 @@ social_copywriter = Agent(
 )
 
 # ----------------------------
-# Core crew (unchanged)
+# Core crew
 # ----------------------------
 def run_marketing_crew(product_brand: str, target_audience: str, objective: str) -> str:
     topic = f"{product_brand} | Audience: {target_audience} | Objective: {objective}"
@@ -116,7 +107,7 @@ def run_marketing_crew(product_brand: str, target_audience: str, objective: str)
         tasks=[market_analysis_task, strategy_task, creative_task],
         verbose=True
     )
-    return crew.kickoff()  # full text output
+    return crew.kickoff()
 
 # ----------------------------
 # Helpers
@@ -142,7 +133,7 @@ def _truncate_chars(s: str, max_chars: int) -> str:
     return s if len(s) <= max_chars else s[:max_chars - 1] + "…"
 
 # ----------------------------
-# Templates (no extra LLM call)
+# 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}."
@@ -195,7 +186,7 @@ def tpl_article(strategy, brand, audience, objective, hashtags, max_words=800):
     return (" ".join(words[:max_words]) + "…") if len(words) > max_words else text
 
 # ----------------------------
-# Optional LLM copywriter (reuses Social Copywriter agent)
+# Optional LLM copywriter
 # ----------------------------
 def llm_copywriter(strategy_text, brand, audience, objective, tone, platform,
                    hashtags, li_words, tweet_chars, article_words):
@@ -241,42 +232,41 @@ 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:
+    # 2) Copy route
+    if use_llm:
         social = llm_copywriter(strategy, product_brand, target_audience, objective,
                                 tone, platform, hashtags,
                                 li_max_words, tweet_max_chars, article_max_words)
     else:
         if platform == "LinkedIn":
             social = tpl_linkedin(strategy, product_brand, target_audience, objective, hashtags, li_max_words)
-            fname = f"linkedin_{product_slug}.md"
+            fname = f"linkedin_{re.sub(r'\\W+','_',product_brand.lower())}.md"
         elif platform == "X (Twitter)":
             social = tpl_tweet(strategy, product_brand, target_audience, objective, hashtags, tweet_max_chars)
-            fname = f"tweet_{product_slug}.txt"
+            fname = f"tweet_{re.sub(r'\\W+','_',product_brand.lower())}.txt"
         else:  # Article
             social = tpl_article(strategy, product_brand, target_audience, objective, hashtags, article_max_words)
-            fname = f"article_{product_slug}.md"
+            fname = f"article_{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)
 
-    # Save LLM result
+    # Save LLM result with reasonable extension
     name_map = {"LinkedIn": "linkedin", "X (Twitter)": "tweet", "Article": "article"}
-    fname = f"{name_map.get(platform,'post')}_{product_slug}.md"
+    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 (two columns + header)
+# Theming & Layout (2 columns with header)
 # ----------------------------
 theme = gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")
+
 CUSTOM_CSS = """
 #header {display:flex; align-items:center; gap:16px; padding:10px 14px; border-radius:12px;
-         background: linear-gradient(90deg, #eef2ff, #f8fafc); border:1px solid #e5e7eb; margin-bottom:8px;}
+         background: linear-gradient(90deg, #eef2ff, #f8fafc); border:1px solid #e5e7eb;}
 #header img {width:44px; height:44px; object-fit:contain; border-radius:8px;}
 #header .title {font-weight:700; font-size:18px; color:#111827;}
 #header .subtitle {font-size:13px; color:#6b7280;}
@@ -290,8 +280,8 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
                 gr.Image(value=LOGO_URL, show_label=False, interactive=False, height=48, width=48)
                 gr.HTML("""
                     <div>
-                      <div class="title">Decoding Data Science — Strategy → Social Generator</div>
-                      <div class="subtitle">Run Analyst → Strategist → Creator, then produce LinkedIn / X / Article.</div>
+                      <div class="title">Decoding Data Science — Marketing Strategy → Social Generator</div>
+                      <div class="subtitle">Run the Analyst → Strategist → Creator pipeline, then produce a platform-ready post.</div>
                     </div>
                 """)
 
@@ -308,7 +298,7 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
                 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 (requires OPENAI_API_KEY)")
+                use_llm = gr.Checkbox(value=False, label="Use LLM Copywriter (higher fidelity)")
 
             with gr.Group(elem_classes="card"):
                 li_max_words = gr.Slider(100, 350, value=180, step=10, label="LinkedIn max words")
@@ -335,6 +325,9 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
         outputs=[strategy_md, social_tb, download_file]
     )
 
-# HF Spaces will start Gradio automatically; leaving default launch is fine.
+# Space-friendly launch
 if __name__ == "__main__":
-    demo.launch()
+    # 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)