src/genai/idea_to_budget_agent/utils/state.py

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.")