Reuse drawing window

python/examples/drawing/demo.py

@@ -1,5 +1,6 @@
 import asyncio
 import sys
+import tkinter as tk
 
 from dotenv import dotenv_values
 
@@ -23,6 +24,12 @@ async def main(file_path: str | None):
     translator = TypeChatJsonTranslator(model, validator, schema.Drawing)
     # print(translator._schema_str)
 
+    window = tk.Tk()
+    window.title("Click to continue...")
+    canvas = tk.Canvas(window, width=800, height=600, bg="white", highlightthickness=0)
+    canvas.pack(padx=10, pady=10)
+    canvas.bind("<Button-1>", lambda event: window.quit())
+
     history: list[str] = []
 
     async def request_handler(request: str):
@@ -39,7 +46,9 @@ async def request_handler(request: str):
                     if isinstance(item, schema.UnknownText):
                         print(" ", item.text)
 
-            render_drawing(value)
+            canvas.delete("all")
+            render_drawing(canvas, value)
+            window.mainloop()
 
     await process_requests("~> ", file_path, request_handler)
 

python/examples/drawing/render.py

@@ -11,13 +11,7 @@
 }
 
 
-def render_drawing(drawing: Drawing):
-    window = tk.Tk()
-    window.title("Drawing")
-    window.configure(bg="white")
-
-    canvas = tk.Canvas(window, width=800, height=600, bg="white", highlightthickness=0)
-    canvas.pack(padx=10, pady=10)
+def render_drawing(canvas: tk.Canvas, drawing: Drawing):
 
     def draw_box(box: Box):
         x1, y1 = box.x, box.y
@@ -69,8 +63,6 @@ def draw_arrow(arrow: Arrow):
             case UnknownText():
                 print(f"Unknown text: {item.text}")
 
-    window.mainloop()
-
 
 if __name__ == "__main__":
     example_drawing = Drawing(
@@ -105,4 +97,9 @@ def draw_arrow(arrow: Arrow):
         ],
     )
 
-    render_drawing(example_drawing)
+    window = tk.Tk()
+    window.title("Drawing")
+    canvas = tk.Canvas(window, width=800, height=600, bg="white", highlightthickness=0)
+    canvas.pack(padx=10, pady=10)
+    render_drawing(canvas, example_drawing)
+    window.mainloop()