Update app.py

app.py

@@ -1,61 +1,65 @@
+import os
 import sys
-from types import ModuleType
+import subprocess  # <--- 确保这行在这里！
+import importlib
+import site
+import time
 
-# --- 🧪 终极暴力：内存级伪造 diso 包 ---
+# --- 🧪 1. 内存级伪造 diso (必须在任何业务 import 之前) ---
 def mock_diso():
+    from types import ModuleType
     print("🧪 Creating emergency mock for diso...")
-    
-    # 1. 创建基础模块对象
     diso = ModuleType("diso")
-    # 2. 伪造 DiffDMC 类
-    # PartCrafter 调用它通常是为了从网格提取几何，我们给它一个空的实现
-    # 如果后续运行报错缺少某个方法，我们再补。通常它是被导入但未被深度调用。
     class FakeDiffDMC:
         def __init__(self, *args, **kwargs): pass
         def __call__(self, *args, **kwargs): return None
-
     diso.DiffDMC = FakeDiffDMC
-    
-    # 3. 注入系统路径，防止其他地方再次触发 import
     sys.modules["diso"] = diso
-    # 针对 diso._C 的报错，也伪造一个空模块
     sys.modules["diso._C"] = ModuleType("diso._C")
-    
+    sys.modules["diso.diso_native"] = ModuleType("diso.diso_native")
     print("✅ diso has been mocked successfully!")
 
 mock_diso()
 
-# --- 正常的安装流程 (保留已成功的 scatter/sparse) ---
+# --- 🚀 2. 极速环境安装 (已经成功的 scatter/sparse) ---
 def install_essential_packages():
-    # scatter 等包你已经安装成功了，这里只做一次性快速确认
+    print("📦 Checking core dependencies...")
+    # 确保基础环境正确
+    subprocess.run([sys.executable, "-m", "pip", "install", "ninja", "setuptools", "wheel", "-q"])
+    
+    # 极速安装 PyG 扩展
     subprocess.run([
         sys.executable, "-m", "pip", "install", 
         "torch-scatter", "torch-sparse", "torch-cluster",
-        "-f", "https://data.pyg.org/whl/torch-2.4.0+cu121.html"
+        "-f", "https://data.pyg.org/whl/torch-2.4.0+cu121.html",
+        "--no-cache-dir"
+    ])
+    
+    # 安装剩下的渲染工具
+    subprocess.run([
+        sys.executable, "-m", "pip", "install", 
+        "pyrender", "pyopengl==3.1.0", "pyyaml", "trimesh", "accelerate", "-q"
     ])
-    # 安装剩下的轻量级包
-    subprocess.run([sys.executable, "-m", "pip", "install", "pyrender", "pyopengl==3.1.0", "pyyaml", "-q"])
+    
+    importlib.invalidate_caches()
+    site.main()
+    print("🎉 Environment Installation Phase Finished.")
 
 install_essential_packages()
 
-# --- 现在可以安全地 import 业务代码了 ---
-from src.pipelines.pipeline_partcrafter import PartCrafterPipeline
-# ... 后续代码 ...
-
+# --- 3. 正式导入业务逻辑 (现在开始这几百行代码就不会报错了) ---
 import spaces
 import gradio as gr
-import os
-# import sys
-import subprocess
-import importlib
-import site
-import time
-from glob import glob
-from typing import Any, Union
 import numpy as np
 import torch
 import uuid
 import shutil
+from huggingface_hub import snapshot_download
+from PIL import Image
+from accelerate.utils import set_seed
+
+# 从这里往下，粘贴你原本所有的业务逻辑代码 (PartCrafterPipeline 等)
+# ...
 
 # --- 🚀 核心修复：强制版本回退以避开编译 ---
 def pre_install_check():