Update app.py

app.py

@@ -64,48 +64,45 @@ def check_for_wav_in_outputs():
     else:
         return None
 
-def infer(text):
+def infer(prompt_in, chords, duration, bpms):
 
     # check if 'outputs' dir exists and empty it if necessary
     check_outputs_folder('./example_1')
     
     # set hparams
     output_dir = 'example_1' ### change this output directory
-    
-    
-    duration = 30
+        
+    duration = duration
     num_samples = 1
     bs = 1
-    
-    
+        
     # load your model
     musicgen = audiocraft.models.MusicGen.get_pretrained('./ckpt/musicongen') ### change this path
     musicgen.set_generation_params(duration=duration, extend_stride=duration//2, top_k = 250)
     
     
-    chords = ['C G A:min F']
+    chords = [def infer(prompt_in, chords):]
     
-    descriptions = ["A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums."] * num_samples
+    descriptions = [prompt_in] * num_samples
     
-    bpms = [120] * num_samples
+    bpms = [bpms] * num_samples
     
     meters = [4] * num_samples
     
     wav = []
     for i in range(num_samples//bs):
-      print(f"starting {i} batch...")
-      temp = musicgen.generate_with_chords_and_beats(descriptions[i*bs:(i+1)*bs], 
-                                                      chords[i*bs:(i+1)*bs],
-                                                      bpms[i*bs:(i+1)*bs], 
-                                                      meters[i*bs:(i+1)*bs]
-                                                      )
-      wav.extend(temp.cpu())
+        print(f"starting {i} batch...")
+        temp = musicgen.generate_with_chords_and_beats(descriptions[i*bs:(i+1)*bs], 
+                                                          chords[i*bs:(i+1)*bs],
+                                                          bpms[i*bs:(i+1)*bs], 
+                                                          meters[i*bs:(i+1)*bs]
+                                                          )
+        wav.extend(temp.cpu())
     
     # save and display generated audio
     for idx, one_wav in enumerate(wav):
-      
-      sav_path = os.path.join('./output_samples', output_dir, chords[idx] + "|" + descriptions[idx]).replace(" ", "_")
-      audio_write(sav_path, one_wav.cpu(), musicgen.sample_rate, strategy='loudness', loudness_compressor=True)
+        sav_path = os.path.join('./output_samples', output_dir, chords[idx] + "|" + descriptions[idx]).replace(" ", "_")
+        audio_write(sav_path, one_wav.cpu(), musicgen.sample_rate, strategy='loudness', loudness_compressor=True)
 
     # Print the outputs directory contents
     print_directory_contents('./output_samples')
@@ -113,17 +110,40 @@ def infer(text):
     print(wav_file_path)
     return wav_file_path
 
-with gr.Blocks() as demo:
-    with gr.Column():
+css="""
+#col-container{
+    max-width: 800px;
+    margin: 0 auto;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
         gr.Markdown("#MusiConGen")
         with gr.Row():
             with gr.Column():
-                text_in = gr.Textbox()
+                prompt_in = gr.Textbox(label="Music description", value="A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums.")
+                chords = gr.Texxtbox(label="Chords progression", value='C G A:min F')
+                duration = gr.Slider(label="Sample duration", minimum=4, maximum=30, step=1, value=20)
+                bpms = gr.Slider(label="BPMs", minimum=50, maximum=220, step=1, value=120)
                 submit_btn = gr.Button("Submit")
             wav_out = gr.Audio(label="Wav Result")
+        gr.Examples(
+            label = "Audio description examples",
+            examples = [
+                ["A laid-back blues shuffle with a relaxed tempo, warm guitar tones, and a comfortable groove, perfect for a slow dance or a night in. Instruments: electric guitar, bass, drums."],
+                ["A smooth acid jazz track with a laid-back groove, silky electric piano, and a cool bass, providing a modern take on jazz. Instruments: electric piano, bass, drums."],
+                ["A classic rock n' roll tune with catchy guitar riffs, driving drums, and a pulsating bass line, reminiscent of the golden era of rock. Instruments: electric guitar, bass, drums."],
+                ["A high-energy funk tune with slap bass, rhythmic guitar riffs, and a tight horn section, guaranteed to get you grooving. Instruments: bass, guitar, trumpet, saxophone, drums."],
+                ["A heavy metal onslaught with double kick drum madness, aggressive guitar riffs, and an unrelenting bass, embodying the spirit of metal. Instruments: electric guitar, bass guitar, drums."]
+            ],
+            inputs = [prompt_in]
+        )
+        
+        
     submit_btn.click(
         fn = infer,
-        inputs = [text_in],
+        inputs = [prompt_in, chords, duration, bpms],
         outputs = [wav_out]
     )
 demo.launch()