Update app.py

app.py

@@ -77,7 +77,7 @@ def submit_job(audio_file, preset, beat_sensitivity, fps, width, height):
     else:
         output_url = tracking_data['links']['outputUrls'][0]
         logging.info(f"Job succeeded. Output URL: {output_url}")
-        return output_url
+        return output_url, None
 
 description1 = """
 # Audio Visualizer Playground
@@ -167,8 +167,10 @@ video, .gr-video {
 """
 
 def main(audio_file, preset, beat_sensitivity, fps, width, height):
-    video_url = submit_job(audio_file, preset, beat_sensitivity, fps, width, height)
-    return video_url
+    result, error = submit_job(audio_file, preset, beat_sensitivity, fps, width, height)
+    if error:
+        return None, error
+    return result, None
 
 with gr.Blocks(css=custom_css) as demo:
     gr.Markdown(description1, elem_id="markdown-text")
@@ -187,11 +189,17 @@ with gr.Blocks(css=custom_css) as demo:
             height = gr.Slider(label="Height", minimum=512, maximum=1024, step=1, value=512)
         submit_button = gr.Button("Submit")
     output_video = gr.Video(label="Output MP4")
+    output_error = gr.Textbox(label="Error", visible=False)
 
-    def update_output(video_url):
-        output_video.update(value=video_url, visible=True)
+    def update_output(video_url, error):
+        if error:
+            output_error.update(value=error, visible=True)
+            output_video.update(value=None, visible=False)
+        else:
+            output_error.update(value="", visible=False)
+            output_video.update(value=video_url, visible=True)
 
-    submit_button.click(main, inputs=[audio_file, preset, beat_sensitivity, fps, width, height], outputs=[output_video])
+    submit_button.click(main, inputs=[audio_file, preset, beat_sensitivity, fps, width, height], outputs=[output_video, output_error])
 
 if __name__ == "__main__":
     demo.launch()