Function main

python/examples/drawing/demo.py:20–55

(file_path: str | None)

async def main(file_path: str | None):
    env_vals = dotenv_values()
    model = create_language_model(env_vals)
    validator = TypeChatValidator(schema.Drawing)
    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):
        print("[Sending request...]")
        history.append(request)
        result: Success[schema.Drawing] | Failure = await translator.translate("\n".join(history))
        if isinstance(result, Failure):
            print("Failure:", result.message)
        else:
            value: schema.Drawing = result.value
            print(value)
            if any(isinstance(item, schema.UnknownText) for item in value.items):
                print("Unknown text detected. Please provide more context:")
                for item in value.items:
                    if isinstance(item, schema.UnknownText):
                        print(" ", item.text)

            canvas.delete("all")
            render_drawing(canvas, value)
            print("Click in drawing to continue...")
            window.mainloop()

    await process_requests("~> ", file_path, request_handler)


if __name__ == "__main__":