修改 无pv不输出 修改 44100Hz

acb2mp3.py

@@ -15,9 +15,10 @@ def convert_awb_to_wav(awb, output_wav):
     run(["vgmstream-cli","-o",output_wav,awb])
 
 
+
 def convert_wav_to_mp3(wav, output_mp3):
-    """使用 ffmpeg 将 WAV 文件转换为 MP3 文件"""
-    run(["ffmpeg", "-i", wav, f"{output_mp3}"])
+    """使用 ffmpeg 将 WAV 文件转换为 44100Hz 的 MP3 文件"""
+    run(["ffmpeg", "-y", "-i", wav, "-ar", "44100", output_mp3])
 
 
 if __name__ == "__main__":

convert.py

@@ -11,35 +11,36 @@ from pv_decode import dat_to_mp4
 # 假设你已实现以下函数
 
 def build_maidata_txt(
-        title: str = "",
-        freemsg: str = "",
-        bpm: str = "",
-        first_notes: dict = None,  # {1: 0.123, 2: ..., ...}
-        levels: dict = None,  # {1: "3", 2: "5", ...}
-        designers: dict = None,  # {1: "作者A", 2: ..., ...}
-        charts: dict = None,  # {1: "谱面数据\n...", ...}，
-        levelnum:int = None,
+    title: str = "",
+    freemsg: str = "",
+    bpm: str = "",
+    first_notes: dict = None,  # {2: 0.123, 3: ..., ...}
+    levels: dict = None,       # {2: "3", 3: "5", ...}
+    designers: dict = None,    # {2: "作者A", 3: ..., ...}
+    charts: dict = None,       # {2: "谱面数据\n...", ...}
+    levelnum: int = None,
 ) -> str:
-    maidata = [f"&title={title}", f"&freemsg={freemsg}", f"&wholebpm={bpm}", "&first=0"]
+    # 限制 levelnum 在 [2, 6]
+    levelnum = max(2, min(levelnum, 6))
 
-    # 1~6 难度
-    for i in range(1, levelnum+1):
+    maidata = [
+        f"&title={title}",
+        f"&artist={freemsg}",
+        f"&wholebpm={bpm}",
+        "&first=0"
+    ]
+
+    for i in range(2, levelnum + 2):
         first = f"{first_notes.get(i):.3f}" if first_notes and i in first_notes else ""
-        lv = levels.get(i, "") if levels else ""
-        des = designers.get(i, "") if designers else ""
-        chart = charts.get(i, "") if charts else ""
-
         maidata.append(f"&first_{i}={first}")
 
-
-
-    for i in range(2, levelnum+1):
+    for i in range(2, levelnum + 2):
         lv = levels.get(i, "") if levels else ""
         des = designers.get(i, "") if designers else ""
         maidata.append(f"&lv_{i}={lv}")
         maidata.append(f"&des_{i}={des}")
 
-    for i in range(2, levelnum):
+    for i in range(2, levelnum + 2):
         chart = charts.get(i, "") if charts else ""
         maidata.append(f"&inote_{i}=")
         maidata.append(chart.strip())
@@ -50,6 +51,7 @@ def build_maidata_txt(
     return "\n".join(maidata)
 
 
+
 def convert_to_simai_folder(result,output_folder):
     npof = output_folder
     output_folder = Path(output_folder)
@@ -61,17 +63,24 @@ def convert_to_simai_folder(result,output_folder):
         os.makedirs(of)
     name = info[1]
     artist = info[2]
-    designers = {i + 1: item["designer"] for i, item in enumerate(info[3])}
-    levels = {i + 1: item["levelshow"] for i, item in enumerate(info[3])}
+    designers = {i + 2: item["designer"] for i, item in enumerate(info[3])}
+    levels = {i + 2: item["levelshow"] for i, item in enumerate(info[3])}
     ma2_list = result[1]
     ab_file = result[2]
     acb_list = result[3]
+    awb_file = next((f for f in acb_list if f.endswith('.awb')), None)
     dat_file = result[4]
     convert_results = {}
-    for mai in ma2_list:
-        convert_results.update({ma2_list.index(mai)+1: ma2tosimai(mai)})
+    for path in ma2_list:
+        filename = os.path.basename(path)
+        try:
+            num = int(filename[-6:-4])  # 提取 _00 → 0
+            level = num + 2  # 转换为 Simai 难度等级
+            convert_results[level] = ma2tosimai(path)
+        except Exception as e:
+            print(f"处理 {filename} 时出错: {e}")
 
-    convert_awb_to_wav(acb_list[1],f"work/{id}/temp.wav")
+    convert_awb_to_wav(awb_file,f"work/{id}/temp.wav")
     convert_wav_to_mp3(f"work/{id}/temp.wav",f"work/{id}/track.mp3")
 
     convert_ab_to_png(ab_file,f"work/{id}/bg.png")
@@ -102,6 +111,7 @@ def convert_to_simai_folder(result,output_folder):
         else:
             print(f"文件 {file_name} 不存在，跳过复制")
 
+
     if os.path.exists(source_folder):
         shutil.rmtree(source_folder)
 
@@ -119,8 +129,6 @@ def convert_to_simai_folder(result,output_folder):
 
 
 
-
-
 # 示例调用
 
 if __name__ == "__main__":

pv_decode.py

@@ -20,7 +20,24 @@ def convert_ivf_to_mp4(ivf_path: Path, output_mp4_path: Path):
         str(output_mp4_path)
     ], check=True)
 
-def dat_to_mp4(dat_file: str,id:str):
+import subprocess
+
+def get_video_duration(path: Path) -> float:
+    """使用 ffprobe 获取视频时长（单位：秒）"""
+    try:
+        result = subprocess.run(
+            ["ffprobe", "-v", "error", "-show_entries", "format=duration",
+             "-of", "default=noprint_wrappers=1:nokey=1", str(path)],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            text=True
+        )
+        return float(result.stdout.strip())
+    except Exception as e:
+        print(f"获取视频时长失败: {e}")
+        return 0.0
+
+def dat_to_mp4(dat_file: str, id: str):
     """ 将 .dat 文件当作 .usm 文件处理，提取并转换为 .mp4 """
     dat_path = Path(dat_file).resolve()
     base_name = dat_path.stem
@@ -41,10 +58,16 @@ def dat_to_mp4(dat_file: str,id:str):
     print(f"[2/3] 转换为 MP4 ...")
     convert_ivf_to_mp4(ivf_path, mp4_path)
 
+    duration = get_video_duration(mp4_path)
+    if duration < 1.0:
+        print(f"⚠️ 视频时长 {duration:.2f}s 太短，跳过生成 pv.mp4")
+        return None
+
     print(f"[3/3] 成功生成：{mp4_path}")
-    process_video(mp4_path, work_dir/"pv.mp4")
+    process_video(mp4_path, work_dir / "pv.mp4")
     return mp4_path
 
+
 # === 示例用法 ===
 if __name__ == "__main__":
     import sys

search.py

@@ -31,6 +31,7 @@ def search_music_by_id(search_id):
                 # ma2 文件
                 ma2_files = [f for f in os.listdir(sub_path) if f.endswith(".ma2")]
                 ma2_paths = [os.path.join(sub_path, f) for f in ma2_files]
+                ma2_paths.sort()
                 if ma2_files:
                     print("  MA2 文件：")
                     for f in ma2_files: