Add application file

app.py

@@ -8,30 +8,40 @@ def greet(n):
     return f"Hello {zero + n} Tensor"
 
 
-with gr.Blocks() as demo:
-    model3d = LitModel3D(label="MESH", exposure=10, height=300)
-    model3d = gr.Model3D(label="MESH2", height=300)
-    model3d = gr.Model3D(label="GS2", height=300)
+import gradio as gr
 
+with gr.Blocks() as demo:
+    gr.Markdown("## 3D Model Viewer")
+    model = gr.Model3D(
+        value="https://modelviewer.dev/shared-assets/models/Astronaut.glb",  # 示例模型
+        clear_color=(0.9, 0.9, 0.9, 1)  # 先设置亮色背景提升基础亮度
+    )
+    
     demo.load(
-        _js="""
+        js="""
         () => {
-            const viewer = document.querySelector('.gr-model3d').viewer;
-            const scene = viewer.scene;
-            
-            // 添加环境光
-            const ambientLight = new THREE.AmbientLight(0xffffff, 1.5);
-            scene.add(ambientLight);
-            
-            // 添加方向光
-            const directionalLight = new THREE.DirectionalLight(0xffffff, 1.5);
-            directionalLight.position.set(5, 5, 5);
-            scene.add(directionalLight);
+            // 等待组件加载完成（避免选择器找不到元素）
+            const observer = new MutationObserver(() => {
+                const viewer = document.querySelector('[data-testid="model3d-component"] .model-viewer');
+                if (viewer) {
+                    const scene = viewer.modelView.scene;
+                    observer.disconnect();  // 找到后停止观察
+                
+                    // 添加环境光（白色，强度 1.5）
+                    const ambientLight = new THREE.AmbientLight(0xffffff, 1.5);
+                    scene.add(ambientLight);
+                    
+                    // 添加方向光（从右上方照射，模拟自然光）
+                    const directionalLight = new THREE.DirectionalLight(0xffffff, 1.8);
+                    directionalLight.position.set(8, 10, 5);  // 调整光源位置
+                    scene.add(directionalLight);
+                }
+            });
+            observer.observe(document.body, { childList: true, subtree: true });
         }
         """
     )
 
 
-
 if __name__ == "__main__":
     demo.launch()