Add application file

app.py

@@ -7,54 +7,73 @@ def greet(n):
     print(zero.device) # <-- 'cuda:0' 🤗
     return f"Hello {zero + n} Tensor"
 
+import gradio as gr
+import spaces
 
 lighting_js = """
 <script>
-function addLightingToModel3D() {
-    const model3dCanvas = document.querySelector('gradio-app model-3d canvas');
-    if (!model3dCanvas) {
-        console.error('Model3D canvas not found');
-        return;
-    }
-    const model3dElement = model3dCanvas.closest('model-3d');
-    if (model3dElement && model3dElement.__vue__) {
-        const vueInstance = model3dElement.__vue__;
-        const scene = vueInstance.scene;
-        if (!scene) {
-            console.error('Three.js scene not accessible');
+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;
         }
-        scene.children = scene.children.filter(child => !(child.isLight));
-        const ambientLight = new THREE.AmbientLight(0xffffff, 0.8);
+        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.5);
+        // 添加平行光
+        const directionalLight = new THREE.DirectionalLight(0xffffff, 0.8); // 高强度平行光
         directionalLight.position.set(1, 1, 1).normalize();
         scene.add(directionalLight);
-        console.log('Lighting added to Model3D');
-    } else {
-        console.error('Vue instance or scene not found');
+        console.log('Enhanced lighting added to model2');
+    } catch (error) {
+        console.error('Error adding lighting to model2:', error.message);
     }
 }
 document.addEventListener('DOMContentLoaded', () => {
-    setTimeout(addLightingToModel3D, 1000);
+    setTimeout(addLightingToModel2, 2000); // 延迟确保加载
 });
 </script>
 """
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 3D Model Viewer")
-    gr.HTML(lighting_js)  # 注入光照脚本
+    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(
         label="3D Model Viewer 1",
-        # value="https://modelviewer.dev/shared-assets/models/Astronaut.glb",  # 示例模型
-        clear_color=(0.9, 0.9, 0.9, 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
     )
+    
+    # model2（提高亮度）
+    model_input2 = gr.File(label="Upload 3D Model for Viewer 2 (.glb, .gltf)")
     model2 = gr.Model3D(
         label="3D Model Viewer2",
-        # value="https://modelviewer.dev/shared-assets/models/Astronaut.glb",  # 示例模型
-        clear_color=(0.1, 0.1, 0.1, 1)  # 先设置亮色背景提升基础亮度
+        clear_color=(0.9, 0.9, 0.9, 1),  # 改为浅色背景，辅助提亮
     )
-    
-    
+    model_input2.change(
+        lambda x: gr.Model3D(value=x.name),
+        inputs=model_input2,
+        outputs=model2
+    )
+
+
 if __name__ == "__main__":
     demo.launch()