Update app.py

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))
 
 
 ###############################################
@@ -168,7 +168,7 @@ def generate_riffuse(prompt_start, steps, num_iterations, feel, prompt_end=None,
 
     if prompt_end is None:
       prompt_end = prompt_start
-    if seed_start is None:
+    if seed_start is 0:
       seed_start = random.randint(0,4294967295)
     if seed_end is None:
       seed_end = seed_start
@@ -188,16 +188,17 @@ 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)
+    # return wav_list_to_wav(wav_list)
 
-    # 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_list_to_wav(wav_list))
 
-    # 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):
@@ -232,8 +233,11 @@ def on_submit(prompt_1, prompt_2, steps, num_iterations, feel, seed):
 def on_num_iterations_change(n, prompt_2):
     if n is None:
         return gr.update(value="")
-    x = 5 if prompt_2 != "" else 2.5
-    total_length = x + x * n
+
+    if prompt_2 != "":
+        total_length = 5 * n
+    else:
+        total_length = 2.5 + 2.5 * n
     return gr.update(value=f"Total length: {total_length:.2f} seconds")
 
 with gr.Blocks() as app:
@@ -257,11 +261,10 @@ with gr.Blocks() as app:
             info = gr.Markdown()
             btn_generate = gr.Button(value="Generate")
         with gr.Column():
-            # video = gr.Video()
-            audio = gr.Audio()
+            video = gr.Video()
 
     inputs = [prompt_1, prompt_2, steps, num_iterations, feel, seed]
-    outputs = [audio, info]
+    outputs = [video, info]
 
     num_iterations.change(on_num_iterations_change, [num_iterations, prompt_2], [info])
     prompt_1.submit(on_submit, inputs, outputs)
@@ -274,13 +277,13 @@ with gr.Blocks() as app:
             ["synthwave", "jazz", "agile", 10],
             ["rap battle freestyle", "", "og_beat", 10],
             ["techno club banger", "", "og_beat", 10],
+            ["reggae dub beat", "sunset chill", "og_beat", 10],
             ["acoustic folk ballad", "", "agile", 10],
             ["blues guitar riff", "", "agile", 5],
             ["jazzy trumpet solo", "", "og_beat", 5],
             ["classical symphony orchestra", "", "vibes", 10],
             ["rock and roll power chord", "", "motorway", 5],
             ["soulful R&B love song", "", "marim", 10],
-            ["reggae dub beat", "sunset chill", "og_beat", 10],
             ["country western twangy guitar", "", "agile", 10]],
         inputs=[prompt_1, prompt_2, feel, num_iterations])