Generate Creative Images from Text

from typing import Any

from camel.prompts import TextPrompt, TextPromptDict
from camel.types import RoleType

class ImageCraftPromptTemplateDict(TextPromptDict):
r"""A dictionary containing :obj:TextPrompt used in the ImageCraft
task.

Attributes:
    ASSISTANT_PROMPT (TextPrompt): A prompt for the AI assistant to create
        an original image based on the provided descriptive captions.
"""

ASSISTANT_PROMPT = TextPrompt(
    """You are tasked with creating an original image based on
    the provided descriptive captions. Use your imagination
    and artistic skills to visualize and draw the images and
    explain your thought process."""
)

def __init__(self, *args: Any, **kwargs: Any) -> None:
    super().__init__(*args, **kwargs)
    self.update(
        {
            RoleType.ASSISTANT: self.ASSISTANT_PROMPT,
        }
    )