feat: updated the prompt

src/_agents.py

@@ -21,10 +21,16 @@ from io import BytesIO
 from IPython.display import display
 from google import genai
 import logging
+import asyncio
 
 
+# anchor_client = Anchorbrowser(
+#     api_key=os.getenv("ANCHOR_API_KEY")
+# )
 
-logging.basicConfig(level=logging.INFO)
+
+
+# logging.basicConfig(level=logging.INFO)
 
 
 aspect_ratios = {
@@ -38,6 +44,8 @@ client = genai.Client(
 )
 
 
+
+
 #----------------------- content agent ----------------------------------
 
 def content_prompt(context: RunContextWrapper, agent: Agent):
@@ -194,7 +202,7 @@ Your goal is to generate attention-grabbing, brand-aligned images that clearly c
 Design Brief:
 {design_brief}
 
-JSON Schema:
+Design Specification:
 {designSpec}
 
 Design Guidelines:
@@ -290,7 +298,7 @@ You are Media Agent, a professional and specialized in creating social media for
 Your task:
 1. Receive a high-level user brief describing a social media post idea.
 2. Generate a detailed DesignSpec (JSON structured specification) from the brief using 'generate_designSpec_from_brief', including platform, style, content, visuals, colors, typography, composition, lighting, mood, and finishing details.
-3. Using the generated DesignSpec, create a high-quality, brand-aligned social media image using 'generate_post_image' tool.
+3. Using the generated DesignSpec, create a high-quality, brand-aligned social media image using 'generate_post_image' tool, (Don't change the schema use same as generated)
 
 Be concise, professional, and strictly follow the structured DesignSpec and design guidelines provided.
 """
@@ -383,7 +391,7 @@ def collect_theme_data(url: str) -> dict:
         return {"error": str(e)}
 
 
-@function_tool
+# @function_tool
 async def extract_text_tool(url: str) -> str:
     """Visits a webpage at the given URL and returns its content as a markdown string.
 
@@ -405,7 +413,7 @@ async def extract_text_tool(url: str) -> str:
         markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
         print(markdown_content)
         return markdown_content
-
+    
     except RequestException as e:
         return f"Error fetching the webpage: {str(e)}"
     except Exception as e:
@@ -477,15 +485,15 @@ if __name__ == "__main__":
     #     )
     # print(result)
 
-    result =  Runner.run_sync(media_agent, 
-            input="""
-Create a media for posting on linkedin plateform, make it modren and professional.
+#     result =  Runner.run_sync(media_agent, 
+#             input="""
+# Create a media for posting on linkedin plateform, make it modren and professional.
 
-companydetails:
-    Design Atom is a subscription-based design service tailored for AI and tech startups, offering comprehensive product design, branding, and web development solutions. With a fixed monthly rate, dedicated designers, and fast 4-hour average turnaround times, they have completed over 200 projects, helped clients raise $45M+, and reached 100M+ users with their designs. Trusted by 45+ global companies, including Y Combinator-backed startups, Design Atom delivers startup-proven results, streamlining UX, boosting retention, and enhancing conversions. Their free design trial eliminates bottlenecks, supporting startups in building investor-ready, user-friendly products.
-""" ,
-        )
-    print(result)
+# companydetails:
+#     Design Atom is a subscription-based design service tailored for AI and tech startups, offering comprehensive product design, branding, and web development solutions. With a fixed monthly rate, dedicated designers, and fast 4-hour average turnaround times, they have completed over 200 projects, helped clients raise $45M+, and reached 100M+ users with their designs. Trusted by 45+ global companies, including Y Combinator-backed startups, Design Atom delivers startup-proven results, streamlining UX, boosting retention, and enhancing conversions. Their free design trial eliminates bottlenecks, supporting startups in building investor-ready, user-friendly products.
+# """ ,
+#         )
+#     print(result)
 
-    # asyncio.run(collect_theme_data("https://www.denovers.com/"))
+    # asyncio.run(extract_text_tool("https://www.denovers.com/"))
     print('_agents file')

src/main.py

@@ -69,7 +69,7 @@ Steps for Generating Content:
 
 Steps for Generating Media:
 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.
+2. Provide the generated post content (do not update use same as generated), the theme colors, user's brief, and the design brief to the Media Tool.
 
 Final Step:
 - Combine all outputs into a coherent, platform-ready social media post.
@@ -88,9 +88,7 @@ async def main():
     
     session = SQLiteSession("conversation_123")
 
-    agent_context = AgentContext(
-        user_input="Write a LinkedIn post about the importance of AI in modern business strategies."
-        )
+    agent_context = AgentContext( user_input="Write a LinkedIn post about the importance of AI in modern business strategies." )
 
 
 
@@ -101,9 +99,7 @@ async def main():
         tools=[
             content_agent.as_tool(
                 tool_name="content_agent",
-                tool_description="""
-Generates engaging social media post content based on user input.
-                """
+                tool_description="""Generates engaging social media post content based on user input."""
             ),
             WebInspectorAgent.as_tool(
                 tool_name="WebInspectorAgent",
@@ -111,8 +107,7 @@ Generates engaging social media post content based on user input.
             ),
             media_agent.as_tool(
                 tool_name="media_agent",
-               tool_description="""
-Generates high-quality, brand-aligned social media images based on the user's brief,
+                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.
 """
@@ -123,7 +118,7 @@ Follows design guidelines for style, composition, typography, colors, dark/light
     )
 
     result = await Runner.run(social_media_post_gen_agent, 
-            input="""Create a LinkedIn post this website https://denovers.com, Post related the services we provide""" ,
+            input="""Create a LinkedIn post for website https://panaversity.org/, tell what they offer in service, can you give the stunning bento grids in middle with logos""" ,
             session=session,
             context=agent_context,
         )

src/model.py

@@ -9,6 +9,7 @@ _: bool = load_dotenv(find_dotenv())
 google_api_key = os.getenv("GOOGLE_API_KEY")
 # grok_api_key = os.getenv("GROK_API_KEY")
 openrouter_api_key = os.getenv("OPENROUTER_API_KEY")
+openai_api_key = os.getenv("OPENAI_API_KEY")
 
 DEEPSEEK_BASE_URL = "https://api.deepseek.com/v1"
 GROK_BASE_URL = "https://api.x.ai/v1"
@@ -21,6 +22,8 @@ openrouter_client = AsyncOpenAI(base_url=OPENROUTER_BASE_URL, api_key=openrouter
 # deepseek_client = AsyncOpenAI(base_url=DEEPSEEK_BASE_URL, api_key=deepseek_api_key)
 # grok_client = AsyncOpenAI(base_url=GROK_BASE_URL, api_key=grok_api_key)
 gemini_client = AsyncOpenAI(base_url=GEMINI_BASE_URL, api_key=google_api_key)
+openai_client = AsyncOpenAI(api_key=openai_api_key)
+
 
 def get_model(model_name: str):
     if "/" in model_name:
@@ -29,6 +32,8 @@ def get_model(model_name: str):
     #     return OpenAIChatCompletionsModel(model=model_name, openai_client=deepseek_client)
     # elif "grok" in model_name:
     #     return OpenAIChatCompletionsModel(model=model_name, openai_client=grok_client)
+    elif "gpt" in model_name:
+        return OpenAIChatCompletionsModel(model=model_name, openai_client=openai_client)
     elif "gemini" in model_name:
         return OpenAIChatCompletionsModel(model=model_name, openai_client=gemini_client)
     else: