Function repair_mxgraphxml

autofigure/generator.py:831–903

Attempts to repair broken mxGraph XML code. Args: mxgraph_code: The broken mxGraph XML code. error_message: The parsing error message. Returns: The repaired mxGraph XML code, or None after multiple attempts.

(mxgraph_code: str, error_message: str)

def repair_mxgraphxml(mxgraph_code: str, error_message: str) -> Optional[str]:
    """
    Attempts to repair broken mxGraph XML code.

    Args:
        mxgraph_code: The broken mxGraph XML code.
        error_message: The parsing error message.

    Returns:
        The repaired mxGraph XML code, or None after multiple attempts.
    """
    prompt = f"""
You are a professional mxGraph XML (draw.io format) debugging expert. The following mxGraph XML code has an error during parsing. Please fix it.

**Error Message:**
{error_message}

**Broken mxGraph XML Code:**
```xml
{mxgraph_code}
```

**Requirements:**
1.  Carefully analyze the error message and the code to locate the issue.
2.  Fix syntax errors, such as unclosed tags, unescaped special characters, incorrect attributes, etc.
3.  Ensure the repaired code is well-formed XML.
4.  Do not change the visual content of the diagram; only perform syntax repairs.
5.  **Please output the complete, repaired mxGraph XML code directly, without any other explanation.**

**Strict mxGraph XML Structure Requirements:**
- The code must start with <mxfile> and end with </mxfile>
- Must contain: mxfile > diagram > mxGraphModel > root
- All attribute values must be enclosed in double quotes.
- All tags must be properly closed; self-closing tags should end with "/>".
- Special characters must be escaped: & -> &amp;, < -> &lt;, > -> &gt;.
- All mxCell elements must be DIRECT children of <root>
- mxCell with id="0" must exist (root cell)
- mxCell with id="1" must exist with parent="0" (default parent)
- All other mxCells must have a 'parent' attribute
- Output format must be strictly: <mxfile>...</mxfile>
"""

    for attempt in range(CONFIG['MAX_REPAIR_RETRIES']):
        print(f"Attempting to repair mxGraph XML (Attempt {attempt + 1}/{CONFIG['MAX_REPAIR_RETRIES']})...")
        try:
            repaired_code = call_google_genai_multimodal([prompt])

            if repaired_code is None:
                print("LLM returned an empty response, repair failed.")
                continue

            mxfile_start = repaired_code.find('<mxfile')
            mxfile_end = repaired_code.rfind('</mxfile>') + 9
            if mxfile_start == -1 or mxfile_end == 8:
                print("No valid mxGraph XML code found in the repaired response.")
                continue

            repaired_mxgraph = repaired_code[mxfile_start:mxfile_end]