Add application file

app.py

@@ -1,79 +1,35 @@
 import gradio as gr
 import spaces
-# from gradio_litmodel3d import LitModel3D
+
+# 只对 id=model2 的组件下 canvas 生效
+lighting_css = """
+<style>
+#model2 canvas {
+    filter: brightness(2.5) !important;
+}
+</style>
+"""
 
 @spaces.GPU
 def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
-import gradio as gr
-import spaces
-
-lighting_js = """
-<script>
-function addLightingToModel2() {
-    try {
-        // 仅选择第二个 model-3d（model2）
-        const model2Canvas = document.querySelector('gradio-app model-3d:nth-child(2) canvas');
-        if (!model2Canvas) {
-            console.error('Model2 canvas not found');
-            return;
-        }
-        const model3dElement = model2Canvas.closest('model-3d');
-        if (!model3dElement || !model3dElement.__vue__?.scene) {
-            console.error('Vue instance or scene not found for model2');
-            return;
-        }
-        const scene = model3dElement.__vue__.scene;
-        // 移除现有光源（可选，保持默认光照可跳过）
-        scene.children = scene.children.filter(child => !child.isLight);
-        // 添加环境光
-        const ambientLight = new THREE.AmbientLight(0xffffff, 1.2); // 高强度环境光
-        scene.add(ambientLight);
-        // 添加平行光
-        const directionalLight = new THREE.DirectionalLight(0xffffff, 0.8); // 高强度平行光
-        directionalLight.position.set(1, 1, 1).normalize();
-        scene.add(directionalLight);
-        console.log('Enhanced lighting added to model2');
-    } catch (error) {
-        console.error('Error adding lighting to model2:', error.message);
-    }
-}
-document.addEventListener('DOMContentLoaded', () => {
-    setTimeout(addLightingToModel2, 2000); // 延迟确保加载
-});
-</script>
-"""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 3D Model Viewer\nUpload .glb or .gltf files to preview models.")
-    gr.HTML(lighting_js)
-    
-    # model1（保持默认亮度）
-    model_input = gr.File(label="Upload 3D Model for Viewer 1 (.glb, .gltf)")
-    model = gr.Model3D(
+    gr.Markdown("## 3D Model Viewer")
+    gr.HTML(lighting_css)  # 注入样式
+
+    model1 = gr.Model3D(
         label="3D Model Viewer 1",
-        clear_color=(0.9, 0.9, 0.9, 1)  # 保持原背景
-    )
-    model_input.change(
-        lambda x: gr.Model3D(value=x.name),
-        inputs=model_input,
-        outputs=model
+        clear_color=(0.9, 0.9, 0.9, 1),
+        elem_id="model1"
     )
-    
-    # model2（提高亮度）
-    model_input2 = gr.File(label="Upload 3D Model for Viewer 2 (.glb, .gltf)")
+
     model2 = gr.Model3D(
-        label="3D Model Viewer2",
-        clear_color=(0.9, 0.9, 0.9, 1),  # 改为浅色背景，辅助提亮
-    )
-    model_input2.change(
-        lambda x: gr.Model3D(value=x.name),
-        inputs=model_input2,
-        outputs=model2
+        label="3D Model Viewer 2 (Brighter)",
+        clear_color=(0.9, 0.9, 0.9, 1),
+        elem_id="model2"
     )
 
-
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()