update

app.py

@@ -26,8 +26,13 @@ def generate_waveform(description):
 
     generated_file_path = "./awesome.wav"
 
+    # if os.path.exists(generated_file_path):
+    #     return generated_file_path
+    # else:
+    #     return "Error: Failed to generate the waveform."
     if os.path.exists(generated_file_path):
-        return generated_file_path
+        waveform_video = gr.make_waveform(audio=generated_file_path, bg_color="#000000", bars_color="#00FF00", bar_count=100, bar_width=1.5)
+        return waveform_video
     else:
         return "Error: Failed to generate the waveform."
 
@@ -54,7 +59,8 @@ Note: The music generation process will take 1-2 minutes 🎶
 iface = gr.Interface(
     fn=generate_waveform,
     inputs=gr.Textbox(lines=2, placeholder="Enter a music description here..."),
-    outputs=gr.Audio(label="Download the Music 🎼"),
+    # outputs=gr.Audio(label="Download the Music 🎼"),
+    outputs=gr.Video(label="Watch the Waveform 🎼"),
     description=intro,
     examples=[
         ["A modern synthesizer creating futuristic soundscapes."],