make_waveform is broken

app.py

@@ -123,11 +123,11 @@ def generate(prompt, steps, num_iterations, feel, seed):
     bg_image = outpaint(prompt, init_image, mask, steps)
     bg_image.save("bg_image.png")
 
-    # return read(wav_bytes)
-    with open("output.wav", "wb") as f:
-        f.write(wav_bytes.read())
+    return read(wav_bytes)
+    # with open("output.wav", "wb") as f:
+    #     f.write(wav_bytes.read())
 
-    return gr.make_waveform("output.wav", bg_image="bg_image.png", bar_count=int(duration_s*25))
+    # return gr.make_waveform("output.wav", bg_image="bg_image.png", bar_count=int(duration_s*25))
 
 
 ###############################################
@@ -188,16 +188,16 @@ def generate_riffuse(prompt_start, steps, num_iterations, feel, prompt_end=None,
         seed_start = seed_end
         seed_end = seed_start + 1
 
-    # return read(wav_bytes)
+    return read(wav_bytes)
 
-    mask = Image.new("RGB", (512, 512), color="white")
-    bg_image = outpaint(f"{prompt_start} and {prompt_end}", init_image, mask, steps)
-    bg_image.save("bg_image.png")
+    # mask = Image.new("RGB", (512, 512), color="white")
+    # bg_image = outpaint(f"{prompt_start} and {prompt_end}", init_image, mask, steps)
+    # bg_image.save("bg_image.png")
 
-    with open("output.wav", "wb") as f:
-        f.write(wav_bytes.read())
+    # with open("output.wav", "wb") as f:
+    #     f.write(wav_bytes.read())
 
-    return gr.make_waveform("output.wav", bg_image="bg_image.png")
+    # return gr.make_waveform("output.wav", bg_image="bg_image.png")
 
 
 def wav_list_to_wav(wav_list):
@@ -257,10 +257,11 @@ with gr.Blocks() as app:
             info = gr.Markdown()
             btn_generate = gr.Button(value="Generate")
         with gr.Column():
-            video = gr.Video()
+            # video = gr.Video()
+            audio = gr.Audio()
 
     inputs = [prompt_1, prompt_2, steps, num_iterations, feel, seed]
-    outputs = [video, info]
+    outputs = [audio, info]
 
     num_iterations.change(on_num_iterations_change, [num_iterations, prompt_2], [info])
     prompt_1.submit(on_submit, inputs, outputs)