Update app.py

app.py

@@ -1,9 +1,14 @@
-# app.py
+
+
+
+
 from crewai import Agent, Task, Crew
 import gradio as gr
 import re
 from datetime import datetime
 from pathlib import Path
+
+
 import os
 from openai import OpenAI  # make sure `openai` is in requirements.txt
 
@@ -15,18 +20,17 @@ if not OPENAI_API_KEY:
         "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 helper --------------------------------------------------------------
+    
+# --- 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")
 
-# ---- Agents ------------------------------------------------------------------
+# ----------------------------
+# Your existing agents (unchanged)
+# ----------------------------
 lead_market_analyst = Agent(
     role="Lead Market Analyst",
     goal="Deliver sharp, data-driven market insights for the brand/product.",
@@ -54,7 +58,7 @@ creative_content_creator = Agent(
     verbose=True,
 )
 
-# Optional social copywriter (only used 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.",
@@ -63,7 +67,9 @@ social_copywriter = Agent(
     verbose=False,
 )
 
-# ---- Tasks & Crew -------------------------------------------------------------
+# ----------------------------
+# Core crew (unchanged)
+# ----------------------------
 def run_marketing_crew(product_brand: str, target_audience: str, objective: str) -> str:
     topic = f"{product_brand} | Audience: {target_audience} | Objective: {objective}"
 
@@ -115,15 +121,16 @@ 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 = []
     for l in lines:
         if l.startswith(("-", "*", "•")) or re.match(r"^\d+[\.\)]", l) or len(l) > 50:
             bullets.append(l.lstrip("-*• ").strip())
-        if len(bullets) >= n:
-            break
+        if len(bullets) >= n: break
     if not bullets:
         parts = [p.strip() for p in text.split("\n\n") if p.strip()]
         bullets = parts[:n]
@@ -136,11 +143,9 @@ 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] + "…"
 
-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) --------------------------------
+# ----------------------------
+# 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)])
@@ -191,7 +196,9 @@ 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 (uses the Social Copywriter agent) ---------------
+# ----------------------------
+# Optional LLM copywriter (reuses Social Copywriter agent)
+# ----------------------------
 def llm_copywriter(strategy_text, brand, audience, objective, tone, platform,
                    hashtags, li_words, tweet_chars, article_words):
     limits = {
@@ -221,7 +228,9 @@ 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,
@@ -239,31 +248,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_{safe}.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_{safe}.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_{safe}.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)
 
-# ---- Theming & Layout (2 columns with header) --------------------------------
+    # 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)
+# ----------------------------
 theme = gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")
 
 CUSTOM_CSS = """
@@ -297,10 +306,8 @@ 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)")
 
@@ -329,8 +336,6 @@ with gr.Blocks(title="DDS Marketing Crew → Social Content", theme=theme, css=C
         outputs=[strategy_md, social_tb, download_file]
     )
 
-# ---- Space-friendly launch ----------------------------------------------------
-if __name__ == "__main__":
-    host = os.getenv("HOST", "0.0.0.0")
-    port = int(os.getenv("PORT", "7860"))
-    demo.launch(server_name=host, server_port=port)
+# In Colab, share=True is handy
+demo.launch(share=True)
+