update

app.py

@@ -41,7 +41,7 @@ except Exception as e:
 @tool
 def generate_quote_tool(niche: str, style: str) -> str:
     """
-    Generate a unique inspirational quote using the HybridQuoteGenerator.
+    Generate a unique, short inspirational quote using the HybridQuoteGenerator.
 
     Args:
         niche: The category of the quote (e.g. Motivation, Fitness, Mindfulness).
@@ -81,7 +81,7 @@ def search_pexels_video_tool(style: str, niche: str) -> dict:
 
     Args:
         style: Visual style (e.g. Cinematic, Nature, Urban, Minimal, Abstract).
-        niche: Content niche (e.g. Motivation, Business/Entrepreneurship, Fitness).
+        niche: Content niche or free-text theme (e.g. Motivation, Soft Life Mornings).
 
     Returns:
         A dictionary with:
@@ -91,7 +91,8 @@ def search_pexels_video_tool(style: str, niche: str) -> dict:
             - pexels_url: The Pexels page URL (or None).
             - error: Optional error message on failure.
     """
-    search_strategies = {
+    # ---- Canonical mapping (your original curated niches) ----
+    canonical_strategies = {
         "Motivation": {
             "Cinematic": ["person climbing mountain", "running sunrise", "achievement success"],
             "Nature": ["sunrise mountain peak", "ocean waves powerful", "forest light"],
@@ -143,7 +144,29 @@ def search_pexels_video_tool(style: str, niche: str) -> dict:
         },
     }
 
-    queries = search_strategies.get(niche, {}).get(style, ["aesthetic nature"])
+    # Normalize style so we don't crash on unexpected input
+    normalized_style = style or "Cinematic"
+    normalized_style = normalized_style.strip().title()
+    if normalized_style not in ["Cinematic", "Nature", "Urban", "Minimal", "Abstract"]:
+        normalized_style = "Cinematic"
+
+    # Decide queries:
+    # 1) If niche is one of the canonical categories -> use curated mapping
+    # 2) Otherwise, treat niche as free-text theme and build dynamic queries
+    if niche in canonical_strategies:
+        queries = canonical_strategies[niche].get(normalized_style, ["aesthetic nature"])
+    else:
+        base = (niche or "").lower().strip()
+        style_word = normalized_style.lower()
+        if not base:
+            base = "aesthetic vertical"
+
+        queries = [
+            f"{base} {style_word} aesthetic vertical video",
+            f"{base} b-roll {style_word}",
+            f"{base} lifestyle aesthetic",
+            f"{base} cinematic b-roll",
+        ]
 
     try:
         if not PEXELS_API_KEY:
@@ -156,8 +179,8 @@ def search_pexels_video_tool(style: str, niche: str) -> dict:
             }
 
         headers = {"Authorization": PEXELS_API_KEY}
-        query = random.choice(queries)
 
+        query = random.choice(queries)
         url = (
             f"https://api.pexels.com/videos/search"
             f"?query={query}&per_page=15&orientation=portrait"
@@ -245,6 +268,7 @@ def create_quote_video_tool(
 
     try:
         print(f"🚀 Processing on Modal (fast!) with text_style={text_style}...")
+        # Note: modal_video_processing.py currently ignores text_style; it's safe to send it
         response = requests.post(
             modal_endpoint,
             json={
@@ -298,7 +322,8 @@ def create_quote_video_tool(
 @tool
 def get_trending_topics_tool(platform: str, niche: str, max_topics: int = 6) -> dict:
     """
-    Suggest trending content angles and hooks for short-form quote videos.
+    Suggest trending content angles and hooks for short-form quote videos,
+    tightly aligned with the given niche.
 
     Args:
         platform: Main platform (e.g. 'TikTok', 'Instagram Reels', 'YouTube Shorts', or 'General').
@@ -320,10 +345,20 @@ def get_trending_topics_tool(platform: str, niche: str, max_topics: int = 6) ->
 You are helping a creator make short quote videos.
 
 Platform: {platform}
-Niche: {niche}
+Creator niche: {niche}
+
+All suggested topics MUST be tightly aligned with this niche.
+
+Examples of alignment:
+- If niche = Mindfulness → topics about calm lifestyle, awareness, mental clarity, soft-life energy, nervous system regulation.
+- If niche = Motivation → topics about discipline era, glow up routines, self-improvement, consistency, habit stacking.
+- If niche = Love & Relationships → topics about healing, attachment, secure love, boundaries, heartbreak recovery.
+- If niche = Stoicism → topics about emotional resilience, fate, control vs. acceptance, inner strength.
+- If niche = Leadership → topics about leading teams, decision-making, emotional intelligence at work.
+- If niche = Business/Entrepreneurship → topics about founders, failure, shipping, building in public, risk.
 
 Task:
-- Propose {max_topics} SPECIFIC trending-style topics/angles that would perform well on {platform}.
+- Propose {max_topics} SPECIFIC trending-style topics/angles that would perform well on {platform} for THIS niche.
 - Think in terms of what actually shows up on TikTok/Reels/Shorts: concrete micro-themes and patterns, not vague ideas.
 
 For each topic, include:
@@ -332,7 +367,7 @@ For each topic, include:
 
 Constraints:
 - No generic motivational fluff like "follow your dreams" or "believe in yourself".
-- Make it feel current: you can use language like "soft life", "discipline era", "glow up", "reset routine" where it fits.
+- Make it feel current: you can use language like "soft life", "discipline era", "glow up", "reset routine", etc., only if it fits the niche.
 - Keep everything under 20 words per topic name and per hook.
 
 Return a JSON object with EXACTLY this structure:
@@ -446,10 +481,10 @@ def mcp_agent_pipeline(
     MAIN PIPELINE: uses smolagents CodeAgent.run to plan & call tools.
 
     The agent:
-      - calls generate_quote_tool
-      - calls search_pexels_video_tool multiple times
+      - calls get_trending_topics_tool to pick a trend
+      - calls generate_quote_tool with style influenced by that trend
+      - calls search_pexels_video_tool using the TRENDING TOPIC as niche (free-text support)
       - calls create_quote_video_tool for each variation
-      - returns JSON with status_log + video_paths
     """
     base_log = ["🤖 **MCP AGENT RUN**"]
 
@@ -469,66 +504,100 @@ def mcp_agent_pipeline(
         base_prefix = os.path.join(output_dir, f"agent_{timestamp}_v")
 
         user_task = f"""
-You are an autonomous Python agent helping creators generate short vertical quote videos.
+You are an autonomous Python agent helping a creator generate short vertical quote videos that follow TikTok-style trends.
+
+You MUST follow these steps exactly, using the provided TOOLS:
 
-Niche: {niche}
-Visual Style: {style}
-Text style for quotes: {text_style}
-Number of variations: {num_variations}
+AVAILABLE TOOLS (with signatures):
 
-You have these TOOLS available:
+1. get_trending_topics_tool(platform: str, niche: str, max_topics: int) -> dict
+   - Returns a dict with:
+     - "topics": list of {{"topic": str, "hook": str}}
+     - "notes": str
 
-1. generate_quote_tool(niche: str, style: str) -> str
-   - Returns a single SHORT quote as plain text.
+2. generate_quote_tool(niche: str, style: str) -> str
+   - Returns a single short quote as plain text.
 
-2. search_pexels_video_tool(style: str, niche: str) -> dict
+3. search_pexels_video_tool(style: str, niche: str) -> dict
    - Returns a dict with:
      - "success": bool
      - "video_url": str or None
 
-3. create_quote_video_tool(
+4. create_quote_video_tool(
        video_url: str,
        quote_text: str,
        output_path: str,
        text_style: str = "classic_center"
    ) -> dict
-   - Writes a video file to output_path and returns a dict with:
+   - Writes a video file to output_path and returns:
      - "success": bool
      - "output_path": str or None
 
-You MAY also optionally call:
-4. get_trending_topics_tool(platform: str, niche: str, max_topics: int = 6) -> dict
-   - Use platform="TikTok" and the given niche to get trending angles.
-   - If you call it, you can use the first topic/hook as inspiration for the quote content.
-
-Your job:
-
-1. (Optional) Call get_trending_topics_tool("TikTok", "{niche}", 4) to see trending angles.
-   - If you do, mention in the status_log which topic you used as inspiration.
-2. Call generate_quote_tool once to obtain quote_text (optionally influenced by a chosen topic).
-3. For each variation i from 1 to {num_variations}:
-   - Call search_pexels_video_tool(style, niche).
-   - If it succeeds, compute output_path exactly as:
-     "{base_prefix}" + str(i) + ".mp4"
-   - Call create_quote_video_tool(video_url, quote_text, output_path, text_style="{text_style}").
-4. Collect only variations where create_quote_video_tool returns success == True and a non-empty output_path.
-5. Build a human-readable status_log string summarizing:
-   - Which tools you called (including whether you used get_trending_topics_tool)
-   - How many videos succeeded or failed
-6. Return ONLY a valid JSON object of the form:
+RUNTIME PARAMETERS:
+
+- base_niche = "{niche}"
+- base_style = "{style}"
+- text_style = "{text_style}"
+- num_variations = {num_variations}
+- base_output_prefix = "{base_prefix}"
+
+STEP-BY-STEP PLAN (YOU MUST FOLLOW):
+
+1. Call get_trending_topics_tool with:
+   - platform = "TikTok"
+   - niche = base_niche
+   - max_topics = 5
+
+   From the returned dict:
+   - If "topics" is non-empty:
+       trending_topic = topics[0]["topic"]
+       trending_hook  = topics[0]["hook"]
+     Otherwise:
+       trending_topic = base_niche
+       trending_hook  = ""
+
+2. Call generate_quote_tool with:
+   - niche = base_niche
+   - style = base_style + " — trend: " + trending_topic
+
+   Store the result as:
+   - quote_text
+
+3. For i from 1 to num_variations (inclusive):
+   a) Call search_pexels_video_tool with:
+        style = base_style
+        niche = trending_topic
+      Let this result be search_result.
+   b) If search_result["success"] is True AND search_result["video_url"] is a non-empty string:
+        - Set:
+            output_path = base_output_prefix + str(i) + ".mp4"
+        - Call create_quote_video_tool with:
+            video_url   = search_result["video_url"]
+            quote_text  = quote_text
+            output_path = output_path
+            text_style  = text_style
+          Let this be video_result.
+        - If video_result["success"] is True AND video_result["output_path"] is a non-empty string:
+            append video_result["output_path"] to a Python list called final_video_paths.
+      Otherwise, skip this variation.
+
+4. Build a multi-line, human-readable string called status_log summarizing:
+   - base_niche, base_style, text_style
+   - The chosen trending_topic and trending_hook (or that you fell back to base_niche)
+   - The final quote_text
+   - How many variations succeeded vs how many were attempted
+
+5. Return ONLY the following JSON object as your final answer (no markdown, no backticks):
 
 {{
-  "status_log": "multi-line human readable description of what you did",
-  "video_paths": [
-    "{base_prefix}1.mp4",
-    "... only paths that actually succeeded ..."
-  ]
+  "status_log": "your multi-line human readable log here",
+  "video_paths": ["... list of successful output_path strings from final_video_paths ..."]
 }}
 
 CRITICAL:
-- Do NOT wrap the JSON in markdown or backticks.
-- Do NOT add extra keys.
-- Do NOT print anything except the JSON object as your final answer.
+- Do NOT include any other keys.
+- Do NOT return Python objects; everything must be JSON-serializable.
+- Do NOT print anything else beyond that single JSON object.
 """
 
         agent_result = agent.run(user_task)
@@ -670,17 +739,17 @@ with gr.Blocks(
 
     **Key Features:**
     - 🌟 Short, non-repeating Gemini quotes (per niche history)
+    - 🔥 Trending topics actually drive the quote + visuals
     - 🤖 smolagents CodeAgent for tool planning
     - 🔗 Optional MCP client integration
     - 🎥 Modal for fast video rendering
     - 🅰️ Text style controls (font & placement)
-    - 🔥 Trending topics / angle helper
     """
     )
 
     with gr.Accordion("📸 Example Gallery - Recent Videos", open=True):
         gr.Markdown(
-            "See what others (or you) have generated. Auto-updates after each run."
+            "See what has been generated. Auto-updates after each run."
         )
 
         with gr.Row():
@@ -828,11 +897,11 @@ with gr.Blocks(
     ---
     ### ✨ Features
     - 🌟 Gemini-powered, short non-repeating quotes (per niche)
+    - 🔥 Trend-aware: topics influence quotes + videos
     - 🎨 Multiple aesthetic video & text layouts
     - ⚡ Modal-accelerated rendering
     - 🤖 smolagents CodeAgent for autonomous tool-calling
     - 🔗 Optional MCP integration via MCPClient
-    - 🔥 Trend helper to shape topics & hooks
 
     ### 🏆 Hackathon: MCP 1st Birthday
     **Track:** Track 2 - MCP in Action