add config

AI-Hobbyist · May 11, 2024 · 1f00ead · 1f00ead
1 parent e28aef2
commit 1f00ead
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 12 deletions.
diff --git a/README.md b/README.md
@@ -41,7 +41,7 @@ Use `audio_processor.py`：
 ```bash
     "-i", "--input_dir", Input directory containing audio files
     "-o", "--output_dir", Output directory to save processed audio files
-    "target_loudness", default=-23, Target loudness
+    "-target", "--target_loudness", default=-23, Target loudness
     "-type", "--loudness_type", default="LUFS", choices=["LUFS", "dBFS", "Peak_dBFS", "RMSdB"], Type of loudness to match
     "--export_format", default="wav", choices=["wav", "flac", "mp3"], Audio export format
     "--mp3_bitrate", default=320, choices=[128, 192, 256, 320], MP3 bitrate in kbps
@@ -53,7 +53,7 @@ Use `audio_processor.py`：
 Example:
 
 ```bash
-python audio_processor.py -i input -o output -23 -type LUFS -export_format mp3 -mp3_bitrate 320
+python audio_processor.py -i input -o output -target -23 -type LUFS --export_format mp3 --mp3_bitrate 320
 ```
 
 ## Notes

diff --git a/README_zh_CN.md b/README_zh_CN.md
@@ -41,7 +41,7 @@ python gui.py
 ```bash
     "-i", "--input_dir", Input directory containing audio files
     "-o", "--output_dir", Output directory to save processed audio files
-    "target_loudness", default=-23, Target loudness
+    "-target", "--target_loudness", default=-23, Target loudness
     "-type", "--loudness_type", default="LUFS", choices=["LUFS", "dBFS", "Peak_dBFS", "RMSdB"], Type of loudness to match
     "--export_format", default="wav", choices=["wav", "flac", "mp3"], Audio export format
     "--mp3_bitrate", default=320, choices=[128, 192, 256, 320], MP3 bitrate in kbps
@@ -53,7 +53,7 @@ python gui.py
 例如：
 
 ```bash
-python audio_processor.py -i input -o output -23 -type LUFS -export_format mp3 -mp3_bitrate 320
+python audio_processor.py -i input -o output -target -23 -type LUFS --export_format mp3 --mp3_bitrate 320
 ```
 
 ## 一些说明

diff --git a/audio_processor.py b/audio_processor.py
@@ -164,13 +164,13 @@ def process_audio(input_dir, output_dir, target_loudness, loudness_type="LUFS",
     shutil.rmtree(temp_dir)
 
 
-def main():
+if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Audio processing API")
     parser.add_argument("-i", "--input_dir", type=str, required=True,
                         help="Input directory containing audio files")
     parser.add_argument("-o", "--output_dir", type=str, required=True,
                         help="Output directory to save processed audio files")
-    parser.add_argument("target_loudness", type=float,
+    parser.add_argument("-target", "--target_loudness", type=float,
                         default=-23, help="Target loudness  (default: -23)")
     parser.add_argument("-type", "--loudness_type", type=str, default="LUFS", choices=["LUFS", "dBFS", "Peak_dBFS", "RMSdB"],
                         help="Type of loudness to match (default: LUFS)")
@@ -189,8 +189,16 @@ def main():
         args.input_dir, args.output_dir, args.target_loudness, args.loudness_type, args.export_format, args.mp3_bitrate, args.ffmpeg_sample_rate, args.ffmpeg_bit_depth
     ))
 
-    with open('config.json', 'r') as config_file:
-        config = json.load(config_file)
+    try:
+        with open('config.json', 'r') as config_file:
+            config = json.load(config_file)
+    except FileNotFoundError:
+        config = {
+            "export_format": "wav",
+            "mp3_bitrate": 320,
+            "ffmpeg_sample_rate": 48000,
+            "ffmpeg_bit_depth": 32
+        }
 
     config["export_format"] = args.export_format
     config["mp3_bitrate"] = args.mp3_bitrate
@@ -211,7 +219,3 @@ def main():
 
     process_audio(args.input_dir, args.output_dir,
                   args.target_loudness, loudness_type)
-
-
-if __name__ == "__main__":
-    main()