Получить статус задачи изображения

Начальное ожидание: 2 секунды
Интервал опроса: 3 секунды
Максимальное ожидание: 120 секунд
Типичное время: 5-30 секунд

import time
import requests

def poll_image_task(task_id, api_key, max_wait=120):
    """Опрос задачи генерации изображения до завершения или таймаута"""
    start_time = time.time()
    interval = 3  # Интервал 3 секунды
    
    while time.time() - start_time < max_wait:
        response = requests.get(
            f'https://toapis.com/v1/images/generations/{task_id}',
            headers={'Authorization': f'Bearer {api_key}'}
        )
        data = response.json()
        
        if data['status'] == 'completed':
            return data['url']
        elif data['status'] == 'failed':
            raise Exception(f"Генерация не удалась: {data['error']['message']}")
        
        time.sleep(interval)
    
    raise TimeoutError("Таймаут задачи")