WIP

app.py

@@ -224,15 +224,21 @@ def download_youtube_audio(url):
                 if status == "completed" or status == "finished":
                     logger.info("Job completed successfully")
                     # Get the output data
-                    output_data = job_data.get("output_0", {})
+                    output_data = job_data.get("outputs", [])
                     if not output_data:
-                        logger.error("No output data found in completed job response")
+                        logger.error(f"No output data found in completed job response. Full response: {job_data}")
                         raise gr.Error("No output data in job response")
                     
+                    # Get the first output which should contain the audio URL
+                    first_output = output_data[0]
+                    if not isinstance(first_output, dict):
+                        logger.error(f"Unexpected output format: {first_output}")
+                        raise gr.Error("Unexpected output format from job response")
+                    
                     # Get the audio URL from the output
-                    audio_url = output_data.get("url")
+                    audio_url = first_output.get("url")
                     if not audio_url:
-                        logger.error("No audio URL found in output data")
+                        logger.error(f"No audio URL found in output data. Output data: {first_output}")
                         raise gr.Error("No audio URL in output data")
                     
                     logger.info(f"Received audio URL from Sieve: {audio_url}")