let5see/AI-Video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化文件保存逻辑

Browse Source
This commit is contained in:
puke
2025-10-28 21:40:51 +08:00
parent d9d501d849
commit bbfc5af35f
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
reelforge/utils/os_util.py
View File

@@ -184,10 +184,10 @@ def create_task_output_dir(task_id: Optional[str] = None) -> Tuple[str, str]:
output/{task_id}/ output/{task_id}/
├── final.mp4 # Final video output ├── final.mp4 # Final video output
├── frames/ # All frame-related files ├── frames/ # All frame-related files
│ ├── 0_audio.mp3 │ ├── 01_audio.mp3
│ ├── 0_image.png │ ├── 01_image.png
│ ├── 0_composed.png │ ├── 01_composed.png
│ ├── 0_segment.mp4 │ ├── 01_segment.mp4
│ └── ... │ └── ...
└── metadata.json # Optional: task metadata └── metadata.json # Optional: task metadata
@@ -245,7 +245,7 @@ def get_task_frame_path(
Args: Args:
task_id: Task ID task_id: Task ID
frame_index: Frame index (0-based) frame_index: Frame index (0-based internally, but filename starts from 01)
file_type: File type (audio/image/composed/segment) file_type: File type (audio/image/composed/segment)
Returns: Returns:
@@ -253,7 +253,7 @@ def get_task_frame_path(
Example: Example:
>>> get_task_frame_path("20251028_143052_ab3d", 0, "audio") >>> get_task_frame_path("20251028_143052_ab3d", 0, "audio")
>>> # Returns: ".../output/20251028_143052_ab3d/frames/0_audio.mp3" >>> # Returns: ".../output/20251028_143052_ab3d/frames/01_audio.mp3"
""" """
ext_map = { ext_map = {
"audio": "mp3", "audio": "mp3",
@@ -262,7 +262,8 @@ def get_task_frame_path(
"segment": "mp4" "segment": "mp4"
} }
filename = f"{frame_index}_{file_type}.{ext_map[file_type]}" # Frame number starts from 01 for better human readability
filename = f"{frame_index + 1:02d}_{file_type}.{ext_map[file_type]}"
return get_task_path(task_id, "frames", filename) return get_task_path(task_id, "frames", filename)