Update app.py

app.py

@@ -30,15 +30,50 @@ def print_directory_contents(path):
         for f in files:
             print(f"{subindent}{f}")
 
+def check_outputs_folder(folder_path):
+    # Check if the folder exists
+    if os.path.exists(folder_path) and os.path.isdir(folder_path):
+        # Delete all contents inside the folder
+        for filename in os.listdir(folder_path):
+            file_path = os.path.join(folder_path, filename)
+            try:
+                if os.path.isfile(file_path) or os.path.islink(file_path):
+                    os.unlink(file_path)  # Remove file or link
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path)  # Remove directory
+            except Exception as e:
+                print(f'Failed to delete {file_path}. Reason: {e}')
+    else:
+        print(f'The folder {folder_path} does not exist.')
+
+def check_for_wav_in_outputs():
+    # Define the path to the outputs folder
+    outputs_folder = './example_1'
+    
+    # Check if the outputs folder exists
+    if not os.path.exists(outputs_folder):
+        return None
+    
+    # Check if there is a .mp4 file in the outputs folder
+    mp4_files = [f for f in os.listdir(outputs_folder) if f.endswith('.wav')]
+    
+    # Return the path to the mp4 file if it exists
+    if mp4_files:
+        return os.path.join(outputs_folder, mp4_files[0])
+    else:
+        return None
 
 def infer(text):
+
+    # 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
-    num_samples = 5
+    num_samples = 1
     bs = 1
     
     
@@ -47,12 +82,7 @@ def infer(text):
     musicgen.set_generation_params(duration=duration, extend_stride=duration//2, top_k = 250)
     
     
-    chords = ['C G A:min F',
-              'A:min F C G',
-              'C F G F',
-              'C A:min F G',
-              'D:min G C A:min',
-              ]
+    chords = ['C G A:min F']
     
     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
     
@@ -78,18 +108,21 @@ def infer(text):
 
     # Print the outputs directory contents
     print_directory_contents('./output_samples')
-    return "done"
+    wav_file_path = check_for_wav_in_outputs()
+    print(wav_file_path)
+    return wav_file_path
 
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("#MusiConGen")
         with gr.Row():
-            text_in = gr.Textbox()
-            submit_btn = gr.Button("Submit")
-            text_out = gr.Textbox()
+            with gr.Column():
+                text_in = gr.Textbox()
+                submit_btn = gr.Button("Submit")
+            wav_out = gr.Audio(label="Wav Result")
     submit_btn.click(
         fn = infer,
         inputs = [text_in],
-        outputs = [text_out]
+        outputs = [wav_out]
     )
 demo.launch()