better code

main.py

@@ -1,6 +1,4 @@
 import flet as ft
-from flet import TextField, ElevatedButton
-from flet_core import ControlEvent
 import flet_fastapi
 
 async def main(page: ft.Page) -> None:
@@ -10,42 +8,37 @@ async def main(page: ft.Page) -> None:
         title=ft.Text("This doesn't work yet!"), on_dismiss=lambda e: print("Dialog dismissed!")
     )
 
-    async def open_dlg(e: ControlEvent):
-        page.dialog = dlg
+    await page.add_async(dlg)
+
+    async def open_dlg(e: ft.ControlEvent) -> None:
         dlg.open = True
         await page.update_async()
 
-    question: TextField = TextField(label="Question", text_align=ft.TextAlign.LEFT, width=200)
-    answer: TextField = TextField(label="Answer", text_align=ft.TextAlign.LEFT, width=200)
+    question = ft.TextField(label="Question", text_align=ft.TextAlign.LEFT, width=200)
+    answer = ft.TextField(label="Answer", text_align=ft.TextAlign.LEFT, width=200)
     
-    button_submit: ElevatedButton = ElevatedButton(text="Send", disabled=True)
+    submit_button = ft.ElevatedButton(text="Send", disabled=True)
 
-    async def validate(e: ControlEvent) -> None:
+    async def validate(e: ft.ControlEvent) -> None:
         if all((question.value, answer.value)):
-            button_submit.disabled = False
+            submit_button.disabled = False
         else:
-            button_submit.disabled = True
+            submit_button.disabled = True
 
         await page.update_async()
     
-    async def submit(e: ControlEvent) -> None:
+    async def submit(e: ft.ControlEvent) -> None:
         await open_dlg(e)
     
     question.on_change = validate
     answer.on_change = validate
-    button_submit.on_click = submit
-
+    submit_button.on_click = submit
 
-    await page.add_async(
-        ft.Container(question, alignment=ft.alignment.center)
-    )
-
-    await page.add_async(
-        ft.Container(answer, alignment=ft.alignment.center)
-    )
 
     await page.add_async(
-        ft.Container(button_submit, alignment=ft.alignment.center)
+        ft.Container(question, alignment=ft.alignment.center),
+        ft.Container(answer, alignment=ft.alignment.center),
+        ft.Container(submit_button, alignment=ft.alignment.center)
     )
 
 app = flet_fastapi.app(main) # type: ignore