Update app.py

app.py

@@ -97,7 +97,7 @@ 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, 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: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)}
 
 def get_music(prompt, musicAI_indx, duration):
     if musicAI_indx == 0: 
@@ -502,7 +502,9 @@ with block:
     share_button.click(None, [], [], _js=share_js)
 
     video_result = gr.Video(type=None, label='Final Merged video', elem_id="output-video")
-    spec_result = gr.Image()
+    imgfile_result = gr.Image(label="Art Cover", elem_id="output-img")
+    musicfile_result = gr.Audio(type='filepath', label="Generated Music Track", elem_id="output-music")
+    spec_result = gr.Image(label="Spectrogram Image")
     
     trigger_component = gr.Textbox(vaule="", visible=False) # This component is used for triggering inference funtion.
     translated_language = gr.Textbox(vaule="", visible=False)
@@ -515,7 +517,7 @@ with block:
     musicAI.change(fn=change_music_generator, inputs=[musicAI], outputs=[duration_input])
     text.submit(translate_language, inputs=[text], outputs=[language_tips_text, status_text, trigger_component, translated_language])
     btn.click(translate_language, inputs=[text], outputs=[language_tips_text, status_text, trigger_component, translated_language])
-    trigger_component.change(fn=get_result, inputs=[translated_language, styles, musicAI, duration_input], outputs=[spec_result, video_result, status_text, share_button, community_icon, loading_icon])
+    trigger_component.change(fn=get_result, inputs=[translated_language, styles, musicAI, duration_input], outputs=[spec_result, imgfile_result, musicfile_result, video_result, status_text, share_button, community_icon, loading_icon])
 
 
     gr.Markdown(
@@ -533,4 +535,4 @@ with block:
     ''')
     
 
-block.queue().launch()
+block.queue().launch()