Created Ideation agent

ideation_agent/agent.py

@@ -0,0 +1,32 @@
+from langgraph.graph import StateGraph, START, END
+from .utils.state import State
+from .utils.nodes import ideator , critic , improver , validator1 , validator2 , validator3 , validator4 , route1_after_validation , route2_after_validation , route3_after_validation , route4_after_validation
+
+
+def ideation_graph():
+    graph_builder= StateGraph(State)
+    graph_builder.add_node(ideator)
+    graph_builder.add_node(critic)
+    graph_builder.add_node(improver)
+    graph_builder.add_node(validator1)
+    graph_builder.add_node(validator2)
+    graph_builder.add_node(validator3)
+    graph_builder.add_node(validator4)
+
+
+    graph_builder.add_edge(START, "ideator")  # Start the graph with node_1
+    graph_builder.add_edge("ideator", "critic")
+    graph_builder.add_edge("critic", "improver")
+    graph_builder.add_edge("improver", "validator1")
+    graph_builder.add_edge("validator1", "validator2")
+    graph_builder.add_edge("validator2", "validator3")
+    graph_builder.add_edge("validator3", "validator4")
+    graph_builder.add_edge("validator4", END)
+
+    # Use conditional routing from validator
+    graph_builder.add_conditional_edges("validator1", route1_after_validation,{False:'improver',True:'validator2'})
+    graph_builder.add_conditional_edges("validator2", route2_after_validation,{False:'improver',True:'validator3'})
+    graph_builder.add_conditional_edges("validator3", route3_after_validation,{False:'improver',True:'validator4'})
+    graph_builder.add_conditional_edges("validator4", route4_after_validation,{False:'improver',True:END})
+
+    return graph_builder.compile()

ideation_agent/utils/nodes.py

@@ -0,0 +1,143 @@
+from .state import State , ValidationFormatter
+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
+from langchain_core.messages import SystemMessage , HumanMessage
+from .prompts import ideator_prompt , critic_prompt , improver_prompt , validator_prompt
+
+def ideator(state:State):
+    tools=[retrieve_tool]
+    react_agent=create_react_agent(
+    model=ideator_llm,
+    tools=tools
+    )
+
+    template = ideator_prompt(state)
+
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n''')]
+
+    response = react_agent.invoke({'messages':messages})
+    response = response['messages'][-1].content
+    print('Ideator Response:',response)
+    state.ideator_response = response
+    print('Ideator Generated the story')
+    return state
+
+def critic(state:State):
+    tools=[retrieve_tool]
+    react_agent=create_react_agent(
+      model=critic_llm,
+      tools=tools
+    )
+
+    template = critic_prompt(state)
+    messages = [SystemMessage(content=template),
+                    HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n. The business_details is\n{state.business_details}\n''')]
+
+    response = react_agent.invoke({'messages':messages})
+    response = response['messages'][-1].content
+    print('Critic Response:',response)
+
+    state.critic_response = response
+    print('Critic Evaluated the story')
+
+    return state
+
+def improver(state:State):
+    react_agent=create_react_agent(
+    model=improver_llm,
+    tools=[]
+    )
+
+    template = improver_prompt(state)
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n The business_details is:\n{state.business_details}''')]
+
+    response = react_agent.invoke({'messages':messages})
+    response = response['messages'][-1].content
+    print('Improver Response:',response)
+    state.improver_response = response
+    print('Improver Improved the story')
+    return state
+
+def validator1(state:State):
+    template = validator_prompt(state)
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n The business_details is:\n{state.business_details}''')]
+
+    response = validator_llm.with_structured_output(ValidationFormatter).invoke(messages)
+    print(f'Validator 1 response: {response}')
+    state.validator1_response = response.result
+    print('The state check:',state.validator1_response)
+    if 'not validated' in response.result:
+        state.disagreement_reason = response.reason
+    return state
+
+def validator2(state:State):
+    template = validator_prompt(state)
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n The business_details is:\n{state.business_details}''')]
+
+    response = ideator_llm.with_structured_output(ValidationFormatter).invoke(messages)
+    print(f'Validator 2 response: {response}')
+    state.validator2_response = response.result
+    print('The state check:',state.validator1_response)
+    if 'not validated' in response.result:
+        state.disagreement_reason = response.reason
+    return state
+
+def validator3(state:State):
+    template = validator_prompt(state)
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n The business_details is:\n{state.business_details}''')]
+
+    response = critic_llm.with_structured_output(ValidationFormatter).invoke(messages)
+    print(f'Validator 3 response: {response}')
+    state.validator3_response = response.result
+    print('The state check:',state.validator1_response)
+    if 'not validated' in response.result:
+        state.disagreement_reason = response.reason
+    return state
+
+def validator4(state:State):
+    template = validator_prompt(state)
+    messages = [SystemMessage(content=template),
+                HumanMessage(content=f'''The topic of the video is:\n{state.topic}\n The business_details is:\n{state.business_details}''')]
+
+    response = improver_llm.with_structured_output(ValidationFormatter).invoke(messages)
+    print(f'Validator 4 response: {response}')
+    state.validator4_response = response.result
+    print('The state check:',state.validator1_response)
+    if 'not validated' in response.result:
+        state.topic=None
+        state.disagreement_reason = response.reason
+    return state
+
+def route1_after_validation(state:State):
+    if 'not validated' in state.validator1_response:
+        return False
+    else:
+        return True
+
+
+def route2_after_validation(state:State):
+    if 'not validated' in state.validator2_response:
+        return False
+    else:
+        return True
+
+
+def route3_after_validation(state:State):
+    if 'not validated' in state.validator3_response:
+        return False
+    else:
+        return True
+
+
+def route4_after_validation(state:State):
+    if 'not validated' in state.validator4_response:
+        return False
+    else:
+        return True
+

ideation_agent/utils/prompts.py

@@ -0,0 +1,159 @@
+
+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.
+
+  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.
+
+  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.
+
+  ---
+
+  **Video Topic**:  
+  {state.topic}
+
+  **Business Details**:  
+  {state.business_details}
+
+  ---
+
+  **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**.
+
+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.
+
+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**:  
+{state.topic}
+
+**Business Details**:  
+{state.business_details}
+
+---
+
+**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.
+
+Now, generate the **core idea paragraph**.
+'''
+
+
+def critic_prompt(state):
+    return f'''
+  You are an expert **video content strategist and creative critic**.
+
+  Your job is to **analyze and critique** the following promotional video idea. Your critique should be insightful, constructive, and based on storytelling quality, creativity, uniqueness, and alignment with the provided video topic and business context.
+
+  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.
+
+  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.
+
+  ---
+
+  **Video Topic**:  
+  {state.topic}
+
+  **Business Details**:  
+  {state.business_details}
+
+  **Original Video Idea** (100 words):  
+  {state.ideator_response}
+
+  ---
+
+   **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.
+'''
+
+
+def improver_prompt(state):
+    disagreement_note = (
+    f"**Your idea was not validated  before because of this reason:\n{state.disagreement_reason}\n**"
+    if state.disagreement_reason 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.
+
+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.
+
+---
+
+**Video Topic**:  
+{state.topic}
+
+**Business Details**:  
+{state.business_details}
+
+**Critic's Response**:
+{state.critic_response}
+
+{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. 
+'''
+
+
+def validator_prompt(state):
+    return f'''
+  You are a strict and unbiased judge responsible for evaluating whether a promotional video idea is ready for publishing.
+
+  You will be given the video topic, business details, and the final version of the video idea (after improvement). Your task is to assess if this idea meets all necessary standards in terms of:
+
+  - Relevance to the video topic
+  - Alignment with the business context
+  - Tone matching the business type
+  - Creativity and clarity
+
+  You can also use the tool's response as your reference to make the validation. The tool's response includes existing influencer-style stories or content for reference.
+
+  ---
+
+
+  **Video Topic**:
+  {state.topic}
+
+  **Business Details**:
+  {state.business_details}
+
+  **Final idea from Improver**:
+  {state.improver_response}
+
+  ---
+
+  **Validation Criteria**:
+
+  - Is the idea **fully aligned** with the topic, business details and the tool's response?
+  - Does the structure flow logically and effectively?
+  - Does it feel complete and professional to use this idea to create a video?
+
+  ---
+
+  **Output Instruction**: There are two outputs you have to give.
+
+      1.Respond strictly just with one of the following values:
+    - `validated` – if the idea meets all the criteria and is validated.
+    - `not validated` – if it fails to meet any key criteria and needs further improvement.
+
+      2. Give a short reason of 30 words why the idea is validated or not validated.
+  '''

ideation_agent/utils/state.py

@@ -0,0 +1,24 @@
+from pydantic import BaseModel, Field
+from typing import Optional
+import operator
+from typing import Annotated
+
+class State(BaseModel):
+  topic: list
+  business_details: Optional[dict]
+  ideator_response: Optional[str] = None
+  critic_response: Optional[str]=None
+  improver_response: Optional[str]=None
+  validator1_response: Optional[str]=None
+  validator2_response: Optional[str]=None
+  validator3_response: Optional[str]=None
+  validator4_response: Optional[str]=None
+  disagreement_reason: Optional[str]=None
+
+
+class QueryFormatter(BaseModel):
+    video_topic: str = Field(description="The video topic that user passes to the agent")
+
+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.")

ideation_agent/utils/tools.py

@@ -0,0 +1,57 @@
+
+from langchain_core.tools import tool
+from .state import QueryFormatter
+import pandas as pd
+import numpy as np
+import ast
+import faiss
+from utils.models_loader import ST
+
+@tool("influencer's data-retrieval-tool", args_schema=QueryFormatter, return_direct=False,description="Retrieve influencer-related data for a given query.")
+def retrieve_tool(video_topic):
+    '''
+    Always invoke this tool.
+    Retrieve influencer's data by semantic search of **video topic**.
+    '''
+    # === Load CSV ===
+    csv_path = 'extracted_data.csv'
+    df = pd.read_csv(csv_path)
+
+    # === Parse stored embeddings ===
+    df['embeddings'] = df['embeddings'].apply(lambda x: ast.literal_eval(x) if isinstance(x, str) else x)
+    embeddings = np.vstack(df['embeddings'].values).astype('float32')
+
+    # === Build FAISS index ===
+    dimension = embeddings.shape[1]
+    index = faiss.IndexFlatL2(dimension)
+    index.add(embeddings)
+
+    # === Load SentenceTransformer model ===
+
+    # === Encode the query and search ===
+    query_embedding = ST.encode(str(video_topic)).reshape(1, -1).astype('float32')
+    top_k=7
+    distances, indices = index.search(query_embedding, top_k)
+
+
+
+    # === Format results ===
+    outer_list = []
+    for i, idx in enumerate(indices[0]):
+        res = {
+            'rank': i + 1,
+            'username': df.iloc[idx]['username'],
+            'story': df.iloc[idx]['story'],
+            'visible_text_or_brandings': df.iloc[idx]['story'],
+            'likesCount': df.iloc[idx]['likesCount'],
+            'commentCount': df.iloc[idx]['commentCount'],
+        }
+
+        inner_list = []
+        inner_list.append(f"[{res['rank']}]. The influencer name is: **{res['username']}** — Likes: **{res['likesCount']}**, Comments: **{res['commentCount']}**")
+        inner_list.append(f"The story of that particular video is:\n{res['story']}")
+        inner_list.append(f"The branding or promotion done is:\n{res['visible_text_or_brandings']}")
+
+        outer_list.append(inner_list)
+
+    return str(outer_list)