Update app.py

app.py

@@ -1,11 +1,18 @@
 import gradio as gr
 import numpy as np
 from PIL import Image, ImageFilter
-from simple_lama_inpainting import SimpleLama
+import torch
+import cv2
 
-lama = SimpleLama()
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-def inpaint(editor_data, brush_size, feather):
+# تحميل lama_large_512px — نفس الموديل اللي في BallonsTranslator
+from iopaint.model.lama import LaMa
+from iopaint.schema import InpaintRequest, HDStrategy
+
+model = LaMa(device=device)
+
+def inpaint(editor_data, feather):
     if editor_data is None:
         return None
 
@@ -16,48 +23,39 @@ def inpaint(editor_data, brush_size, feather):
         return None
 
     image = bg.convert("RGB")
+    image_np = np.array(image)
 
     if not layers or layers[0] is None:
         return image
 
-    # استخراج الماسك من alpha channel طبقة الفرشاة
+    # استخراج الماسك
     layer = layers[0].convert("RGBA")
-    alpha = np.array(layer)[:, :, 3]
-
-    # تكبير الماسك شوية حتى يغطي حواف النص
-    mask_img = Image.fromarray(alpha)
+    mask_np = np.array(layer)[:, :, 3]
 
-    # feather (ناعم) اختياري
+    # feather اختياري
     if feather > 0:
-        mask_img = mask_img.filter(ImageFilter.GaussianBlur(radius=feather))
-
-    # تطبيق LaMa
-    result = lama(image, mask_img)
-    return result
-
-
-with gr.Blocks(
-    title="SFX Cleaner",
-    css="""
-    .gradio-container { max-width: 1100px !important; margin: auto; }
-    #run-btn { font-size: 1.2em; padding: 12px 30px; }
-    """
-) as demo:
-
-    gr.Markdown(
-        """
-        # 🧹 SFX Cleaner — LaMa Inpainting
-        **كيفية الاستخدام:**
-        1. ارفع صورة المانهوا
-        2. ارسم بالفرشاة البيضاء على النص / SFX
-        3. اضغط **تبييض**
-        """
-    )
+        mask_pil = Image.fromarray(mask_np)
+        mask_pil = mask_pil.filter(ImageFilter.GaussianBlur(radius=feather))
+        mask_np = np.array(mask_pil)
+
+    mask_np = (mask_np > 127).astype(np.uint8) * 255
+
+    config = InpaintRequest(hd_strategy=HDStrategy.ORIGINAL)
+    result = model(image_np, mask_np, config)
+
+    return Image.fromarray(result)
+
+
+with gr.Blocks(title="SFX Cleaner - LaMa Large") as demo:
+    gr.Markdown("""
+    # 🧹 SFX Cleaner — lama_large_512px
+    **الطريقة:** ارفع الصورة ← ارسم بالفرشاة على النص/SFX ← اضغط **تبييض**
+    """)
 
     with gr.Row():
         with gr.Column(scale=2):
             editor = gr.ImageEditor(
-                label="📌 ارسم على النص اللي تريد تمسحه",
+                label="📌 ارسم على النص",
                 brush=gr.Brush(
                     colors=["#ffffff"],
                     color_mode="fixed",
@@ -65,42 +63,19 @@ with gr.Blocks(
                 ),
                 eraser=gr.Eraser(default_size=20),
                 type="pil",
-                height=600,
+                height=650,
             )
-
         with gr.Column(scale=1):
             output = gr.Image(
                 label="✅ النتيجة",
                 type="pil",
-                height=600,
+                height=650,
             )
 
     with gr.Row():
-        with gr.Column():
-            brush_size = gr.Slider(
-                minimum=5, maximum=80, value=20, step=1,
-                label="حجم الفرشاة (Brush Size)"
-            )
-            feather = gr.Slider(
-                minimum=0, maximum=10, value=2, step=1,
-                label="نعومة الحواف (Feather)"
-            )
-        with gr.Column():
-            btn = gr.Button("🧹 تبييض", variant="primary", size="lg", elem_id="run-btn")
-            download = gr.DownloadButton(label="💾 حفظ الصورة", visible=False)
-
-    btn.click(
-        fn=inpaint,
-        inputs=[editor, brush_size, feather],
-        outputs=[output]
-    )
-
-    # تفعيل زر الحفظ بعد النتيجة
-    def enable_download(img):
-        if img is None:
-            return gr.update(visible=False)
-        return gr.update(visible=True)
-
-    output.change(enable_download, inputs=output, outputs=download)
+        feather = gr.Slider(0, 8, value=2, step=1, label="نعومة الحواف (Feather)")
+        btn = gr.Button("🧹 تبييض", variant="primary", size="lg")
+
+    btn.click(fn=inpaint, inputs=[editor, feather], outputs=output)
 
 demo.launch()