WIP

app.py

@@ -171,7 +171,7 @@ def download_youtube_audio(url):
                     f"{SIEVE_API_URL}/push",
                     headers={"X-API-Key": SIEVE_API_KEY, "Content-Type": "application/json"},
                     json=payload,
-                    timeout=30  # Add timeout
+                    timeout=1800  # Add timeout
                 )
                 response.raise_for_status()
                 response_data = response.json()
@@ -197,10 +197,10 @@ def download_youtube_audio(url):
         
         logger.info(f"Received job ID: {job_id}")
         
-        # Poll for job completion with timeout
+        # Poll for job completion
         poll_count = 0
-        max_polls = 60  # Maximum number of polls (2 minutes with 2-second delay)
-        start_time = time.time()
+        max_polls = 180  # Maximum number of polls (6 minutes with 2-second delay)
+        last_status = None
         
         while True:
             poll_count += 1
@@ -210,14 +210,16 @@ def download_youtube_audio(url):
                 job_response = requests.get(
                     f"{SIEVE_API_URL}/jobs/{job_id}",
                     headers={"X-API-Key": SIEVE_API_KEY},
-                    timeout=10
+                    timeout=1800,
                 )
                 job_response.raise_for_status()
                 job_data = job_response.json()
                 logger.debug(f"Job status response: {job_data}")
                 
                 status = job_data.get("status")
-                logger.info(f"Current job status: {status}")
+                if status != last_status:
+                    logger.info(f"Job status changed: {status}")
+                    last_status = status
                 
                 if status == "completed":
                     logger.info("Job completed successfully")
@@ -247,18 +249,18 @@ def download_youtube_audio(url):
                     with open(output_path, "wb") as f:
                         f.write(audio_response.content)
                     logger.info(f"Successfully saved audio to: {output_path}")
-                        
-                    return output_path
+                    
+                    # Break out of the polling loop after successful download
+                    break
                     
                 elif status == "failed":
                     error_msg = job_data.get("error", "Unknown error")
                     logger.error(f"Job failed with error: {error_msg}")
                     raise gr.Error(f"Job failed: {error_msg}")
                 
-                # Check for timeout
-                if time.time() - start_time > 120:  # 2 minutes timeout
-                    logger.error("Job polling timed out after 2 minutes")
-                    raise gr.Error("Download timed out. Please try again.")
+                if poll_count >= max_polls:
+                    logger.error("Maximum polling attempts reached")
+                    raise gr.Error("Download took too long. Please try again or check if the video is accessible.")
                 
                 logger.info("Job still processing, waiting 2 seconds before next poll...")
                 time.sleep(2)
@@ -275,6 +277,8 @@ def download_youtube_audio(url):
     except Exception as e:
         logger.exception(f"Unexpected error during YouTube download: {str(e)}")
         raise gr.Error(f"Failed to download YouTube audio: {str(e)}")
+        
+    return output_path
 
 def transcribe_youtube(url, return_timestamps, generate_subs):
     """Transcribe audio from YouTube video.