AI API for Production Apps

1import requests
2import time
3from typing import Optional
4
5class ModelsLabClient:
6    """Production-ready ModelsLab API client with retry logic."""
7
8    def __init__(self, api_key: str, max_retries: int = 3):
9        self.api_key = api_key
10        self.base_url = "https://modelslab.com/api/v7"
11        self.max_retries = max_retries
12
13    def generate_image(self, prompt: str, model: str = "flux", **kwargs) -> Optional[list]:
14        payload = {
15            "key": self.api_key,
16            "model_id": model,
17            "prompt": prompt,
18            "width": kwargs.get("width", 1024),
19            "height": kwargs.get("height", 1024),
20            "samples": kwargs.get("samples", 1),
21        }
22
23        for attempt in range(self.max_retries):
24            try:
25                response = requests.post(
26                    f"{self.base_url}/images/text-to-image",
27                    json=payload,
28                    timeout=30
29                )
30
31                if response.status_code == 429:
32                    retry_after = int(response.headers.get("Retry-After", 5))
33                    time.sleep(retry_after)
34                    continue
35
36                response.raise_for_status()
37                data = response.json()
38
39                if data.get("status") == "success":
40                    return data["output"]
41                elif data.get("status") == "processing":
42                    return self._poll_result(data["fetch_result"])
43
44            except requests.exceptions.Timeout:
45                if attempt < self.max_retries - 1:
46                    time.sleep(2 ** attempt)
47                    continue
48                raise
49
50        return None
51
52# Usage
53client = ModelsLabClient("YOUR_API_KEY")
54images = client.generate_image("professional product photo, studio lighting")

1// Set up webhook endpoint for async video generation
2const express = require('express');
3const app = express();
4app.use(express.json());
5
6// Trigger video generation
7async function generateVideo(prompt) {
8  const response = await fetch('https://modelslab.com/api/v6/video/text2video', {
9    method: 'POST',
10    headers: { 'Content-Type': 'application/json' },
11    body: JSON.stringify({
12      key: process.env.MODELSLAB_API_KEY,
13      model_id: 'kling',
14      prompt: prompt,
15      webhook: 'https://your-app.com/webhooks/modelslab',
16      track_id: 'video-' + Date.now()
17    })
18  });
19  return response.json();
20}
21
22// Receive webhook when video is ready
23app.post('/webhooks/modelslab', (req, res) => {
24  const { status, output, track_id } = req.body;
25
26  if (status === 'success') {
27    console.log(`Video ready: ${output[0]}`);
28    // Notify user, update database, trigger next step
29  }
30
31  res.status(200).send('OK');
32});
33
34app.listen(3000);