Spaces:
Sleeping
Sleeping
File size: 2,301 Bytes
0362b52 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
from typing import Dict, Any, Optional
import requests
import base64
def generative_fill(
api_key: str,
image_data: bytes,
mask_data: bytes,
prompt: str,
negative_prompt: Optional[str] = None,
num_results: int = 4,
sync: bool = False,
seed: Optional[int] = None,
content_moderation: bool = False,
mask_type: str = "manual"
) -> Dict[str, Any]:
"""
Generate content in a masked area of an image using a text prompt.
Args:
api_key: Bria AI API key
image_data: Image data in bytes
mask_data: Mask image data in bytes
prompt: Description of what to generate in the masked area
negative_prompt: Description of what to avoid (optional)
num_results: Number of variations to generate (1-4)
sync: Whether to wait for results
seed: Optional seed for reproducible results
content_moderation: Whether to enable content moderation
mask_type: Type of mask ('manual' or 'automatic')
"""
url = "https://engine.prod.bria-api.com/v1/gen_fill"
headers = {
'api_token': api_key,
'Accept': 'application/json',
'Content-Type': 'application/json'
}
# Convert image and mask to base64
image_base64 = base64.b64encode(image_data).decode('utf-8')
mask_base64 = base64.b64encode(mask_data).decode('utf-8')
# Prepare request data
data = {
'file': image_base64,
'mask_file': mask_base64,
'mask_type': mask_type,
'prompt': prompt,
'num_results': num_results,
'sync': sync,
'content_moderation': content_moderation
}
# Add optional parameters
if negative_prompt:
data['negative_prompt'] = negative_prompt
if seed is not None:
data['seed'] = seed
try:
print(f"Making request to: {url}")
print(f"Headers: {headers}")
print(f"Data: {data}")
response = requests.post(url, headers=headers, json=data)
response.raise_for_status()
print(f"Response status: {response.status_code}")
print(f"Response body: {response.text}")
return response.json()
except Exception as e:
raise Exception(f"Generative fill failed: {str(e)}") |