utils.py

import re

def parse_resolution(resolution_str):
    """Parse resolution string into width and height"""
    match = re.search(r"(\d+)\s*[×x]\s*(\d+)", resolution_str)
    if match:
        return int(match.group(1))), int(match.group(2)))
    return 1024, 1024  # Default fallback

def clean_model_output(text):
    """Clean model output for display"""
    text = re.sub(r"```.*?```", "", text, flags=re.DOTALL)
    text = re.sub(r"\n+", "\n", text).strip()
    return text