import requests
import base64
from PIL import Image
import io

def call_inpaint_api(image_path, mask_path, prompt):
    # Update this with your actual space URL after deployment
    url = "https://your-username-your-space-name.hf.space/inpaint"
    
    files = {
        'image': open(image_path, 'rb'),
        'mask': open(mask_path, 'rb')
    }
    
    data = {
        'prompt': prompt
    }
    
    response = requests.post(url, files=files, data=data)
    
    if response.status_code == 200:
        # Decode base64 image
        img_data = base64.b64decode(response.json()['image'])
        img = Image.open(io.BytesIO(img_data))
        img.save('result.png')
        return 'result.png'
    else:
        print(f"Error: {response.text}")
        return None