Function parse_generated_content

app.py:16–66

Parse the generated content into a dictionary of file paths and contents. Expects the format: file_content here

(content: str)

warnings.filterwarnings("ignore")

def parse_generated_content(content: str) -> Dict[str, str]:
    """
    Parse the generated content into a dictionary of file paths and contents.
    Expects the format:

    <file_path>
    <begin_content>
    file_content here
    <end_content>
    """
    files = {}
    current_file = None
    current_content = []
    state = 'expect_file'

    lines = content.strip().split('\n')
    line_num = 0
    while line_num < len(lines):
        line = lines[line_num]
        stripped = line.strip()
        line_num += 1

        if state == 'expect_file':
            if stripped.startswith('<') and stripped.endswith('>'):
                current_file = stripped[1:-1].strip()
                if not current_file:
                    current_file = None
                    continue
                state = 'expect_begin_content'
            else:
                continue  # Ignore any text before the first <file_path>
        elif state == 'expect_begin_content':
            if stripped == '<begin_content>':
                current_content = []
                state = 'collect_content'
            else:
                current_file = None
                state = 'expect_file'
        elif state == 'collect_content':
            if stripped == '<end_content>':
                if current_file:
                    files[current_file] = '\n'.join(current_content).strip()
                current_file = None
                current_content = []
                state = 'expect_file'
            else:
                current_content.append(line)
    # Handle incomplete file definitions
    if state == 'collect_content' and current_file:
        files[current_file] = '\n'.join(current_content).strip()
    return files

def generate_feature_descriptions(client: Groq, model_name: str, features: List[str], max_retries: int = 5) -> Dict[str, str]:
    """