Update app.py

app.py

@@ -2,35 +2,23 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-from PIL import Image, ImageDraw
 
 HF_TOKEN = os.environ["HF_TOKEN"]
 client = InferenceClient(token=HF_TOKEN)
 
-def create_center_mask(image: Image.Image) -> Image.Image:
-    """Создаёт маску в центре изображения (пример для окна)"""
-    w, h = image.size
-    # Пример: маска 60% ширины и 80% высоты по центру
-    left = int(w * 0.2)
-    top = int(h * 0.1)
-    right = int(w * 0.8)
-    bottom = int(h * 0.9)
-    mask = Image.new("L", (w, h), 0)
-    draw = ImageDraw.Draw(mask)
-    draw.rectangle([left, top, right, bottom], fill=255)
-    return mask
-
-def add_curtains(image, style="classic", color="beige"):
-    if image is None:
-        return "❌ Загрузите фото окна"
+def add_curtains(editor_data, style="classic", color="beige"):
+    if editor_data is None:
+        return "❌ Пожалуйста, загрузите фото и нарисуйте область окна."
     
-    try:
-        # Создаём маску (позже можно заменить на ручной выбор)
-        mask = create_center_mask(image)
+    image = editor_data["image"]      # PIL.Image
+    mask = editor_data["mask"]        # PIL.Image (чёрно-белая маска)
 
-        prompt = f"{style} {color} curtains on the window, photorealistic, natural lighting, soft fabric"
+    if image is None or mask is None:
+        return "❌ Нарисуйте маску на окне."
 
-        # Вызываем inpainting
+    prompt = f"{style} {color} curtains on the window, photorealistic, natural lighting"
+
+    try:
         result = client.inpainting(
             image=image,
             mask_image=mask,
@@ -44,14 +32,16 @@ def add_curtains(image, style="classic", color="beige"):
 demo = gr.Interface(
     fn=add_curtains,
     inputs=[
-        gr.Image(type="pil", label="Фото вашего окна"),
+        gr.ImageEditor(
+            label="1. Загрузите фото окна\n2. Нажмите 'Edit' → выберите кисть → закрасьте область окна белым",
+            type="pil"
+        ),
         gr.Radio(["classic", "modern", "roman"], value="classic", label="Стиль штор"),
         gr.Textbox(value="beige", label="Цвет (например: white, navy blue)")
     ],
     outputs=gr.Image(label="Результат со шторами"),
     title="🪟 Виртуальная примерка штор",
-    description="Загрузите фото окна — ИИ добавит шторы!"
+    description="Нарисуйте маску на окне — ИИ добавит туда шторы!"
 )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()