Function improve_code

autofigure/generator.py:1562–1734

Improvement Agent: Generate improved code (SVG or mxGraph XML) based on previous critique Use initial generation prompt + previous evaluation result to guide improvement

(
    code: str,
    code_image: Image.Image,
    paper_content: str,
    reference_figures: List[Image.Image],
    iteration: int,
    previous_critique: Optional[Dict],
    human_guidance: Optional[str] = None,
    topic: str = 'paper',
    output_format: str = None
)

def improve_code(
    code: str,
    code_image: Image.Image,
    paper_content: str,
    reference_figures: List[Image.Image],
    iteration: int,
    previous_critique: Optional[Dict],
    human_guidance: Optional[str] = None,
    topic: str = 'paper',
    output_format: str = None
) -> Optional[str]:
    """
    Improvement Agent: Generate improved code (SVG or mxGraph XML) based on previous critique
    Use initial generation prompt + previous evaluation result to guide improvement
    """
    output_format = output_format or CONFIG.get('OUTPUT_FORMAT', 'svg')
    format_name = 'mxGraph XML' if output_format == 'mxgraphxml' else 'SVG'

    # Determine code extraction patterns based on format
    if output_format == 'mxgraphxml':
        start_tag = '<mxfile'
        end_tag = '</mxfile>'
        end_offset = 9
    else:
        start_tag = '<svg'
        end_tag = '</svg>'
        end_offset = 6

    try:
        # Get initial generation prompt as base
        base_prompt = get_initial_prompt_template(topic, paper_content, output_format)

        # Add previous evaluation feedback on top of initial prompt
        improvement_prompt = base_prompt

        # Add previous evaluation result as improvement guidance
        if previous_critique:
            try:
                previous_eval_json_str = json.dumps(previous_critique, ensure_ascii=False, indent=2)
                improvement_prompt += f"""

**Previous Iteration Evaluation (Iteration {iteration-1}):**
The previous design received the following evaluation:

{previous_eval_json_str}

**CRITICAL IMPROVEMENT REQUIREMENTS:**
1. You MUST systematically address each item in 'specific_issues'
2. You MUST implement each item in 'improvement_suggestions'
3. Focus especially on fixing:
   - Component overlap and alignment issues
   - Connector correctness and clarity
   - Layout balance and aesthetic quality
   - Content accuracy and completeness
   - Placeholder compliance
"""
            except Exception as e:
                print(f"[Improvement Agent] Cannot format previous_critique: {e}")