update

.gitignore

@@ -0,0 +1,79 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environments
+venv/
+.venv/
+env/
+.env/
+.anaconda
+.conda
+
+# IDE / System
+.idea/
+.vscode/
+*.swp
+.DS_Store
+Thumbs.db
+
+# UniSH specific outputs
+output.glb
+visualize.glb
+scene_clouds_per_frame/
+smpl_meshes_per_frame/
+output/
+outputs/
+results/
+tmp/
+temp/
+
+# Model weights and large files
+*.pt
+*.pth
+*.ckpt
+*.safetensors
+*.bin
+*.onnx
+*.pkl
+
+# SMPL models
+body_models/
+smpl_models/
+
+# Media and Datasets
+*.mp4
+*.avi
+*.mov
+*.mkv
+*.jpg
+*.png
+*.jpeg
+
+# 3D Intermediate files
+*.ply
+*.obj
+*.gltf
+
+# Runtime temporary directories
+_tmp_sam2/
+.gradio/
+flagged/

app.py

@@ -17,16 +17,10 @@ import html
 import base64
 import inspect
 
-# ==========================================
-# 1. 核心路径与环境配置
-# ==========================================
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 STATIC_DIR = os.path.join(BASE_DIR, "static")
 EXAMPLES_DIR = os.path.join(BASE_DIR, "examples")
 
-# ==========================================
-# 2. 静态资源准备
-# ==========================================
 def prepare_local_assets():
     os.makedirs(STATIC_DIR, exist_ok=True)
     base_url = "https://registry.npmmirror.com/three/0.160.0/files"
@@ -50,9 +44,6 @@ def prepare_local_assets():
 
 prepare_local_assets()
 
-# ==========================================
-# 3. 环境 Patch (SAM 2)
-# ==========================================
 def install_sam2():
     try:
         import sam2
@@ -72,7 +63,6 @@ install_sam2()
 
 sys.path.append(BASE_DIR)
 
-# 引入核心模型代码
 from unish.utils.inference_utils import (
     load_model, process_video, run_inference,
     generate_mixed_geometries_in_memory,
@@ -82,9 +72,6 @@ from unish.utils.inference_utils import (
 MODEL = None
 BODY_MODELS_PATH = "body_models/"
 
-# ==========================================
-# 4. 辅助函数
-# ==========================================
 def download_smpl_assets(body_models_path):
     if 'smpl' not in body_models_path:
         model_path = os.path.join(body_models_path, 'smpl')
@@ -112,7 +99,6 @@ def pack_sequence_to_glb(base_dir, output_path, start_frame, end_frame, scene_ra
     scene_cloud_dir = os.path.join(base_dir, "scene_clouds_per_frame")
     smpl_mesh_dir = os.path.join(base_dir, "smpl_meshes_per_frame")
     
-    # 60,000 点/帧
     MAX_POINTS_PER_FRAME = 60000 
 
     for i in range(start_frame, end_frame):
@@ -171,10 +157,6 @@ def get_video_duration(video_path):
     except:
         return 10.0
 
-# ==========================================
-# 5. 前端逻辑: Player & Loading HTML
-# ==========================================
-
 def get_loading_html(message="Processing..."):
     return f"""
     <div style="height: 600px; width: 100%; background: #f9fafb; border-radius: 12px; border: 1px solid #e5e7eb; display: flex; flex-direction: column; align-items: center; justify-content: center; font-family: sans-serif; color: #4b5563;">
@@ -290,9 +272,6 @@ def get_player_html(abs_glb_path):
     """
     return f'<iframe srcdoc="{html.escape(raw_html)}" width="100%" height="600px" style="border:none; border-radius: 12px;"></iframe>'
 
-# ==========================================
-# 6. 推理逻辑封装
-# ==========================================
 if os.environ.get("SPACE_ID"):
     from spaces import GPU
     gpu_decorator = GPU(duration=120)
@@ -307,7 +286,6 @@ else:
 
 @gpu_decorator
 def predict(video_path, start_time, end_time):
-    # 1. 时长限制检测
     duration_input = end_time - start_time
     if duration_input > 10.0:
         raise gr.Error(f"Video limit exceeded ({duration_input:.1f}s). Please keep it under 10 seconds.")
@@ -315,7 +293,6 @@ def predict(video_path, start_time, end_time):
     if start_time >= end_time:
         raise gr.Error("Error: End time must be greater than Start time.")
     
-    # 2. 正常流程
     yield get_loading_html("Processing...")
     
     output_dir = tempfile.mkdtemp()
@@ -355,10 +332,6 @@ def predict(video_path, start_time, end_time):
         
     yield get_player_html(tmp_glb_path)
 
-# ==========================================
-# 7. UI 构建
-# ==========================================
-
 examples_list = []
 if os.path.exists(EXAMPLES_DIR):
     examples_list = [[os.path.join("examples", f)] for f in os.listdir(EXAMPLES_DIR) if f.endswith(".mp4")]
@@ -412,7 +385,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", radius_size="md"), css
                     start_time = gr.Slider(minimum=0, maximum=10, value=0, step=0.01, label="Start Time (s)")
                     end_time = gr.Slider(minimum=0, maximum=10, value=3, step=0.01, label="End Time (s)")
             
-            # --- Grid Layout 还原完整文案 ---
             gr.HTML("""
             <div class="alert-box tip-box">
                 <strong>💡 Use Tips:</strong>

index.html

@@ -15,7 +15,7 @@
   <link href="https://fonts.googleapis.com/css?family=Google+Sans|Noto+Sans|Castoro" rel="stylesheet">
 
   <style>
-    /* 1. 轮播图容器样式 */
+
     .carousel .item {
       position: relative;
       border-radius: 12px;
@@ -32,7 +32,6 @@
       height: auto;
     }
     
-    /* 2. 悬浮计数胶囊样式 */
     .video-overlay-label {
       position: absolute;
       bottom: 15px;
@@ -51,7 +50,6 @@
       z-index: 10;
     }
 
-    /* 3. Player 样式 */
     #canvas-container {
       width: 100%;
       height: 500px;
@@ -62,7 +60,6 @@
       box-shadow: inset 0 0 20px rgba(0,0,0,0.05);
     }
 
-    /* 4. Teaser 媒体容器样式 */
     .teaser-video-wrapper {
       position: relative;
       border-radius: 8px; 
@@ -71,7 +68,7 @@
       box-shadow: 0 8px 20px rgba(0,0,0,0.12);
       background: #000;
       width: 100%;
-      margin-bottom: 20px; /* 视频和图片之间的间距 */
+      margin-bottom: 20px;
     }
     
     .teaser-video-wrapper video {
@@ -80,13 +77,11 @@
       height: auto;
     }
 
-    /* 作者名字样式 */
     .author-block {
       margin-right: 10px;
       white-space: nowrap; 
     }
     
-    /* 自定义 Slider */
     input[type=range].custom-slider {
       -webkit-appearance: none;
       width: 100%;