working on post generation

src/_agents.py

@@ -19,8 +19,11 @@ import fal_client
 from PIL import Image
 from io import BytesIO
 from IPython.display import display
-
+from google import genai
 import logging
+
+
+
 logging.basicConfig(level=logging.INFO)
 
 
@@ -30,6 +33,11 @@ aspect_ratios = {
 }
 
 
+client = genai.Client(
+    # vertexai=True, project='1089055075981', location='us-central1'
+)
+
+
 #----------------------- content agent ----------------------------------
 
 def content_prompt(context: RunContextWrapper, agent: Agent):
@@ -123,7 +131,7 @@ post_schema = """
 
 
 @function_tool
-def generate_designSpec_from_brief(design_brief: str, defaultValues:str) -> dict:
+def generate_designSpec_from_brief(design_brief: str) -> dict:
     """Generates a detailed design specification in JSON format from a high-level design brief.
 
     This function takes a natural language description of a social media post and uses an AI model
@@ -132,7 +140,6 @@ def generate_designSpec_from_brief(design_brief: str, defaultValues:str) -> dict
 
     Args:
         design_brief: A string containing the high-level creative brief for the social media post.
-        defaultValues: A string providing default values to consider, such as aspect ratio (e.g., "aspect_ratio:1:1").
 
     Returns:
         A dictionary containing the detailed design specification in JSON format.
@@ -158,11 +165,10 @@ Rules:
 design_brief:
 {design_brief}
 
-defaultValues:{defaultValues}
 """
 
 
-    prompt = schema_template.format(design_brief=design_brief,defaultValues=defaultValues,post_schema=post_schema, )
+    prompt = schema_template.format(design_brief=design_brief,post_schema=post_schema, )
 
     
     response = client.models.generate_content(
@@ -183,7 +189,6 @@ defaultValues:{defaultValues}
 
 def image_generate_prompt(design_brief, designSpec):
     return f"""
-You are a professional social media AI specialized in creating high-quality, visually striking images for social media posts. 
 Your goal is to generate attention-grabbing, brand-aligned images that clearly communicate the given message, following design best practices and platform-specific guidelines.
 
 Design Brief:
@@ -219,10 +224,6 @@ async def generate_post_image(
 ):
     """Generates a social media post image based on a design brief and a detailed specification.
 
-    This function uses an AI image generation service to create a visual based on a natural
-    language design brief and a structured JSON design specification. It returns the URL
-    of the generated image.
-
     Args:
         design_brief: A string containing the high-level creative brief for the image.
         designSpec: A string containing the detailed JSON design specification.
@@ -284,7 +285,7 @@ async def generate_post_image(
 
 def media_agent_prompt(context: RunContextWrapper, agent: Agent):
     return """
-You are Media Agent, a professional AI specialized in creating social media post. 
+You are Media Agent, a professional and specialized in creating social media for post. 
 
 Your task:
 1. Receive a high-level user brief describing a social media post idea.

src/main.py

@@ -2,7 +2,7 @@ import os
 import asyncio
 from dataclasses import dataclass
 from agents import set_tracing_disabled, Runner, SQLiteSession, Agent, AgentHooks , RunHooks,  RunContextWrapper, TContext, Tool
-from _agents import content_agent, WebInspectorAgent
+from _agents import content_agent, WebInspectorAgent, media_agent
 from pydantic import BaseModel, Field, ConfigDict
 from model import get_model
 from typing import Optional
@@ -56,32 +56,31 @@ class AgentContext(BaseModel):
 
 
 def media_post_prompt(context: RunContextWrapper, agent: Agent):
-
-    # 4. Provide the generated content and relevant details to the Media Generation Agent to create images or media.
-    # 5. Combine all outputs into a coherent final post suitable for the requested platform.
-
-    return """You are the Post Generation Agent for Social media. 
+    return """You are the Post Generation Agent for Social Media. 
 Your task is to coordinate multiple agents to produce a complete social media post.
 
-You can execute the follow tasks:
+You can execute the following tasks:
 - Content Generation
 - Media Generation 
 
 Steps for Generating Content:
-1. Extract text website.
-2. Pass the extracted text and user brief to the Content Agent to generate engaging platform-specific post text.
+1. Extract website text using the Web Inspector Tool.
+2. Provide the extracted text along with the user's brief to the Content Tool to generate engaging, platform-specific post text.
 
 Steps for Generating Media:
-1. Extract text website.
-
+1. Extract key text and theme colors from the website.
+2. Provide the extracted website content, the user's brief, and the design brief to the Media Tool.
 
-AT THE END RETURN FULL REPORT TO THE UESR
+Final Step:
+- Combine all outputs into a coherent, platform-ready social media post.
+- Return a full report to the user, including the generated content, images, and any design details.
 
 Rules:
 - Do not generate text or media yourself.
-- Always follow the above order.
+- Always follow the steps in order.
 - Validate each agent's output before proceeding to the next step.
-- Ask the user for clarification only if the input is missing or ambiguous."""
+- Ask the user for clarification only if the input is missing or ambiguous.
+"""
 
 
 async def main():
@@ -110,20 +109,26 @@ Generates engaging social media post content based on user input.
                 tool_name="WebInspectorAgent",
                 tool_description="""Extracts details from a website using given URL, such as visible text content, theme colors etc."""
             ),
+            media_agent.as_tool(
+                tool_name="media_agent",
+               tool_description="""
+Generates high-quality, brand-aligned social media images based on the user's brief,
+design specifications, and extracted website content (text, colors, and theme). 
+Follows design guidelines for style, composition, typography, colors, dark/light theme and platform-specific requirements.
+"""
+            )
         ],
         model=get_model('gemini-2.0-flash'),
       
     )
 
     result = await Runner.run(social_media_post_gen_agent, 
-            input="""
-                Create a LinkedIn post about the importance of AI in modern business strategies for this website https://denovers.com
-            """ ,
+            input="""Create a LinkedIn post this website https://denovers.com, Post related the services we provide""" ,
             session=session,
             context=agent_context,
         )
     print('-----------------------------session--------------------------')
-    print(await session.get_items())
+    # print(await session.get_items())
     # print(result)