Update app.py

app.py

@@ -2,34 +2,56 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
+from PIL import Image, ImageDraw
 
 HF_TOKEN = os.environ["HF_TOKEN"]
-client = InferenceClient(model="black-forest-labs/FLUX.1-dev", token=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 None
-    prompt = f"Add {style} {color} curtains to the window in this photo, photorealistic, natural lighting, soft fabric folds, no distortion"
+        return "❌ Загрузите фото окна"
+    
     try:
-        result = client.image_to_image(
+        # Создаём маску (позже можно заменить на ручной выбор)
+        mask = create_center_mask(image)
+
+        prompt = f"{style} {color} curtains on the window, photorealistic, natural lighting, soft fabric"
+
+        # Вызываем inpainting
+        result = client.inpainting(
             image=image,
+            mask_image=mask,
             prompt=prompt,
-            strength=0.75  # 0.0 = оригинал, 1.0 = полностью новое
+            model="diffusers/stable-diffusion-xl-1.0-inpainting-0.1"
         )
         return result
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Ошибка: {str(e)}"
 
 demo = gr.Interface(
     fn=add_curtains,
     inputs=[
-        gr.Image(type="pil", label="Photo of your window"),
-        gr.Radio(["classic", "modern", "roman", "austrian"], value="classic", label="Curtain style"),
-        gr.Textbox(value="beige", label="Color (e.g. white, navy blue, gray)")
+        gr.Image(type="pil", label="Фото вашего окна"),
+        gr.Radio(["classic", "modern", "roman"], value="classic", label="Стиль штор"),
+        gr.Textbox(value="beige", label="Цвет (например: white, navy blue)")
     ],
-    outputs=gr.Image(label="Result with curtains"),
-    title="🪟 Virtual Curtain Try-On",
-    description="Upload a photo of your window — AI will add realistic curtains!"
+    outputs=gr.Image(label="Результат со шторами"),
+    title="🪟 Виртуальная примерка штор",
+    description="Загрузите фото окна — ИИ добавит шторы!"
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()