Blog with Images using Ray

blog_with_images_ray.py

import os

import ray
from dotenv import load_dotenv
from langchain_community.tools import DuckDuckGoSearchRun

from motleycrew.agents.crewai import CrewAIMotleyAgent
from motleycrew.agents.langchain import ReActToolCallingMotleyAgent
from motleycrew.common import configure_logging
from motleycrew.tools.image.dall_e import DallEImageGeneratorTool

configure_logging(verbose=True)
load_dotenv()


@ray.remote
def produce_blog_post():
    configure_logging(verbose=True)

    search_tool = DuckDuckGoSearchRun()

    researcher = CrewAIMotleyAgent(
        role="Senior Research Analyst",
        goal="Uncover cutting-edge developments in various fields, doing web search if necessary",
        backstory="""You work at a leading tech think tank.
    Your expertise lies in identifying emerging trends.
    You have a knack for dissecting complex data and presenting actionable insights.""",
        verbose=True,
        tools=[search_tool],
    )

    writer = ReActToolCallingMotleyAgent(
        name="AI writer agent",
        tools=[researcher],
        verbose=True,
    )

    prompt = """You are an experienced blog writer.
Conduct a comprehensive analysis of the latest advancements in AI in 2024.
Identify key trends, breakthrough technologies, and potential industry impacts.
Using the insights provided by a thorough research, develop an engaging blog post
that highlights the most significant advancements in AI.
Your post should be informative yet accessible, catering to a tech-savvy audience.
Make it sound cool, avoid complex words so it doesn't sound like AI.
Create a blog post of at least 4 paragraphs, in markdown format."""

    blog_post = writer.invoke({"prompt": prompt})
    return blog_post


@ray.remote
def create_illustrations(blog_post: str):
    configure_logging(verbose=True)

    image_generator_tool = DallEImageGeneratorTool(
        os.path.realpath("./images"),
        model="dall-e-3",
        refine_prompt_with_llm=True,
        size="1024x1024",
        style="vivid",
    )

    illustrator = ReActToolCallingMotleyAgent(
        name="Illustrator",
        description="Create beautiful and insightful illustrations for a blog post",
        tools=[image_generator_tool],
    )

    prompt = f"""Create beautiful and insightful illustrations to accompany the provided blog post on AI advancements.
The blog post will be provided to you in markdown format.
Make sure to use the illustration tool provided to you, once per illustration, and embed the URL provided by
the tool into the blog post.
Create between 2 and 3 illustrations, neither more nor less
Only return the full text of the blog post with embedded URLs, no backticks, no other text.
Only use slashes as path separators, never backslashes.

The blog post is:

{blog_post}"""

    illustrated_blog_post = illustrator.invoke({"prompt": prompt})
    return illustrated_blog_post


# This creates a lazy computation graph: https://docs.ray.io/en/latest/ray-core/ray-dag.html
blog_post_ref = produce_blog_post.bind()  # Create writing node
illustrated_blog_post_ref = create_illustrations.bind(
    blog_post=blog_post_ref
)  # Create illustration node

# Evaluate the relevant node, which will also evaluate all the nodes that it depends on, topologically
illustrated_blog_post = ray.get(illustrated_blog_post_ref.execute())

here = os.path.realpath(".").replace(os.sep, "/")
print(illustrated_blog_post.replace(here, "."))