Function get_initial_prompt_template

autofigure/generator.py:266–300

(topic: str, content: str, output_format: str = None)

    return reference_figures

def get_initial_prompt_template(topic: str, content: str, output_format: str = None) -> str:
    if topic not in CONFIG['SUPPORTED_TOPICS']:
        raise ValueError(f"Unsupported topic: {topic}. Supported topics are: {CONFIG['SUPPORTED_TOPICS']}")

    output_format = output_format or CONFIG.get('OUTPUT_FORMAT', 'svg')

    common_placeholder_spec = """
**Placeholder Specification:**
    *   To prepare for final illustration, an icon needs a placeholder.
    *   **Function**: The placeholder's role is to reserve space and provide a clear directive for the illustrator.
    *   **Recommended Implementation**: A clean, professional way to do this is with a gray, rounded-corner rectangle (`<rect rx="8" ry="8" style="fill:#cccccc; stroke:#666666; stroke-width:1;" />`).
    *   **Content (CRITICAL)**: Each placeholder MUST contain two pieces of text:
        *   **Exterior Label**: A concise name for the component, placed **outside** the box (e.g., above it).
        *   **Interior Description**: A detailed English phrase describing the desired icon using the format `[icon]: <description>`, placed **inside** the box (e.g., `[icon]: An icon showing a robot meticulously reviewing a paper`). This description MUST NOT appear in the final illustration but is a crucial instruction and it must be detailed and concrete.
"""

    # Select appropriate prompt template based on output format
    if output_format == 'mxgraphxml':
        if topic == 'paper':
            return _get_paper_mxgraphxml_prompt_template(content, common_placeholder_spec)
        elif topic == 'survey':
            return _get_survey_mxgraphxml_prompt_template(content, common_placeholder_spec)
        elif topic == 'blog':
            return _get_blog_mxgraphxml_prompt_template(content, common_placeholder_spec)
        elif topic == 'textbook':
            return _get_textbook_mxgraphxml_prompt_template(content, common_placeholder_spec)
    else:  # default to svg
        if topic == 'paper':
            return _get_paper_prompt_template(content, common_placeholder_spec)
        elif topic == 'survey':
            return _get_survey_prompt_template(content, common_placeholder_spec)
        elif topic == 'blog':
            return _get_blog_prompt_template(content, common_placeholder_spec)
        elif topic == 'textbook':
            return _get_textbook_prompt_template(content, common_placeholder_spec)

def extract_layout_type(svg_code: str) -> Optional[str]:
    try: