Update app.py

app.py

@@ -23,6 +23,7 @@ import moviepy.video.io.ImageSequenceClip
 from moviepy.editor import *
 import mutagen
 from mutagen.mp3 import MP3
+from mutagen.wave import WAVE
 
 img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
 text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music") 
@@ -97,9 +98,11 @@ def get_result(text_prompts, style_indx, musicAI_indx, duration):
     spec_image, music_output = get_music(interrogate_prompt + ", " + style_list_EN[style_indx], musicAI_indx, duration)
     
     video_merged = merge_video(music_output, image_output)
-    return {spec_result:spec_image, imgfile_result:image_output, musicfile_result:music_output, video_result:video_merged, status_text:'Success', share_button:gr.update(visible=True), community_icon:gr.update(visible=True), loading_icon:gr.update(visible=True)}
+    return {spec_result:spec_image, imgfile_result:image_output, musicfile_result:"audio.wav", video_result:video_merged, status_text:'Success', share_button:gr.update(visible=True), community_icon:gr.update(visible=True), loading_icon:gr.update(visible=True)}
 
 def get_music(prompt, musicAI_indx, duration):
+    mp3file_name = "audio.mp3"
+    wavfile_name = "audio.wav"
     if musicAI_indx == 0: 
         if duration == 5:
             width_duration=512
@@ -108,13 +111,12 @@ def get_music(prompt, musicAI_indx, duration):
         spec = pipe2(prompt, height=512, width=width_duration).images[0]
         print(spec)
         wav = wav_bytes_from_spectrogram_image(spec)
-        with open("output.wav", "wb") as f:
+        with open(wavfile_name, "wb") as f:
             f.write(wav[0].getbuffer())
         
           
-        #Convert to mp3
-        mp3file_name = "audio.mp3"
-        wavfile = AudioSegment.from_wav("output.wav")
+        #Convert to mp3, for video merging function
+        wavfile = AudioSegment.from_wav(wavfile_name)
         wavfile.export(mp3file_name, format="mp3")
         return spec, mp3file_name
     else: 
@@ -128,7 +130,6 @@ def get_music(prompt, musicAI_indx, duration):
         """)
         
         url = result
-        mp3file_name = "file.mp3"
         
         data = urllib.request.urlopen(url)
         
@@ -136,10 +137,10 @@ def get_music(prompt, musicAI_indx, duration):
         f.write(data.read())
         f.close()
         
-        #wave_file="file.wav"
+        #Convert to wav, for sharing function only supports wav file
+        mp3file = AudioSegment.from_mp3(mp3file_name)
+        mp3file.export(wavfile_name, format="wav")
         
-        #sound = AudioSegment.from_mp3(save_as)
-        #sound.export(wave_file, format="wav")
         return None, mp3file_name