Updated commit

business_interaction_agent/agent.py

@@ -5,7 +5,7 @@ from langgraph.prebuilt import create_react_agent
 from .utils.state import State,StateUpdateFormatter
 # from .utils.nodes import business_interaction_node, cleanup_messages
 from utils.models_loader import llm
-from langchain_core.messages import SystemMessage
+from langchain_core.messages import SystemMessage, ToolMessage
 from .utils.prompts import business_retrieval_prompt, check_state_update_prompt
 from .utils.utils import manual_retrieval
 from context_analysis_agent.utils.utils import save_to_db
@@ -42,8 +42,8 @@ class BusinessInteractionChatbot:
     def _call_model(self, state):
         print('Entered into callmodel')
         retrievals = manual_retrieval(str([msg['content'] for msg in self.messages if msg['role'] == 'user']),business_state.business_details)
-        template = business_retrieval_prompt(str([msg['content'] for msg in self.messages if msg['role'] == 'user']),retrievals,str(business_state.business_details))
-        messages = [SystemMessage(content=template)] + state["messages"]
+        template = business_retrieval_prompt(str([msg['content'] for msg in self.messages if msg['role'] == 'user']),str(business_state.business_details))
+        messages = [SystemMessage(content=template),ToolMessage(content="Tool's response:\n"+retrievals,tool_call_id='call_business_interaction')] + state["messages"]
         print('The message is:',messages)
         backup_response = self.react_agent.invoke({'messages':messages})['messages'][-1]
         print('Backup response:',backup_response.content)
@@ -57,7 +57,7 @@ class BusinessInteractionChatbot:
         messages = [SystemMessage(content=template)]
         response = llm.with_structured_output(StateUpdateFormatter).invoke(messages)
         # response= llm.invoke(messages)
-        print('Response of state check:',response)
+        # print('Response of state check:',response)
         return response.model_dump()
 
 
@@ -66,6 +66,8 @@ class BusinessInteractionChatbot:
         business_state.business_details=business_details
         self.messages.append({"role": "user", "content": f'{user_input}'})
         checked_details = self.check_state_update()
+        print('Checked details:',checked_details)
+        print('Business details:',business_state.business_details)
         if checked_details!= business_state.business_details:
             save_to_db(checked_details)
             print('Database Updated as the state changed....')

business_interaction_agent/utils/nodes.py

@@ -1,4 +1,4 @@
-from langchain_core.messages import SystemMessage
+from langchain_core.messages import SystemMessage, ToolMessage
 from .prompts import business_retrieval_prompt
 from .utils import manual_retrieval
 
@@ -8,7 +8,7 @@ def business_interaction_node(state, llm, react_agent, messages, business_detail
     user_inputs = str([msg['content'] for msg in messages if msg['role'] == 'user'])
     retrievals = manual_retrieval(user_inputs, business_details)
     template = business_retrieval_prompt(user_inputs, retrievals)
-    formatted_messages = [SystemMessage(content=template)] + state["messages"]
+    formatted_messages = [SystemMessage(content=template) , ToolMessage(content="Tool's response:\n"+retrievals)] + state["messages"]
     response = react_agent.invoke({'messages': formatted_messages})['messages'][-1]
     print('Backup response:', response.content)
     return {"messages": [response.content]}

business_interaction_agent/utils/prompts.py

@@ -1,10 +1,10 @@
-def business_retrieval_prompt(user_message, retrievals, business_details):
+def business_retrieval_prompt(user_message, business_details):
     return f'''
-You are a professional AI assistant helping users understand how influencers can support their business. You will be given:
+You are a professional AI assistant helping users understand how influencers can support their business but only on the basis of the tool's response provided to you. You will be given:
 
 - A **user message**
-- A list of **retrieved influencer data**
-- The **business details** provided by the user
+- The response of the tool. The tool's response contains the video stories of the influencers.
+- The **business details** provided by the user.
 
 Your job is to:
 1. **First**, analyze the user message and decide if it is actually a business-related question or query that could be answered using influencer content.
@@ -14,7 +14,7 @@ Your job is to:
     a. Analyze the **business details** and check if the user message aligns with the type of business.
         - If there is a mismatch (e.g., business is a restaurant but the user asks about clothing), politely **alert the user** about the mismatch.
         - Still, go ahead and provide a helpful answer using relevant influencer data if available.
-    b. Analyze the influencer data and explain how each influencer might support the business based only on the retrieved content.
+    b. Analyze the influencer data and explain how each influencer might support the business based only on the tool's response.
 
 --- USER MESSAGE ---
 {user_message}
@@ -22,29 +22,26 @@ Your job is to:
 --- BUSINESS DETAILS ---
 {business_details}
 
---- START OF RETRIEVALS ---
-{retrievals}
---- END OF RETRIEVALS ---
 
 Rules:
-- If the user message is **not relevant to influencer or business help**, politely respond in a general helpful way and ignore the retrievals and business details.
+- If the user message is **not relevant to influencer or business help**, politely respond in a general helpful way and ignore the tool's response and business details.
 - If the message **is relevant**, then:
   - First verify if the user’s business type matches the context of their message.
   - If not, display a short alert to the user like: “Note: Your query seems to focus on [X], but your business is about [Y].”
   - Then analyze the influencer stories:
     - Identify which influencer content is relevant.
     - Explain what they are promoting and how it might help the business.
-    - You **must mention influencer usernames** and only use what is in the retrievals.
-- Do NOT invent or assume any information beyond what is explicitly provided in the retrievals and business details.
+    - You **must mention influencer usernames** and only use what is in the tool's response.
+- Do NOT invent or assume any information beyond what is explicitly provided in the tool's response and business details.
 
 Keep your response:
 - Context-aware, direct and short as possible
-- Grounded only in the given data
+- Grounded only in the tool's response.
 - Helpful, concise, and user-friendly
 '''.strip()
 
 
-def check_state_update_prompt_old(business_details: dict, user_messages: str):
+def check_state_update_prompt(business_details: dict, user_messages: str):
     return f"""
 You are an expert business assistant tasked with updating business details by thoroughly analyzing user messages. Your goal is to identify and apply changes to the business details only when the user message explicitly provides new or conflicting information relevant to the business details. Follow these guidelines:
 
@@ -72,23 +69,7 @@ Output:
 Return the updated business details in the same dictionary format as provided. If no changes are warranted, return the original business details unchanged.
 """
 
-def check_state_update_prompt(business_details: dict, user_messages: str):
-    return f"""
-You are a business assistant. Your job is to check if the user's message changes any of the current business details.
-
-Rules:
-1. Only change details if the user gives **clear new or different information**.
-2. If the message does **not change anything**, return the same business details.
-3. **Never return empty values.** If you can't update something, keep the current value.
-4. Keep the **same format** as the original business details (same keys and data types).
-
-Input:
-Current business details: {str(business_details)}
-User message: {user_messages}
 
-Output:
-Return the updated business details in the same dictionary format. If nothing changed, return the original business details.
-""".strip()
 
 
 

idea_to_budget_agent/agent.py

@@ -0,0 +1,18 @@
+from .utils.prompts import budgeting_prompt
+from langchain_core.messages import SystemMessage , ToolMessage, FunctionMessage
+from.utils.tools import cost_table,budget_tiers_schema
+from utils.models_loader import llm
+from .utils.state import json_schema, VideoIdeasBudgetSchema
+
+def budget_calculator(business_details, final_ideation):
+    print('The final ideation is:',final_ideation)
+    print('type:',type(final_ideation))
+    template = budgeting_prompt(final_ideation,business_details)
+    messages = [SystemMessage(content=template),
+                ToolMessage(content="The cost table is:\n"+ str(cost_table) + "\n" + "The budget tier schema is:\n"+ str(budget_tiers_schema) ,tool_call_id='call_cost_table'),
+                # FunctionMessage(name="cost_and_budget",content="The cost table is:\n"+ str(cost_table) + "\n" + "The budget tier schema is:\n"+ str(budget_tiers_schema) ,tool_call_id='call_cost_table')]
+                ]
+    response = llm.with_structured_output(VideoIdeasBudgetSchema).invoke(messages)
+    print('Budget Calculator Response:\n',response)
+    return response
+

idea_to_budget_agent/utils/prompts.py

@@ -0,0 +1,43 @@
+
+def budgeting_prompt(ideas_block,business_details) -> str:
+    return f"""
+You are a video production strategist for Instagram and TikTok reels. You will receive a list of 4 creative short-form video ideas. These ideas are designed to promote businesses from various industries (e.g., fitness, fashion, tech, finance, food, etc.).
+
+Your task is to analyze each video idea and assign them to the most appropriate budget range, based on the **cost table** and **budget tiers schema** which is provided to you as a tool.
+
+
+Instructions: (For each video idea)
+1. Evaluate what it would realistically take to produce each video idea.
+2. Assign a suitable budget range and label to each.
+3. Provide a justification for your choice, based on production complexity.
+4. Output your answer in the following format:
+
+Output Format: (For each video idea, Give the result in the following format)
+"idea": "Display the video idea as it is.",
+"mapped_budget_range": "The best suitable budget range from budget tiers schema. Include the price range mentioned too.", 
+"budget_label": "The best fit budget label from the budget tiers schema.",
+"justification": "Clean and precise reason why this idea can be mapped in this budget range and budget label.",
+"recommended_team_and_tools": "Display the cost detailing by deeply analyzing the cost table. You have to give a clear response that which portion needs how much money. Remember, the cost detailing don't have to exceed the fitted budget range."
+
+🎬 Here are the video ideas to evaluate along with the business_details:
+**4 video ideas**
+1. **First Idea**
+{ideas_block[0]}
+
+2. **Second Idea**
+{ideas_block[1]}
+
+3. **Third Idea**
+{ideas_block[2]}
+
+4. **Fourth Idea**
+{ideas_block[3]}
+
+**Business Details**:
+{business_details}
+
+📝 Notes:
+- Be objective and realistic in budget assessment.
+- Do not assign high-end budgets unless absolutely necessary.
+- Assume the creator wants maximum value with minimal overspending.
+"""

idea_to_budget_agent/utils/state.py

@@ -0,0 +1,45 @@
+json_schema = {
+    "idea1":{
+        "idea": "Display the first video idea as it is.",
+        "mapped_budget_range": "The best suitable budget range from budget tiers schema. Include the price range mentioned too.", 
+        "budget_label": "The best fit budget label from the budget tiers schema.",
+        "justification": "Clean and precise reason why this idea can be mapped in this budget range and budget label.",
+        "recommended_team_and_tools": "Display the cost detailing by deeply analyzing the cost table. You have to give a clear response that which portion needs how much money. Remember, the cost detailing don't have to exceed the fitted budget range."
+   },
+
+    "idea2":{
+        "idea": "Display the second video idea as it is.",
+        "mapped_budget_range": "The best suitable budget range from budget tiers schema. Include the price range mentioned too.", 
+        "budget_label": "The best fit budget label from the budget tiers schema.",
+        "justification": "Clean and precise reason why this idea can be mapped in this budget range and budget label.",
+        "recommended_team_and_tools": "Display the cost detailing by deeply analyzing the cost table. You have to give a clear response that which portion needs how much money. Remember, the cost detailing don't have to exceed the fitted budget range."
+   },
+
+    "idea3":{
+        "idea": "Display the third video idea as it is.",
+        "mapped_budget_range": "The best suitable budget range from budget tiers schema. Include the price range mentioned too.", 
+        "budget_label": "The best fit budget label from the budget tiers schema.",
+        "justification": "Clean and precise reason why this idea can be mapped in this budget range and budget label.",
+        "recommended_team_and_tools": "Display the cost detailing by deeply analyzing the cost table. You have to give a clear response that which portion needs how much money. Remember, the cost detailing don't have to exceed the fitted budget range."
+   },
+
+    "idea4":{
+        "idea": "Display the fourth video idea as it is.",
+        "mapped_budget_range": "The best suitable budget range from budget tiers schema. Include the price range mentioned too.", 
+        "budget_label": "The best fit budget label from the budget tiers schema.",
+        "justification": "Clean and precise reason why this idea can be mapped in this budget range and budget label.",
+        "recommended_team_and_tools": "Display the cost detailing by deeply analyzing the cost table. You have to give a clear response that which portion needs how much money. Remember, the cost detailing don't have to exceed the fitted budget range."
+   },
+}
+
+from pydantic import BaseModel, Field
+
+class IdeaMapping(BaseModel):
+    idea: str = Field(..., description="The video idea as it is.")
+    mapped_budget_range: str = Field(..., description="Best suitable budget range from budget tiers schema. Include price range.")
+    budget_label: str = Field(..., description="Best fit budget label from the budget tiers schema.")
+    justification: str = Field(..., description="Reason why this idea fits in the given budget.")
+    recommended_team_and_tools: str = Field(..., description="Detailed cost breakdown. Must not exceed budget.")
+
+class VideoIdeasBudgetSchema(BaseModel):
+    ideas: list[IdeaMapping] = Field(..., description="List of 4 video ideas mapped with budget, justification, and cost breakdown.")

idea_to_budget_agent/utils/tools.py

@@ -0,0 +1,101 @@
+cost_table={
+  "production_cost_table": [
+    {
+      "item": "camera_rent_daily",
+      "label": "Camera Rental (Daily)",
+      "price_range_npr": "2,000 – 5,000",
+      "description": "Depends on camera quality – DSLR to semi-pro cinema cams"
+    },
+    {
+      "item": "lighting_kit",
+      "label": "Lighting Kit Rental",
+      "price_range_npr": "1,000 – 3,000",
+      "description": "Basic LED lights to full 3-point lighting kits"
+    },
+    {
+      "item": "editor_per_reel",
+      "label": "Video Editing (per reel)",
+      "price_range_npr": "2,000 – 6,000",
+      "description": "Includes transitions, sound sync, color grading"
+    },
+    {
+      "item": "actor_per_day",
+      "label": "Actor Cost (per day)",
+      "price_range_npr": "1,500 – 4,000",
+      "description": "Based on experience and role complexity"
+    },
+    {
+      "item": "voiceover_artist",
+      "label": "Voiceover Artist",
+      "price_range_npr": "1,000 – 3,000",
+      "description": "For narration or character lines in the reel"
+    },
+    {
+      "item": "location_per_day",
+      "label": "Location Rental (per day)",
+      "price_range_npr": "1,000 – 5,000",
+      "description": "Standard gym, studio, or unique outdoor location"
+    },
+    {
+      "item": "props_and_set",
+      "label": "Props and Set Design",
+      "price_range_npr": "500 – 3,000",
+      "description": "Includes small items, backdrops, thematic decoration"
+    },
+    {
+      "item": "costume_and_makeup",
+      "label": "Costume and Makeup",
+      "price_range_npr": "1,000 – 3,500",
+      "description": "If any character styling or transformation is needed"
+    },
+    {
+      "item": "background_music_license",
+      "label": "Background Music License",
+      "price_range_npr": "0 – 1,500",
+      "description": "Royalty-free or platform-safe music use"
+    },
+    {
+      "item": "sound_engineering",
+      "label": "Sound Mixing & Sync",
+      "price_range_npr": "1,000 – 2,500",
+      "description": "Cleans background noise, balances music/voice"
+    },
+    {
+      "item": "crew_support",
+      "label": "Crew Support (Assistant, Director)",
+      "price_range_npr": "2,000 – 6,000",
+      "description": "Helps with camera, lighting, direction during shoot"
+    }
+  ]
+}
+
+budget_tiers_schema = {
+    "budget_tiers": [
+        {
+            "range_npr": "< 5,000",
+            "label": "Ultra-Low"
+        },
+        {
+            "range_npr": "5,000 – 10,000",
+            "label": "Low Budget"
+        },
+        {
+            "range_npr": "10,000 – 15,000",
+            "label": "Moderate Budget"
+        },
+        {
+            "range_npr": "15,000 – 20,000",
+            "label": "Enhanced"
+        },
+        {
+            "range_npr": "20,000 – 25,000",
+            "label": "Premium"
+        },
+        {
+            "range_npr": "25,000+",
+            "label": "High-End"
+        }
+    ]
+}
+
+

ideation_agent/utils/nodes.py

@@ -26,7 +26,6 @@ def ideator(state:State):
     template = ideator_prompt(state)
     messages = [SystemMessage(content=template),
                 HumanMessage(content=f'''The topic of the video is:\n{state.topic[-1]}\n''')]
-    print('Ideator prompt:', messages)
     try:
         response = ideator_agent.invoke({'messages':messages})
         response = response['messages'][-1].content
@@ -45,7 +44,6 @@ def critic(state:State):
     template = critic_prompt(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]}\n''')]
-    print('Critic prompt:', messages)
     
     try:
         response = critic_agent.invoke({'messages':messages})
@@ -63,29 +61,23 @@ def critic(state:State):
 
 
 def improver(state:State):
+    response_list = []
     template = improver_prompt(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)
+    print('Improver Prompt:',messages)
     response = improver_llm.with_structured_output(ImproverResponseFormatter).invoke(messages)
-    state.improver_response.append(response.improved_idea)
+    response_list.append(response.improved_idea1)
+    response_list.append(response.improved_idea2)
+    response_list.append(response.improved_idea3)
+    response_list.append(response.improved_idea4)
+
+    state.improver_response.append(str(response_list))
     state.critic_fault.append(response.faults)
-    print('Improver response:',response.improved_idea)
+    print('Improver response:',response_list)
     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,42 +1,49 @@
-example = '''
-**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"
+def ideator_prompt(state):
+     return f"""
+You are a **bold, imaginative, and culturally-attuned video ideator** trusted by top global brands to craft **crisp, original, and high-impact video concepts** for social platforms like TikTok, Instagram, and YouTube.
 
-**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.
+Your task is to create **exactly 4 highly original video ideas**, each in **exactly 40 words**. These are not scripts or taglines — they are **powerful conceptual seeds**: short, visual, emotionally charged ideas that can spark full videos. Your ideas should feel fresh, scroll-stopping, and deeply resonant with today’s audiences.
 
 ---
-'''
 
 
+### What is a 40-word video idea?
 
-def ideator_prompt(state):
-    return f'''
-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**.
+A video idea is a **standalone, high-concept creative spark**. It hints at a compelling moment, scenario, or transformation — not the full story. It’s short, but strong enough to fuel the entire video direction.
 
-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.
+This differs from a full story in that:
+- You’re generating **4 completely distinct mini-concepts**, not one long arc.
+- Each idea stands alone — a unique emotional core.
+- Think in **cinematic micro-moments**, full of emotion, surprise, or visual contrast.
 
-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.
+### Output & Formatting Guidelines:
+- Provide **exactly 4 ideas**, each in **exactly 40 words** .
+- Do **not** number or label the ideas.
+- Each idea should be a **self-contained, cinematic pitch** — visual, emotionally compelling, and easily translatable into video.
+- Avoid lists, taglines, or instructions. Focus on **visually-driven storytelling**.
+- Be **creative with characters, symbols, or twists** — do not simply rephrase the business info.
+- Start from **an unexpected or abstract scene**, character, or event — then subtly anchor it to the business concept in the conclusion.
+- Ensure every idea is **platform-native**, emotionally viral, and rooted in **the business details provided**.
 
+---
+### Very Important Reminder:
+- Do not start or potray the idea directly by  context of the topic and business details. Start from entirely different plot , event or character and conclude to the topic and details at last making it creative and unique just like in the example that includes coffee slipping before the date , earthquake surviving , security guard in the luxury mall etc. These starting plots differs from the clothing business but they creatively concluded at the end.
+- Every 4 ideas must be very diverse. They have to be completely different from each other.
+- NEVER use the contexts of topics as opening scenarios - Begin with completely unrelated situations like accidents, mysteries, competitions, discoveries, or social experiments. For eg: If the context is about restaurant don't start with chef, restaurant or related things directly.
+- Avoid repeating similar character types - **Do not use same characters , or same workplace scenarios, or same situations etc across the 4 ideas**.
+- Please think **Out of the box** ideas. Surprise like a creative genius. Your idea can never be thought by a simple human. 
+
+---
 
 ---
 
+### Your Assignment:
+
 **Video Topic**:  
 {state.topic[-1]}
 
@@ -45,155 +52,203 @@ Use the response of the provided tool (which includes video stories from influen
 
 ---
 
-**Important Instructions**:
-- 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.
+### Example Reference:
 
----
+**Business Details**:
+"business_type": "women’s ethnic fashion brand",  
+"platform": "Instagram and TikTok",  
+"target_audience": "young Nepali women (ages 18–35) who value tradition with a modern twist",  
+"business_goals": "to grow a loyal, style-forward female customer base and expand across all major cities in Nepal",  
+"offerings": "authentic Nepali ethnic wear reimagined with bold colors, modern cuts, and cultural pride — from dhaka dresses to fusion kurtis and handcrafted accessories",  
+"Challenges_faced": "struggling to break into urban youth markets and stand out in a saturated online fashion scene"  
 
-**To better understand the expected tone, depth, and style, here are two sample ideas with their exact business contexts**:
+**Video Ideas (each exactly 40 words):**
 
----
+A woman accidentally spills coffee on her modern outfit before a date. Panicking, she raids her grandmother’s trunk and pieces together a fusion look using dhaka fabrics. Her date is mesmerized. Twist: he’s also in traditional wear. The moment is magical — a perfect blend of heritage and attraction.
 
-**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"
+An earthquake survivor finds her childhood home in ruins. Among the debris, she discovers her mother’s preserved ethnic dress. She wears it to job interviews, her graduation, and eventually her wedding. The same outfit evolves into her symbol of survival, growth, and deep cultural pride.
 
-**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.
+A security guard watches a woman silently admire expensive western outfits daily. One day, she walks in wearing a vibrant dhaka fusion kurti. Cameras click, people stare. A luxury brand manager approaches her. The message: true style doesn’t come with a price tag — it comes with roots.
 
 ---
 
-**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"
+Now, based on the video topic and business details above, generate **four original, 40-word promotional video ideas** that can inspire unforgettable video content.
+"""
 
-**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 a sharp, imaginative, and detail-oriented **video idea critic**. You’ve been assigned to **critique 4 video ideas** created by another ideator and then **refine or improve** them.
 
+Your job is twofold:
+1. **Identify flaws** (if any) in one or more of the ideas — repetition, weak twist, boring start, lack of connection to business, etc.
+2. **Generate a better version** of each idea — keeping it true to the core if it’s already good, or replacing it if needed.
 
-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.
+### Very Important Creative Guidelines:
 
-  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.
+- **Each idea must be exactly 40 words**.
+- Ideas must start with an **unrelated or abstract plot, character, or situation** — never directly with the business context.
+- Please **Don't potray the topic and  business details in the beginning of the idea**. The topic and business details have to come into picture as the main conclusion of the idea.
+- Every 4 ideas must be very diverse. They have to be completely different from each other.
+- Each idea must **creatively conclude** by connecting to the business.
+- **Avoid repeating** characters, locations, or similar scenarios across the four ideas.
+- **Refine only when needed**. If an original idea is already great, make light edits — don’t change it completely.
+- Make ideas **scroll-stopping**, emotional, and suitable for Instagram/TikTok/YouTube.
+- Do **not output critique per idea**. Instead, summarize faults collectively in one section if any.
 
-  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.
+---
 
-  ---
+### Video Topic:
+{state.topic[-1]}
 
-  **Video Topic**:  
-  {state.topic[-1]}
+### Business Details:
+{state.business_details[-1]}
 
-  **Business Details**:  
-  {state.business_details[-1]}
+### Original Ideas by Ideator:
+{state.ideator_response[-1]}
 
-  **Original Video Idea** (100 words):  
-  {state.ideator_response[-1]}
+---
 
-  ---
+**Format your response like this:**
+
+---
+Faults: Faults in any of the 4 ideas of ideator  
+
+---
+Improved Ideas from Critic:  
+**improved_first idea of 40 words**  
+**improved_second idea of 40 words**  
+**improved_third idea of 40 words**  
+
+---
+"""
 
-   **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"**Note:** This idea was not validated earlier because:\n{state.disagreement_reason[-1]}\n"
+        f"**Note:** The previous version was not validated 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 **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:
+    return f"""
+You are a highly skilled, creative, and discerning **video concept improver**. Two previous agents — the *Ideator* and the *Critic* — have worked on this task. Now, it’s your job to **analyze both**, identify any faults in the *critic's refinements*, and deliver the **final, most powerful version** of the 4 video ideas.
+
+---
+
+### Your Responsibilities:
 
-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.
+1. **Read the original 4 ideas from the Ideator.**
+2. **Read the improved 4 ideas from the Critic.**
+4. Identify any issues in the *critic’s version*, such as:
+   - Loss of originality from ideator's idea
+   - Overcorrection or unnecessary replacement
+   - Weak twists, lack of emotional resonance, dull opening
+   - Repetition of themes, settings, characters across the 4 ideas
 
-If the critic’s idea is already strong and has no faults, simply **repeat it** as your final version.
+You do **not need to critique the ideator again** — focus **only on identifying the issues in the critic's improved ideas**.
 
+4. Then, write **your final improved version of the 4 ideas**, using the best elements from both versions (or inventing better ones).
+   - Stick to **exactly 40 words per idea**
+   - Start with a surprising or unrelated event/character/symbol
+   - Conclude creatively with the business offering or mission
+   - Ensure **diversity** between the 4 ideas — no repetition of theme, tone, or character
+   - Refine only when needed. If the improved idea of critic is already great, make light edits or no edits — don’t change it completely.
+   -
+
+---
+
+### Very Important Reminder:
+- Do not start or potray the idea directly by  context of the topic and business details. Start from entirely different plot , event or character and conclude to the topic and details at last making it creative and unique just like in the example that includes coffee slipping before the date , earthquake surviving , security guard in the luxury mall etc. These starting plots differs from the clothing business but they creatively concluded at the end.
+- Every 4 ideas must be very diverse. They have to be completely different from each other.
+- NEVER use the contexts of topics as opening scenarios - Begin with completely unrelated situations like accidents, mysteries, competitions, discoveries, or social experiments. For eg: If the context is about restaurant don't start with chef, restaurant or related things directly.
+- Avoid repeating similar character types - **Do not use same characters , or same workplace scenarios, or same situations etc across the 4 ideas**.
+- Please **Don't potray the topic and  business details in the beginning of the idea**. The topic and business details have to come into picture as the main conclusion of the idea.
+- Please think **Out of the box** ideas. Surprise like a creative genius. Your idea can never be thought by a simple human.
 ---
 
+### Output Format (Strict):
+
+---
+Faults: Faults in any of the 4 ideas of critic  
+
+---
+Improved Ideas from Improver:  
+**improved_final_first idea of 40 words**  
+**improved_final_second idea of 40 words**  
+**improved_final_third idea of 40 words**  
+
+---
+
+### Input:
+
 **Video Topic**:  
 {state.topic[-1]}
 
 **Business Details**:  
 {state.business_details[-1]}
 
-**Critic's Response**:  
-{state.critic_response[-1]}
+---
 
-{disagreement_note}
+**Ideas from Ideator**:  
+{state.ideator_response[-1]}
 
 ---
 
-**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.**
+**Faults of ideator and improved Ideas from Critic**:  
+{state.critic_response[-1]}
 
-{example}
-'''
+**{disagreement_note}**
 
+---
 
-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.
+Now begin your final analysis and output your improved 4 ideas in the required format.
+"""
 
-  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.
 
-  ---
 
+def validator_prompt(state):
+    return f'''
+You are reviewing 4 short video ideas meant for a social media promotional campaign.
 
-  **Video Topic**:
-  {state.topic[-1]}
+Each idea creatively tells a short story that connects indirectly to the business details — starting from an unrelated or surprising scenario and ending with a clever link back to the business theme.
 
-  **Business Details**:
-  {state.business_details[-1]}
+---
 
-  **Final idea from Improver**:
-  {state.improver_response[-1]}
+**Video Topic**:  
+{state.topic[-1]}
+
+**Business Details**:  
+{state.business_details[-1]}
 
-  ---
+**Final 4 Ideas from Improver**:  
+{state.improver_response[-1]}
 
-  **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?
+### Friendly Review Guidelines:
+- Each idea should begin with an **unusual or imaginative scenario** (like a mystery, surprise, or random event) and **end by cleverly connecting to the business details** (like goals, services, or audience).
+- The ideas should be **visually different** and **emotionally engaging**, avoiding repetition of characters or scenes across all four.
+- It's okay if the ideas are metaphorical, humorous, or slightly abstract — as long as the **final message feels connected to the business**.
+- Only mark “not validated” if one or more ideas:  
+  • start too obviously with the business context,  
+  • never connect to the business at all,  
+  • or feel almost identical to each other.
+---
 
-  ---
 
-  **Output Instruction**: There are two outputs you have to give.
+### Output Instructions:
+Return exactly 2 things: 
 
-      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.
+1. **Validation status** — Only respond with `validated` or `not validated`.  
+2.**If not validated**, give a short, kind explanation (under 40 words) of what could be improved, or why isn't it validated.
+'''
 
-      2. Give a short reason of 30 words why the idea is validated or not validated.
-  '''

ideation_agent/utils/state.py

@@ -22,7 +22,14 @@ class QueryFormatter(BaseModel):
 
 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.")
+   improved_idea1: str = Field(description="Returns the first improved idea of exactly 40 words. If no faults found, repeat the critic's version.")
+   improved_idea2: str = Field(description="Returns the second improved idea of exactly 40 words. If no faults found, repeat the critic's version.")
+   improved_idea3: str = Field(description="Returns the third improved idea of exactly 40 words. If no faults found, repeat the critic's version.")
+   improved_idea4: str = Field(description="Returns the fourth improved idea of exactly 40 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.")

main.py

@@ -9,6 +9,7 @@ from business_interaction_agent.agent import BusinessInteractionChatbot
 from context_analysis_agent.utils.utils import save_to_db
 import ast
 from brainstroming_agent.utils.utils import encode_image_to_base64 , generate_final_story, generate_image
+from idea_to_budget_agent.agent import budget_calculator
 from ideation_agent.agent import ideation_graph
 from langgraph.errors import GraphRecursionError
 
@@ -23,8 +24,22 @@ idea_graph = ideation_graph()
 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['final_ideation'] = {}
+# stored_data['business_details']={"business_type": "fitness and gym", "platform": "instagram, tiktok", "target_audience": "every software needed clients", "business_goals": "to expand in every cities of nepal", "offerings": "nepali cloths and products", "Challenges_faced": "finding new customers, attracting large customers"}
+stored_data['business_details'] = {
+    "business_type": "fitness and gym",
+    "platform": "Instagram, TikTok",
+    "target_audience": "young Nepali adults (ages 18–40) who are health-conscious and active on social media",
+    "business_goals": "to expand gym branches across all major cities of Nepal and build a strong fitness community",
+    "offerings": "personal training, group fitness classes, modern workout equipment, nutrition guidance, and wellness programs",
+    "Challenges_faced": "attracting loyal members, standing out in a competitive market, and promoting consistent engagement"
+}
+
+
+
+stored_data['final_ideation']= ['''A street magician performs tricks, leaving a crowd awestruck. One trick falters, inspiring him to hit the gym. Months later, he fuses fitness with magic, showcasing strength through performance, proving that transformation can captivate and inspire a community''',
+                                                  '''A group of strangers attends a quirky food festival, each trying bizarre dishes. One adventurer finds a fitness coach among them. They bond over health, sparking a weekly workout group that brings the community together, blending fun and fitness effortlessly.''',
+                                                    '''A lost hiker discovers a hidden waterfall, but slips on the rocks, nearly falling. A passing fitness enthusiast helps him up, encouraging regular exercise for safety and strength. Together, they start a hiking club, uniting fitness and nature lovers in their city''',
+                                                    '''A lost hiker discovers a hidden waterfall, but slips on the rocks, nearly falling. A passing fitness enthusiast helps him up, encouraging regular exercise for safety and strength. Together, they start a hiking club, uniting fitness and nature lovers in their city.''']
 class UserMessage(BaseModel):
     message: str
 @app.post("/context-analysis")
@@ -56,7 +71,7 @@ class IdeationRequest(BaseModel):
 
 @app.post("/ideation")
 def ideation_endpoint(request:IdeationRequest):
-    config={"recursion_limit":10, "configurable": {"thread_id": "ideation_thread123"}}
+    config={"recursion_limit":15, "configurable": {"thread_id": "ideation_thread123"}}
     try:
         result = idea_graph.invoke(
         {
@@ -66,12 +81,20 @@ def ideation_endpoint(request:IdeationRequest):
         config=config,
         )
         stored_data['final_ideation'] = result['improver_response'][-1]
+        stored_data['final_ideation']=ast.literal_eval(stored_data['final_ideation'])
+
         return {'response':result}
     except GraphRecursionError:
         result = idea_graph.get_state({"configurable": {"thread_id": "ideation_thread123"}})
         return {'response': result[0]}
     
+class BudgetMappingRequest(BaseModel):
+    pass
 
+@app.post("/budget-mapping")
+def budget_mapping_endpoint():
+    result = budget_calculator(stored_data["business_details"],stored_data['final_ideation'])
+    return {'response':result}
 
 
 class BrainstormRequest(BaseModel):
@@ -84,7 +107,7 @@ def brainstroming_endpoint(
     request: BrainstormRequest,  # 🔥 Full JSON body here
 ):
     result = brainstrom_graph.invoke({
-        'topic': [stored_data['final_ideation']],
+        'topic': [stored_data['final_ideation'][0]],
         '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)
@@ -104,7 +127,9 @@ def generate_final_story_endpoint():
     return {
         'response': final_story
     }
-
+stored_data['final_story']= '''A cinematic journey follows a street magician\'s 
+metamorphosis from a mere trickster to a powerful performer, as he transforms his act with newfound physical strength, effortlessly executing death-defying stunts, and inspiring a captivated crowd to take action, all set against a 
+backdrop of urban grandeur and pulsing energy.'''
 @app.post("/generate-image")  
 def generate_image_endpoint():
     image = generate_image(str(stored_data['final_story']))

utils/models_loader.py

@@ -1,7 +1,7 @@
 from langchain_groq import ChatGroq
 from langchain_openai import ChatOpenAI
-from sentence_transformers import SentenceTransformer
-from huggingface_hub import InferenceClient
+# from sentence_transformers import SentenceTransformer
+# from huggingface_hub import InferenceClient
 from huggingface_hub import login
 from dotenv import load_dotenv
 load_dotenv()
@@ -17,14 +17,12 @@ 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",
 #     temperature=0.3,
-#     max_tokens=500,
 # )
 
 
@@ -55,37 +53,42 @@ ST = HFEmbeddingAPI(
 
 # Initiated the models for ideation
 
-ideator_llm = ChatGroq(
-    model="llama3-8b-8192",
-    temperature=0.3,
-    max_tokens=500,
+# ideator_llm = ChatGroq(
+#     model="llama-3.1-8b-instant",
+#     temperature=0.7,
+#     max_tokens=500,
 
-)
+# )
 
-critic_llm = ChatGroq(
-    model="llama-3.3-70b-versatile",
-    temperature=0.3,
-    max_tokens=500,
+# critic_llm = ChatGroq(
+#     model="llama-3.3-70b-versatile",
+#     temperature=0.7,
+#     max_tokens=500,
 
-)
+# )
 
 # improver_llm = ChatOpenAI(
 #     model="gpt-4o-mini",
-#     temperature=0.3,
+#     temperature=0.7,
 #     max_tokens=500,
 # )
 
 improver_llm = ChatGroq(
-    model="meta-llama/llama-4-scout-17b-16e-instruct",
-    temperature=0.3,
+    model="llama3-8b-8192",
+    temperature=0.7,
     max_tokens=500,
 
 )
-validator_llm = ChatGroq(
-    model="llama-3.3-70b-versatile",
-    temperature=0.3,
-    max_tokens=500,
+ideator_llm = improver_llm
+critic_llm = improver_llm
+validator_llm = improver_llm
 
-)
+
+# validator_llm = ChatGroq(
+#     model="llama-3.3-70b-versatile",
+#     temperature=0.7,
+#     max_tokens=500,
+
+# )