fix filename output

app.py

@@ -66,7 +66,7 @@ def get_args(parser: argparse.ArgumentParser):
         "-template",
         type=bool,
         default=True,
-        help="whether to generate by features",
+        help="whether to generate by template",
     )
     return parser.parse_args()
 
@@ -121,7 +121,7 @@ def generate_music(
     temperature = args.temperature
     seed = args.seed
     show_control_code = args.show_control_code
-    use_template = args.template
+    fname_prefix = emo if args.template else "Melody"
     print(" Hyper parms ".center(60, "#"), "\n")
     args_dict: dict = vars(args)
     for arg in args_dict.keys():
@@ -130,7 +130,7 @@ 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" if use_template else "T:Melody Fragment\n"
+        title = f"T:{fname_prefix} 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)
@@ -174,7 +174,7 @@ def generate_music(
             tokens = None
             if predicted_patch[0] != patchilizer.eos_token_id:
                 next_bar = patchilizer.decode([predicted_patch])
-                if show_control_code or next_bar[:2] not in ["S:", "B:", "E:"]:
+                if show_control_code or next_bar[:2] not in ["S:", "B:", "E:", "D:"]:
                     print(next_bar, end="")
                     tune += next_bar
 
@@ -243,8 +243,8 @@ def generate_music(
                     fix_pitch,
                 )
                 tunes = tunes.replace(title + title, title)
-                os.rename(xml, f"{outdir}/[{emo}]{timestamp}.musicxml")
-                xml = f"{outdir}/[{emo}]{timestamp}.musicxml"
+                os.rename(xml, f"{outdir}/[{fname_prefix}]{timestamp}.musicxml")
+                xml = f"{outdir}/[{fname_prefix}]{timestamp}.musicxml"
 
         else:
             if mode == "minor":
@@ -258,11 +258,11 @@ def generate_music(
                     offset,
                 )
                 tunes = tunes.replace(title + title, title)
-                os.rename(xml, f"{outdir}/[{emo}]{timestamp}.musicxml")
-                xml = f"{outdir}/[{emo}]{timestamp}.musicxml"
+                os.rename(xml, f"{outdir}/[{fname_prefix}]{timestamp}.musicxml")
+                xml = f"{outdir}/[{fname_prefix}]{timestamp}.musicxml"
 
             else:
-                xml = abc2xml(tunes, f"{outdir}/[{emo}]{timestamp}.musicxml")
+                xml = abc2xml(tunes, f"{outdir}/[{fname_prefix}]{timestamp}.musicxml")
 
         audio = xml2(xml, "wav")
         if fix_volume != None:
@@ -413,7 +413,7 @@ if __name__ == "__main__":
         with gr.Row():
             with gr.Column():
                 gr.Video(
-                    "./tutorial.mp4",
+                    "./demo.mp4",
                     label="Video demo",
                     show_download_button=False,
                     show_share_button=False,
@@ -445,7 +445,7 @@ if __name__ == "__main__":
                     label="Generate chords (coming soon)",
                     value=False,
                 )
-                gen_btn = gr.Button("Generate")
+                gen_btn_1 = gr.Button("Generate")
                 gr.Markdown("# Generate by feature control")
                 std_option = gr.Radio(["Low", "High"], label="Pitch SD", value="High")
                 mode_option = gr.Radio(["Minor", "Major"], label="Mode", value="Major")
@@ -516,7 +516,7 @@ if __name__ == "__main__":
                     allow_flagging="never",
                 )
 
-        gen_btn.click(
+        gen_btn_1.click(
             fn=infer_by_template,
             inputs=[dataset_option, valence_radio, arousal_radio, chord_check],
             outputs=[