From b2381397738b5a6066d7a70cadfb2db000958013 Mon Sep 17 00:00:00 2001
From: yinsx <yinsx@163.com>
Date: Mon, 5 Jan 2026 18:37:42 +0800
Subject: [PATCH] 1

---
 src/main.ts | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/main.ts b/src/main.ts
index c539c33..0ccba6a 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -13,7 +13,7 @@ declare global {
 }
 
 type InitParams = {
-    container?: string;
+    container?: string | HTMLCanvasElement;
     modelUrlList?: string[];
     apiConfig?: ConfiguratorParams;
     onSuccess?: () => void;
@@ -38,8 +38,15 @@ const kernel = {
         if (!params) { console.error('params is required'); return; }
 
         mainApp = new MainApp();
+
+        const container = (typeof params.container === 'string'
+            ? (document.querySelector(params.container) || document.getElementById(params.container))
+            : params.container || document.querySelector('#renderDom')) as HTMLCanvasElement | null;
+
+        if (!container) { throw new Error('Render canvas not found'); }
+
         mainApp.loadAConfig({
-            container: params.container || document.querySelector(params.container),
+            container,
             modelUrlList: params.modelUrlList || [],
             env: params.env
         });