Function repair_json

autofigure/generator.py:709–757

Attempts to repair a broken JSON string. Args: broken_json: The broken JSON text. error_message: The error message from parsing. Returns: A repaired JSON dictionary, or None on failure.

(broken_json: str, error_message: str)

        raise Exception(f"Failed to generate initial {format_name}: {error_msg}")

def repair_json(broken_json: str, error_message: str) -> Optional[Dict]:
    """
    Attempts to repair a broken JSON string.
    
    Args:
        broken_json: The broken JSON text.
        error_message: The error message from parsing.
        
    Returns:
        A repaired JSON dictionary, or None on failure.
    """
    prompt = f"""
You are a JSON syntax repair expert. The following text was intended to be a JSON object, but it has syntax errors. Please fix it.

**Error message:**
{error_message}

**Broken JSON text:**
```
{broken_json}
```

**Instructions:**
1.  Analyze the error and the text to locate the syntax issue.
2.  Correct the syntax. Common issues are missing quotes, unescaped characters, or trailing commas.
3.  Return **ONLY** the corrected, valid JSON object, without any additional explanations or surrounding text.
"""
    for attempt in range(CONFIG['MAX_REPAIR_RETRIES']):
        print(f"Attempting to repair JSON (Attempt {attempt + 1}/{CONFIG['MAX_REPAIR_RETRIES']})...")
        try:
            response = call_google_genai_multimodal([prompt])
            if response is None:
                continue
            
            fixed_json_match = re.search(r'\{.*\}', response, re.DOTALL)
            if not fixed_json_match:
                continue

            fixed_json_str = fixed_json_match.group(0)
            return json.loads(fixed_json_str)

        except json.JSONDecodeError as e:
            error_message = str(e)
            print(f"Repaired JSON is still invalid: {error_message}")
        except Exception as e:
            print(f"Error calling LLM for JSON repair: {e}")
            
    print("Reached max retries, JSON repair failed.")
    return None

def repair_svg(svg_code: str, error_message: str) -> Optional[str]:
    """