WIP

app.py

@@ -84,16 +84,9 @@ def get_youtube_audio_url(url):
             "inputs": {
                 "url": url,
                 "download_type": "audio",
-                "resolution": "highest-available",
-                "include_audio": True,
+                "audio_format": "mp3",
                 "start_time": 0,
-                "end_time": -1,
-                "include_metadata": False,
-                "metadata_fields": ["title", "thumbnail", "description", "tags", "duration"],
-                "include_subtitles": False,
-                "subtitle_languages": ["en"],
-                "video_format": "mp4",
-                "audio_format": "mp3"
+                "end_time": -1
             }
         }
         
@@ -103,6 +96,7 @@ def get_youtube_audio_url(url):
         
         for attempt in range(max_retries):
             try:
+                logger.info(f"Sending request to Sieve API (attempt {attempt + 1}/{max_retries})")
                 response = requests.post(
                     f"{SIEVE_API_URL}/push",
                     headers={"X-API-Key": SIEVE_API_KEY, "Content-Type": "application/json"},
@@ -111,10 +105,12 @@ def get_youtube_audio_url(url):
                 )
                 response.raise_for_status()
                 response_data = response.json()
+                logger.info(f"Sieve API response: {response_data}")
                 
                 job_id = response_data.get("id")
                 if not job_id:
                     if attempt < max_retries - 1:
+                        logger.warning(f"No job ID received, retrying in {retry_delay} seconds")
                         time.sleep(retry_delay)
                         continue
                     raise gr.Error("Failed to get job ID from Sieve API")
@@ -122,6 +118,7 @@ def get_youtube_audio_url(url):
                 
             except requests.exceptions.RequestException as e:
                 if attempt < max_retries - 1:
+                    logger.warning(f"Request failed: {str(e)}, retrying in {retry_delay} seconds")
                     time.sleep(retry_delay)
                     continue
                 raise
@@ -134,6 +131,7 @@ def get_youtube_audio_url(url):
             poll_count += 1
             
             try:
+                logger.info(f"Polling job status (attempt {poll_count}/{max_polls})")
                 job_response = requests.get(
                     f"{SIEVE_API_URL}/jobs/{job_id}",
                     headers={"X-API-Key": SIEVE_API_KEY},
@@ -141,6 +139,7 @@ def get_youtube_audio_url(url):
                 )
                 job_response.raise_for_status()
                 job_data = job_response.json()
+                logger.info(f"Job response: {job_data}")
                 
                 status = job_data.get("status")
                 
@@ -164,10 +163,12 @@ def get_youtube_audio_url(url):
                     if not audio_url.startswith(('http://', 'https://')):
                         raise gr.Error(f"Invalid audio URL scheme: {audio_url}")
                     
+                    logger.info(f"Successfully got audio URL: {audio_url}")
                     return audio_url
                     
                 elif status == "failed":
                     error_msg = job_data.get("error", "Unknown error")
+                    logger.error(f"Job failed: {error_msg}")
                     raise gr.Error(f"Job failed: {error_msg}")
                 
                 if poll_count >= max_polls:
@@ -178,9 +179,11 @@ def get_youtube_audio_url(url):
             except requests.exceptions.RequestException as e:
                 if poll_count >= max_polls:
                     raise gr.Error("Failed to check job status. Please try again.")
+                logger.warning(f"Request failed: {str(e)}, retrying in 2 seconds")
                 time.sleep(2)
             
     except Exception as e:
+        logger.exception(f"Error during YouTube URL fetch: {str(e)}")
         raise gr.Error(f"Failed to get YouTube audio URL: {str(e)}")
 
 def check_api_health():