Function parse_generated_content

pygen/generate_package.py:12–62

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

(content: str)

    return api_key

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.split('\n')
    for line_num, line in enumerate(lines, 1):
        stripped = line.strip()
        if state == 'expect_file':
            if stripped.startswith('<') and stripped.endswith('>'):
                current_file = stripped[1:-1].strip()
                if not current_file:
                    print(f"Warning: Empty file path at line {line_num}")
                    current_file = None
                    continue
                state = 'expect_begin_content'
            elif stripped == '':
                continue  # Skip empty lines
            else:
                print(f"Warning: Expected <file_path> at line {line_num}, got: {stripped}")
        elif state == 'expect_begin_content':
            if stripped == '<begin_content>':
                current_content = []
                state = 'collect_content'
            else:
                print(f"Warning: Expected <begin_content> after <{current_file}> at line {line_num}, got: {stripped}")
                current_file = None
                state = 'expect_file'
        elif state == 'collect_content':
            if stripped == '<end_content>':
                if current_file in files:
                    print(f"Warning: Duplicate file path '{current_file}' at line {line_num}. Overwriting previous content.")
                files[current_file] = '\n'.join(current_content).strip()
                current_file = None
                current_content = []
                state = 'expect_file'
            else:
                current_content.append(line)
    # Check if any file was not properly closed
    if state != 'expect_file':
        print(f"Warning: Incomplete file definition for {current_file}")
    return files

def generate_python_package(package_name: str, package_description: str, features: List[str]) -> Optional[Dict[str, str]]:
    """Generate a Python package structure and content using Google Gemini Pro 002 model."""