Edit an Image with AI

npm install axios

pip install requests

import axios from "axios";

const API_BASE_URL = "https://api.pictory.ai/pictoryapis";
const API_KEY = "YOUR_API_KEY"; // Replace with your actual API key

// Edit an existing image
const imageRequest = {
  prompt: "Replace the blue sofa in the first image with a tan leather armchair, keeping the lighting, floor, and everything else unchanged",
  model: "nanobanana2",
  aspectRatio: "16:9",
  referenceImageUrls: [
    "https://example.com/images/living-room.png"
  ]
};

import requests
import time

API_BASE_URL = "https://api.pictory.ai/pictoryapis"
API_KEY = "YOUR_API_KEY"  # Replace with your actual API key

# Edit an existing image
image_request = {
    "prompt": "Replace the blue sofa in the first image with a tan leather armchair, keeping the lighting, floor, and everything else unchanged",
    "model": "nanobanana2",
    "aspectRatio": "16:9",
    "referenceImageUrls": [
        "https://example.com/images/living-room.png"
    ]
}

async function editImage() {
  try {
    console.log("Submitting image edit request...");

    const response = await axios.post(
      `${API_BASE_URL}/v1/aistudio/images`,
      imageRequest,
      {
        headers: {
          "Content-Type": "application/json",
          Authorization: API_KEY,
        },
      }
    );

    const jobId = response.data.data.jobId;
    console.log("Image edit started.");
    console.log("Job ID:", jobId);

    return jobId;
  } catch (error) {
    console.error("Error submitting request:", error.response?.data || error.message);
    throw error;
  }
}

def edit_image():
    try:
        print("Submitting image edit request...")

        response = requests.post(
            f"{API_BASE_URL}/v1/aistudio/images",
            json=image_request,
            headers={
                "Content-Type": "application/json",
                "Authorization": API_KEY
            }
        )
        response.raise_for_status()

        job_id = response.json()["data"]["jobId"]
        print("Image edit started.")
        print(f"Job ID: {job_id}")

        return job_id

    except requests.exceptions.RequestException as error:
        print(f"Error submitting request: {error}")
        raise

async function waitForImage(jobId) {
  console.log("\nPolling for image edit result...");

  while (true) {
    const response = await axios.get(
      `${API_BASE_URL}/v1/jobs/${jobId}`,
      {
        headers: { Authorization: API_KEY },
      }
    );

    const data = response.data;
    const status = data.data.status;
    console.log("Status:", status);

    if (status === "completed") {
      console.log("\nImage edited successfully!");
      console.log("Image URL:", data.data.url);
      console.log("AI Credits Used:", data.data.aiCreditsUsed);
      return data;
    }

    if (status === "failed") {
      throw new Error("Image edit failed: " + JSON.stringify(data));
    }

    // Wait 15 seconds before polling again
    await new Promise(resolve => setTimeout(resolve, 15000));
  }
}

// Run the complete workflow
editImage()
  .then(jobId => waitForImage(jobId))
  .then(result => console.log("\nDone!"))
  .catch(error => console.error("Error:", error));

def wait_for_image(job_id):
    print("\nPolling for image edit result...")

    while True:
        response = requests.get(
            f"{API_BASE_URL}/v1/jobs/{job_id}",
            headers={"Authorization": API_KEY}
        )
        response.raise_for_status()

        data = response.json()
        status = data["data"]["status"]
        print(f"Status: {status}")

        if status == "completed":
            print("\nImage edited successfully!")
            print(f"Image URL: {data['data']['url']}")
            print(f"AI Credits Used: {data['data']['aiCreditsUsed']}")
            return data

        if status == "failed":
            raise Exception(f"Image edit failed: {data}")

        # Wait 15 seconds before polling again
        time.sleep(15)

# Run the complete workflow
if __name__ == "__main__":
    job_id = edit_image()
    result = wait_for_image(job_id)
    print("\nDone!")