Flux Image Editing

from openai import OpenAI

# Initialize client
client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://api.laozhang.ai/v1"
)

def edit_local_image(image_path, prompt, aspect_ratio="1:1", mask_path=None, 
                    seed=None, safety_tolerance=1, output_format="png"):
    """Edit local image"""
    try:
        # Prepare base parameters
        edit_params = {
            "model": "flux-kontext-max",
            "image": open(image_path, "rb"),
            "prompt": prompt,
            "extra_body": {
                "aspect_ratio": aspect_ratio,
                "safety_tolerance": safety_tolerance,
                "output_format": output_format
            }
        }
        
        # Optional parameters
        if seed is not None:
            edit_params["extra_body"]["seed"] = seed
        
        # Add mask if provided
        if mask_path:
            edit_params["mask"] = open(mask_path, "rb")
        
        # Call edit API
        response = client.images.edit(**edit_params)
        
        # Return edited image URL (note: valid for 10 minutes)
        return response.data[0].url
    
    except Exception as e:
        print(f"Edit failed: {e}")
        return None
    
    finally:
        # Ensure file handles are closed
        if 'edit_params' in locals():
            if 'image' in edit_params:
                edit_params['image'].close()
            if 'mask' in edit_params:
                edit_params['mask'].close()

# Usage example
edited_url = edit_local_image(
    image_path="portrait.jpg",
    prompt="Change the background to a beautiful sunset sky",
    aspect_ratio="2:3",
    seed=42,  # Reproducible results
    safety_tolerance=1,  # Safety level (0-2)
    output_format="jpeg"  # or "png"
)

if edited_url:
    print(f"Edit complete: {edited_url}")
    print("Note: URL will expire after 10 minutes, download promptly!")

# Add text to image
add_text = edit_local_image(
    image_path="poster.jpg",
    prompt='Add the text "SALE 50% OFF" in bold red letters at the top center',
    aspect_ratio="3:2"
)

# Modify existing text
modify_text = edit_local_image(
    image_path="sign.jpg", 
    prompt='Change the text from "Open" to "Closed"',
    aspect_ratio="1:1"
)

# Use quotes for text precision
precise_text = edit_local_image(
    image_path="banner.jpg",
    prompt='Replace the existing text with exactly "Welcome to Our Store"',
    aspect_ratio="7:3"
)

# Maintain lighting consistency
lighting_edit = edit_local_image(
    image_path="interior.jpg",
    prompt="Add a lamp in the corner, matching the existing warm lighting",
    aspect_ratio="16:9"
)

# Maintain perspective and scale
perspective_edit = edit_local_image(
    image_path="street.jpg", 
    prompt="Add a bicycle parked next to the building, matching the street perspective",
    aspect_ratio="3:2"
)

# Maintain character features
character_edit = edit_local_image(
    image_path="person.jpg",
    prompt="Change the person's shirt to blue while keeping all facial features identical",
    aspect_ratio="2:3",
    seed=42  # Fixed seed for consistency
)

# Replace background
background_edit = edit_local_image(
    image_path="portrait.jpg",
    prompt="Replace the background with a modern office environment",
    aspect_ratio="2:3"
)

# Remove background
remove_bg = edit_local_image(
    image_path="product.jpg", 
    prompt="Remove the background, make it transparent",
    aspect_ratio="1:1"
)

# Add elements
add_element = edit_local_image(
    image_path="room.jpg",
    prompt="Add a beautiful plant in the corner of the room",
    aspect_ratio="16:9"
)

# Remove objects
remove_object = edit_local_image(
    image_path="street.jpg",
    prompt="Remove the car from the street, replace with empty road",
    aspect_ratio="3:2"
)

# Replace objects
replace_object = edit_local_image(
    image_path="table.jpg",
    prompt="Replace the apple on the table with an orange", 
    aspect_ratio="1:1"
)

# Apply artistic style
artistic_edit = edit_local_image(
    image_path="photo.jpg",
    prompt="Transform this photo into a watercolor painting style",
    aspect_ratio="3:2"
)

# Color adjustment
color_edit = edit_local_image(
    image_path="landscape.jpg",
    prompt="Make the image more vibrant with saturated colors",
    aspect_ratio="16:9"
)

from PIL import Image, ImageDraw

def create_circular_mask(image_path, center_x, center_y, radius):
    """Create circular mask"""
    # Open original image to get dimensions
    img = Image.open(image_path)
    width, height = img.size
    
    # Create black mask
    mask = Image.new('RGB', (width, height), 'black')
    draw = ImageDraw.Draw(mask)
    
    # Draw white circle
    draw.ellipse(
        [(center_x - radius, center_y - radius),
         (center_x + radius, center_y + radius)],
        fill='white'
    )
    
    # Save mask
    mask_path = 'mask_circular.png'
    mask.save(mask_path)
    return mask_path

# Usage
mask = create_circular_mask('photo.jpg', 512, 512, 200)
edited = edit_local_image(
    image_path='photo.jpg',
    prompt='Change the center area to a starry night sky',
    mask_path=mask
)

# ❌ Vague prompt
prompt = "make it better"

# ✅ Specific and detailed
prompt = """
Replace the cloudy sky with a clear blue sky with some white clouds.
Keep the foreground buildings exactly as they are.
Maintain the natural lighting and shadows.
"""

# ✅ Technical precision
prompt = """
Change the car color from red to metallic blue.
Preserve all reflections and highlights.
Match the lighting of the original image.
No other changes to composition.
"""

import requests
import time

def download_and_save(url, filename):
    """Download edited image immediately"""
    try:
        # Download immediately (URL expires in 10 minutes)
        response = requests.get(url, timeout=30)
        response.raise_for_status()
        
        with open(filename, 'wb') as f:
            f.write(response.content)
        
        print(f"✅ Saved: {filename} ({len(response.content) // 1024}KB)")
        return True
    
    except Exception as e:
        print(f"❌ Download failed: {e}")
        return False

# Generate and save immediately
edited_url = edit_local_image('photo.jpg', 'your prompt')
if edited_url:
    download_and_save(edited_url, f'edited_{int(time.time())}.png')