import requests
from PIL import Image
import base64
from io import BytesIO

url = "https://sezer91-sam.hf.space/segment/"
file_path = "img.jpeg"

try:
    # Görüntü dosyasını kontrol et
    img = Image.open(file_path)
    if img.size[0] < 64 or img.size[1] < 64:
        raise ValueError(f"Görüntü boyutu çok küçük: {img.size[0]}x{img.size[1]}. Minimum 64x64 piksel olmalı.")
    print(f"Görüntü boyutu: {img.size[0]}x{img.size[1]}")

    with open(file_path, "rb") as file:
        files = {"file": file}
        response = requests.post(url, files=files)

    if response.status_code == 200:
        result = response.json()
        print("Başarılı! Maske alındı.")
        # Base64'ü PNG olarak kaydet
        base64_string = result["mask"].split(",")[1]  # "data:image/png;base64," kısmını atla
        img_data = base64.b64decode(base64_string)
        img = Image.open(BytesIO(img_data))
        img.save("output_mask.png")
        print("Maske 'output_mask.png' olarak kaydedildi.")
    else:
        print(f"Hata: {response.status_code}, {response.text}")

except FileNotFoundError:
    print(f"Hata: {file_path} dosyası bulunamadı.")
except ValueError as ve:
    print(f"Hata: {str(ve)}")
except Exception as e:
    print(f"Hata: {str(e)}")