Refined ideation prompts

.gitignore

@@ -9,4 +9,17 @@ my_agent/utils/check.py
 streamlit_app.py
 data
 extracted_data.csv
-Research
+Research
+__pycache__/
+*.pyc
+
+# __pycache__
+# context_analysis_agent/__pycache__
+# context_analysis_agent/utils/__pycache__
+# brainstroming_agent/__pycache__
+# brainstroming_agent/utils__pycache__
+# business_interaction_agent/__pycache__
+# business_interaction_agent/utils/__pycache__
+# ideation_agent/__pycache__
+# ideation_agent/utils/__pycache__
+

ideation_agent/utils/nodes.py

@@ -1,4 +1,4 @@
-from .state import State , ValidationFormatter
+from .state import State , ValidationFormatter , ImproverResponseFormatter
 from .tools import retrieve_tool
 from langgraph.prebuilt import create_react_agent
 from utils.models_loader import ideator_llm, critic_llm , improver_llm , validator_llm
@@ -6,6 +6,7 @@ from langchain_core.messages import SystemMessage , HumanMessage
 from .prompts import ideator_prompt , critic_prompt , improver_prompt , validator_prompt
 
 
+
 ideator_agent = create_react_agent(
     model=ideator_llm,
     tools=[retrieve_tool]
@@ -66,20 +67,25 @@ def improver(state:State):
     messages = [SystemMessage(content=template),
                 HumanMessage(content=f'''The topic of the video is:\n{state.topic[-1]}\n The business_details is:\n{state.business_details[-1]}''')]
     print('improver prompt:', messages)
+    response = improver_llm.with_structured_output(ImproverResponseFormatter).invoke(messages)
+    state.improver_response.append(response.improved_idea)
+    state.critic_fault.append(response.faults)
+    print('Improver response:',response.improved_idea)
+    return state
     
-    try:
-        response = improver_agent.invoke({'messages':messages})
-        response = response['messages'][-1].content
-        print('Improver Response:',response)
-        state.improver_response.append(response)
-        print('Improver Improved the story')
-        return state
-
-    except:
-        response = improver_llm.invoke(messages)
-        print('Improver backup Response:',response.content)
-        state.improver_response.append(response.content)
-        return state
+    # try:
+    #     response = improver_agent.invoke({'messages':messages})
+    #     response = response['messages'][-1].content
+    #     print('Improver Response:',response)
+    #     state.improver_response.append(response)
+    #     print('Improver Improved the story')
+    #     return state
+
+    # except:
+    #     response = improver_llm.invoke(messages)
+    #     print('Improver backup Response:',response.content)
+    #     state.improver_response.append(response.content)
+    #     return state
 
 def validator1(state:State):
     template = validator_prompt(state)

ideation_agent/utils/prompts.py

@@ -1,37 +1,40 @@
+example = '''
+**To better understand the expected tone, depth, and style, here are two sample ideas with their exact business contexts**:
 
-def ideator_prompt_old(state):
-    return f'''
+---
 
-  You are a highly creative video content strategist who specializes in generating original, compelling video ideas in exactly 100 words.
+**Example 1: Restaurant Business**  
+**Business Details**:  
+"business_type": "restaurant", "platform": "instagram", "target_audience": "youths", "business_goals": "to go global", "offerings": "nepali foods", "Challenges_faced": "finding new customers, attracting large customers"
 
-  Your task is to develop a **unique, creative and imaginative root idea** for a promotional video based on the following **video topic** and **business details**. The idea should be highly creative, catchy, and distinct—something that instantly grabs attention.
+**Core Idea (100 words)**:  
+The influencer challenges the chef: “Can your food impress people from 5 different countries?” The restaurant serves the same Nepali dish (like momo or gundruk soup) to five diverse international students/youths without telling them it’s Nepali. Each gives raw reactions and ratings. Once they fall in love with it, the reveal hits. Final line: “If the world loves it—why isn’t it global yet?” This idea plays on suspense, universal taste appeal, and social proof—turning traditional Nepali food into an underdog hero.
 
-  Another important thing is that you have to give the response focusing on the response of the tool provided to you. The tool contains the video stories and contents created by the influencers.  
-  Use those responses for your reference. You can use your creativity but stay within the boundaries of the tool's response.
+---
 
-  ---
+**Example 2: Clothing Business**  
+**Business Details**:  
+"business_type": "clothing business", "platform": "instagram", "target_audience": "youths", "business_goals": "to go global", "offerings": "nepali clothing products", "Challenges_faced": "finding new customers, attracting large customers"
 
-  **Video Topic**:  
-  {state.topic[-1]}
+**Core Idea (100 words)**:  
+The influencer opens with: “What if your drip had 1,000 years of history behind it?” They showcase traditional Nepali garments reimagined with modern streetwear styling—hoodies with Dhaka patterns, joggers inspired by daura suruwal. Youths on the street try them on and react: “I’d totally wear this!” The story ends with: “Nepali fashion isn’t old—it’s timeless. And now, it’s global.” A core idea blending heritage with youth-driven reinvention, sparking both pride and curiosity.
 
-  **Business Details**:  
-  {state.business_details[-1]}
+---
+'''
 
-  ---
 
-  **Important Instructions**:
-  Do **not** generate a complete storyline. Focus only on the **main root idea** that is clear, unique,  bold, and creatively promising. Keep it exactly 100 words.
-  Now, generate the final idea for the video topic.
-'''
 
 def ideator_prompt(state):
     return f'''
-You are a highly imaginative and strategic **video ideator** who specializes in crafting powerful promotional video **core ideas** in **exactly 100 words**.
+You are a highly imaginative and strategic **video ideator** with a reputation for crafting unforgettable promotional video **core ideas** that go viral — all in **exactly 100 words**.
 
-Your task is to develop a **single, bold, unique, and creative paragraph** that represents the **root idea** of a promotional video. This is **not** a full concept or storyline — just the **main idea**, presented creatively and concisely.
+Your task is to create a **single, bold, deeply creative paragraph** that captures the **central narrative engine** of a promotional video. Think of it as the **beating heart** of a blockbuster: one core idea that flows smoothly, emotionally, and memorably. It should resonate instantly with the business’s target audience, platform, and goals — while staying tight, rooted, and crystal clear.
+
+Do not write scattered ideas or concepts that jump in multiple directions. Focus on **one clear concept** that could inspire an entire successful video campaign. This is not a title, slogan, or script — it’s the foundational idea the rest of the content could build on.
 
 Use the response of the provided tool (which includes video stories from influencers) as inspiration. Stay within the creative boundaries of that tool's content, but feel free to push the edges with originality.
 
+
 ---
 
 **Video Topic**:  
@@ -43,15 +46,42 @@ Use the response of the provided tool (which includes video stories from influen
 ---
 
 **Important Instructions**:
-- Write only **one paragraph**, not a list.
-- Do **not** include titles, headers, sections, or formatting like bullet points.
-- The output must be exactly **100 words** — no more, no less.
-- Focus on uniqueness, imagination, and viral potential.
+- Output must be **one paragraph** only — not a list or outline.
+- Do **not** use headers, titles, sections, or formatting like bullet points.
+- The idea must be **exactly 100 words** — no more, no less.
+- Prioritize originality, emotional engagement, and shareability.
+- Keep it firmly grounded in the business details provided — product, audience, platform, challenge, and goal.
+- Imagine this idea being executed by a top influencer with millions of views.
+
+---
 
-Now, generate the **core idea paragraph**.
+**To better understand the expected tone, depth, and style, here are two sample ideas with their exact business contexts**:
+
+---
+
+**Example 1: Restaurant Business**  
+**Business Details**:  
+"business_type": "restaurant", "platform": "instagram", "target_audience": "youths", "business_goals": "to go global", "offerings": "nepali foods", "Challenges_faced": "finding new customers, attracting large customers"
+
+**Core Idea (100 words)**:  
+The influencer challenges the chef: “Can your food impress people from 5 different countries?” The restaurant serves the same Nepali dish (like momo or gundruk soup) to five diverse international students/youths without telling them it’s Nepali. Each gives raw reactions and ratings. Once they fall in love with it, the reveal hits. Final line: “If the world loves it—why isn’t it global yet?” This idea plays on suspense, universal taste appeal, and social proof—turning traditional Nepali food into an underdog hero.
+
+---
+
+**Example 2: Clothing Business**  
+**Business Details**:  
+"business_type": "clothing business", "platform": "instagram", "target_audience": "youths", "business_goals": "to go global", "offerings": "nepali clothing products", "Challenges_faced": "finding new customers, attracting large customers"
+
+**Core Idea (100 words)**:  
+The influencer opens with: “What if your drip had 1,000 years of history behind it?” They showcase traditional Nepali garments reimagined with modern streetwear styling—hoodies with Dhaka patterns, joggers inspired by daura suruwal. Youths on the street try them on and react: “I’d totally wear this!” The story ends with: “Nepali fashion isn’t old—it’s timeless. And now, it’s global.” A core idea blending heritage with youth-driven reinvention, sparking both pride and curiosity.
+
+---
+
+Now, generate your **core idea paragraph**.
 '''
 
 
+
 def critic_prompt(state):
     return f'''
   You are an expert **video content strategist and creative critic**.
@@ -60,6 +90,8 @@ def critic_prompt(state):
 
   Also, you have to check whether the given idea is relevant to the tool's response or not. The tool's response includes the video stories of the influencers. If it's not relevant to the tool's response, you have to specify it.
 
+  Also, you have to check whether the given idea is scattered in several dimensions or not. If it's scattered or bit vague, you have to mainrtain it at the  root idea flowing smoothly in a core concept.
+
   After identifying the weaknesses or faults in the idea, you must **rewrite a new, improved version** of the core idea that fixes those flaws or faults while keeping the concept bold, creative, and engaging. The improved idea should also be exactly 100 words.
 
   ---
@@ -78,20 +110,26 @@ def critic_prompt(state):
    **Your Task as the Critic**:
   1. List the **key faults** or weak areas in the original idea.
   2. Provide a **revised version** of the idea that improves upon the original, maintains high creativity, and stays within 100 words.
+
+  {example}
 '''
 
 
 def improver_prompt(state):
     disagreement_note = (
-    f"**Your idea was not validated  before because of this reason:\n{state.disagreement_reason[-1]}\n**"
-    if len(state.disagreement_reason)>0 else ""
+        f"**Note:** This idea was not validated earlier because:\n{state.disagreement_reason[-1]}\n"
+        if len(state.disagreement_reason) > 0 else ""
     )
+    
     return f'''
 You are a master-level **video concept developer** known for turning creative ideas into standout, refined gems.
 
-Your task is to carefully evaluate the **critic's response** to a promotional video idea. The critic has already listed faults in the original idea and provided a revised version. You must now go a step further and generate an **even more creative, emotionally engaging, and original 100-word idea** that corrects any lingering weaknesses or missed potential in the critic’s revision.
+Your task is to **evaluate the critic’s response** to a promotional video idea. The critic has already listed faults and provided a revised version. Now, you must go a step further by:
 
-Focus on making the idea sharper, bolder, and more impactful — while staying aligned with the video topic, business details, and the style of influencer content referenced in the tool’s response.
+1. **Listing the key faults or missed opportunities** in the critic’s revised idea (if any), such as lack of emotional engagement, weak narrative flow, unoriginality, poor alignment with brand/influencer style, etc.
+2. **Generating a final, improved version** of the idea that is more creative, emotionally resonant, and narratively strong — while still aligned with the business context and influencer content style.
+
+If the critic’s idea is already strong and has no faults, simply **repeat it** as your final version.
 
 ---
 
@@ -101,16 +139,18 @@ Focus on making the idea sharper, bolder, and more impactful — while staying a
 **Business Details**:  
 {state.business_details[-1]}
 
-**Critic's Response**:
+**Critic's Response**:  
 {state.critic_response[-1]}
 
 {disagreement_note}
 
 ---
 
-**Your Task**:  
-Generate a **new, final improved idea** based on the critic’s analysis and revision. Your idea must be exactly **100 words**, stand out with high creativity, and reflect a strong understanding of narrative appeal and brand relevance.
-If the critic's revision doesn't have any problem, give the same critic's idea. 
+**Your Tasks**:  
+**A. List faults (if any) in the critic's revised idea.**  
+**B. Write a final improved idea (100 words exactly). If no faults found, repeat the critic’s version.**
+
+{example}
 '''
 
 

ideation_agent/utils/state.py

@@ -8,6 +8,7 @@ class State(BaseModel):
   business_details : Annotated[list[dict],operator.add] = []
   ideator_response: Annotated[list[str],operator.add] = []
   critic_response: Annotated[list[str],operator.add] = []
+  critic_fault: Annotated[list[str],operator.add] = []
   improver_response: Annotated[list[str],operator.add] = []
   validator1_response: Annotated[list[str],operator.add] = []
   validator2_response: Annotated[list[str],operator.add] = []
@@ -19,6 +20,10 @@ class State(BaseModel):
 class QueryFormatter(BaseModel):
     video_topic: str = Field(description="The video topic that user passes to the agent")
 
+class ImproverResponseFormatter(BaseModel):
+   faults: str= Field(description="Returns the faults of the critic's response if any.")
+   improved_idea: str = Field(description="Returns the improved idea of exactly 100 words. If no faults found, repeat the critic's version.")
+
 class ValidationFormatter(BaseModel):
   result: str = Field(description="Returns **validated** if the story is validated. Returns **not validated** if story is not validated.")
   reason: str = Field(description="Returns the reason why the story is validated or not validated.")

main.py

@@ -24,7 +24,7 @@ brainstrom_graph = brainstroming_graph()
 
 stored_data={}
 stored_data['business_details']={"business_type": "restaurant", "platform": "instagram", "target_audience": "youths", "business_goals": "to go global", "offerings": "nepali foods", "Challenges_faced": "finding new customers, attracting large customers"}
-# stored_data['business_details']={}
+stored_data['final_ideation'] = {}
 class UserMessage(BaseModel):
     message: str
 @app.post("/context-analysis")
@@ -65,6 +65,7 @@ def ideation_endpoint(request:IdeationRequest):
         },
         config=config,
         )
+        stored_data['final_ideation'] = result['improver_response'][-1]
         return {'response':result}
     except GraphRecursionError:
         result = idea_graph.get_state({"configurable": {"thread_id": "ideation_thread123"}})
@@ -74,18 +75,16 @@ def ideation_endpoint(request:IdeationRequest):
 
 
 class BrainstormRequest(BaseModel):
-    query: List[str]
-    preferred_topics: Optional[List] = []
-    image_base64_list: Optional[List] = []
+    preferred_topics: Optional[list] = []
+    image_base64_list: Optional[list] = []
     thread_id: Optional[str]="default-session"
 
 @app.post("/brainstrom")
 def brainstroming_endpoint(
     request: BrainstormRequest,  # 🔥 Full JSON body here
-    # thread_id: Optional[str] = Query("default-session"),  # Separate query param
 ):
     result = brainstrom_graph.invoke({
-        'topic': request.query,
+        'topic': [stored_data['final_ideation']],
         'images': request.image_base64_list,
         'latest_preferred_topics': request.preferred_topics,
         'business_details': (lambda d: d['business_details'] if 'business_details' in d else {})(stored_data)

utils/models_loader.py

@@ -14,19 +14,19 @@ login(os.environ['HUGGINGFACEHUB_ACCESS_TOKEN'])
 os.environ['GROQ_API_KEY']=os.getenv('GROQ_API_KEY')
 
 
-# llm = ChatGroq(
-#     model="llama3-8b-8192",
-#     temperature=0.3,
-#     max_tokens=500,
-
-# )
-
-llm = ChatOpenAI(
-    model="gpt-4o-mini",
+llm = ChatGroq(
+    model="llama3-8b-8192",
     temperature=0.3,
     max_tokens=500,
+
 )
 
+# llm = ChatOpenAI(
+#     model="gpt-4o-mini",
+#     temperature=0.3,
+#     max_tokens=500,
+# )
+
 
 class HFEmbeddingAPI:
     def __init__(self, api_url, token):