sync ms

app.py

@@ -62,6 +62,12 @@ def get_args(parser: argparse.ArgumentParser):
         default=False,
         help="whether to show control code",
     )
+    parser.add_argument(
+        "-template",
+        type=bool,
+        default=True,
+        help="whether to generate by features",
+    )
     return parser.parse_args()
 
 
@@ -115,6 +121,7 @@ def generate_music(
     temperature = args.temperature
     seed = args.seed
     show_control_code = args.show_control_code
+    use_template = args.template
     print(" Hyper parms ".center(60, "#"), "\n")
     args_dict: dict = vars(args)
     for arg in args_dict.keys():
@@ -123,14 +130,14 @@ def generate_music(
     print("\n", " Output tunes ".center(60, "#"))
     start_time = time.time()
     for i in range(num_tunes):
-        title = f"T:{emo} Fragment\n"
+        title = f"T:{emo} Fragment\n" if use_template else "T:Melody Fragment\n"
         artist = f"C:Generated by AI\n"
         tune = f"X:{str(i + 1)}\n{title}{artist}{prompt}"
         lines = re.split(r"(\n)", tune)
         tune = ""
         skip = False
         for line in lines:
-            if show_control_code or line[:2] not in ["S:", "B:", "E:"]:
+            if show_control_code or line[:2] not in ["S:", "B:", "E:", "D:"]:
                 if not skip:
                     print(line, end="")
                     tune += line
@@ -279,7 +286,7 @@ def generate_music(
         return generate_music(args, emo, weights)
 
 
-def inference(dataset: str, v: str, a: str, add_chord: bool):
+def infer_by_template(dataset: str, v: str, a: str, add_chord: bool):
     if os.path.exists(TEMP_DIR):
         shutil.rmtree(TEMP_DIR)
 
@@ -296,6 +303,7 @@ def inference(dataset: str, v: str, a: str, add_chord: bool):
 
     parser = argparse.ArgumentParser()
     args = get_args(parser)
+    args.template = True
     return generate_music(
         args,
         emo=emotion,
@@ -303,7 +311,7 @@ def inference(dataset: str, v: str, a: str, add_chord: bool):
     )
 
 
-def infer(
+def infer_by_features(
     dataset: str,
     pitch_std: str,
     mode: str,
@@ -328,6 +336,7 @@ def infer(
 
     parser = argparse.ArgumentParser()
     args = get_args(parser)
+    args.template = False
     return generate_music(
         args,
         emo=emotion,
@@ -405,7 +414,7 @@ if __name__ == "__main__":
             with gr.Column():
                 gr.Video(
                     "./tutorial.mp4",
-                    label="Tutorial",
+                    label="Video demo",
                     show_download_button=False,
                     show_share_button=False,
                 )
@@ -433,7 +442,7 @@ if __name__ == "__main__":
                     value="High",
                 )
                 chord_check = gr.Checkbox(
-                    label="Generate chords (Coming soon)",
+                    label="Generate chords (coming soon)",
                     value=False,
                 )
                 gen_btn = gr.Button("Generate")
@@ -462,7 +471,7 @@ if __name__ == "__main__":
                     label="Volume (dB)",
                 )
                 chord_check_2 = gr.Checkbox(
-                    label="Generate chords (Coming soon)",
+                    label="Generate chords (coming soon)",
                     value=False,
                 )
                 gen_btn_2 = gr.Button("Generate")
@@ -508,7 +517,7 @@ if __name__ == "__main__":
                 )
 
         gen_btn.click(
-            fn=inference,
+            fn=infer_by_template,
             inputs=[dataset_option, valence_radio, arousal_radio, chord_check],
             outputs=[
                 wav_audio,
@@ -522,7 +531,7 @@ if __name__ == "__main__":
         )
 
         gen_btn_2.click(
-            fn=infer,
+            fn=infer_by_features,
             inputs=[
                 dataset_option,
                 std_option,