[import flet_code_editor as fce

import flet as ft

CODE = """# 1
# 2
# 3
import json
import textwrap

print("Folding demo")
"""


def main(page: ft.Page):
    editor = fce.CodeEditor(
        language=fce.CodeLanguage.PYTHON,
        value=CODE,
        selection=ft.TextSelection(base_offset=41, extent_offset=62),
        autofocus=True,
        
        expand=True,
        on_selection_change=lambda e: print("Selection:", e),
    )

    async def fold_imports():
        await editor.fold_imports()

    async def fold_comment():
        await editor.fold_comment_at_line_zero()

    page.add(
        ft.Row(
            [
                ft.Button("Fold imports", on_click=fold_imports),
                ft.Button("Fold comment", on_click=fold_comment),
            ]
        ),
        editor,
    )


if __name__ == "__main__":
    ft.run(main)]