Update app.py

app.py

@@ -97,14 +97,12 @@ def segment(mesh_path, output_dir, postprocess=True, postprocess_threshold=0.95,
     mesh_save = mesh.copy()
     mesh_save.visual.face_colors = face_colors
 
-    # 保存结果
-    input_filename = Path(mesh_path).stem
-    
-    seg_mesh_path = os.path.join(output_dir, f'{input_filename}_segmented.glb')
+    # 保存结果 (使用固定文件名)
+    seg_mesh_path = os.path.join(output_dir, 'output_segmented.glb')
     mesh_save.export(seg_mesh_path)
     print(f"  -> Saved segmented mesh to: {seg_mesh_path}")
 
-    face_id_path = os.path.join(output_dir, f'{input_filename}_face_ids.npy')
+    face_id_path = os.path.join(output_dir, 'output_face_ids.npy')
     np.save(face_id_path, face_ids)
     print(f"  -> Saved face IDs to: {face_id_path}")
 
@@ -131,12 +129,10 @@ def generate(mesh_path, aabb, output_dir, seed=42):
         **additional_params,
     )
 
-    # 导出所有结果
-    input_filename = Path(mesh_path).stem
-    
-    obj_path = os.path.join(output_dir, f'{input_filename}_gen_parts.glb')
-    out_bbox_path = os.path.join(output_dir, f'{input_filename}_gen_bbox.glb')
-    explode_path = os.path.join(output_dir, f'{input_filename}_gen_exploded.glb')
+    # 导出所有结果 (使用固定文件名)
+    obj_path = os.path.join(output_dir, 'output_gen_parts.glb')
+    out_bbox_path = os.path.join(output_dir, 'output_gen_bbox.glb')
+    explode_path = os.path.join(output_dir, 'output_gen_exploded.glb')
     
     obj_mesh.export(obj_path)
     print(f"  -> Saved generated parts to: {obj_path}")
@@ -173,7 +169,6 @@ def main():
         default='results',
         help="Directory to save the output files. (default: 'results')"
     )
-    # ... 其他参数保持不变 ...
     parser.add_argument(
         '--seed',
         type=int,