Update app.py

app.py

@@ -1,42 +1,60 @@
 import gradio as gr
 import numpy as np
 from PIL import Image
+import cv2
 
 def get_mask_bbox(mask: Image.Image):
     mask_np = np.array(mask)
     if len(mask_np.shape) == 3:
-        # إذا كان القناع ملون، استخدم قناة أخيرة (alpha إذا RGBA)
         mask_np = mask_np[..., -1] if mask_np.shape[2] == 4 else np.mean(mask_np, axis=-1)
     ys, xs = np.where(mask_np > 0)
     if ys.size == 0 or xs.size == 0:
         return None
     return int(xs.min()), int(ys.min()), int(xs.max()), int(ys.max())
 
-def mask_to_alpha(mask: Image.Image):
-    # إذا كان القناع RGBA وفيه قناة ألفا واضحة
-    if mask.mode == "RGBA":
-        alpha = mask.split()[-1]
-        return alpha
-    # إذا كان القناع ملون (RGB)، حوله إلى رمادي واستخدمه كشفافية
-    elif mask.mode == "RGB":
-        gray = mask.convert("L")
-        return gray
-    # إذا كان القناع بالفعل رمادي أو ثنائي
-    elif mask.mode == "L":
-        return mask
-    # إذا كان فيه وضع آخر، حوله لرمادي
-    else:
-        return mask.convert("L")
+def mask_to_alpha(mask: Image.Image, smooth_edges=True, morph_level=0):
+    # --- الحواف الناعمة ---
+    # إذا كان smooth_edges مفعّل، استخدم القناع كما هو (تدرج رمادي)
+    # إذا غير مفعّل، حول القناع إلى أبيض/أسود
+    mask_np = np.array(mask.convert("L"))
 
-def crop_masked_object(img: Image.Image, mask: Image.Image):
+    if not smooth_edges:
+        mask_np = np.where(mask_np > 127, 255, 0).astype(np.uint8)
+
+    # --- التنعيم المورفولوجي ---
+    if morph_level > 0:
+        # نستخدم GaussianBlur لتنعيم الحواف (يمكن استخدام غيره حسب رغبتك)
+        mask_np = cv2.GaussianBlur(mask_np, (0, 0), sigmaX=morph_level, sigmaY=morph_level, borderType=cv2.BORDER_DEFAULT)
+
+        # يمكن استخدام cv2.morphologyEx أيضاً
+        # kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (morph_level, morph_level))
+        # mask_np = cv2.morphologyEx(mask_np, cv2.MORPH_CLOSE, kernel)
+
+        # بعد التنعيم، قد يتحول إلى float، فلنجعله uint8
+        mask_np = np.clip(mask_np, 0, 255).astype(np.uint8)
+
+    return Image.fromarray(mask_np)
+
+def crop_masked_object(img: Image.Image, mask: Image.Image, smooth_edges=True, morph_level=0, scale=1.0, rotation=0, transparency=1.0):
     bbox = get_mask_bbox(mask)
     if bbox is None:
         return None, None
     xmin, ymin, xmax, ymax = bbox
     cropped_img = img.crop((xmin, ymin, xmax + 1, ymax + 1)).convert("RGBA")
     cropped_mask = mask.crop((xmin, ymin, xmax + 1, ymax + 1))
-    alpha_mask = mask_to_alpha(cropped_mask)
+    alpha_mask = mask_to_alpha(cropped_mask, smooth_edges, morph_level)
     cropped_img.putalpha(alpha_mask)
+    # تكبير/تصغير
+    if scale != 1.0:
+        new_size = (int(cropped_img.size[0] * scale), int(cropped_img.size[1] * scale))
+        cropped_img = cropped_img.resize(new_size, Image.LANCZOS)
+    # تدوير
+    if rotation != 0:
+        cropped_img = cropped_img.rotate(rotation, expand=True)
+    # شفافية
+    if transparency < 1.0:
+        alpha = cropped_img.split()[-1].point(lambda p: int(p * transparency))
+        cropped_img.putalpha(alpha)
     return cropped_img, bbox
 
 def paste_object_on_mask(base_img, base_mask, obj_img, obj_bbox):
@@ -52,7 +70,8 @@ def paste_object_on_mask(base_img, base_mask, obj_img, obj_bbox):
     result.paste(obj_img, (x_paste, y_paste), obj_img)
     return result
 
-def process_transfer(image1_editor, mask1_upload, image2_editor, mask2_upload):
+def process_transfer(image1_editor, mask1_upload, image2_editor, mask2_upload,
+                    scale, transparency, rotation, smooth_edges, morph_level):
     if not image1_editor or not image2_editor:
         return None, None, None, "**❌ يرجى رفع الصورتين وتحديد الأقنعة."
     if image1_editor.get("background") is None or image2_editor.get("background") is None:
@@ -67,8 +86,7 @@ def process_transfer(image1_editor, mask1_upload, image2_editor, mask2_upload):
         return None, None, None, "**❌ يجب رسم أو رفع قناع على كلتا الصورتين."
     base_img = image1_editor["background"].convert("RGBA")
     obj_img = image2_editor["background"].convert("RGBA")
-    # لاحظ: لا حاجة لتحويل القناع هنا، سيتم معالجته في crop_masked_object
-    cropped_obj, obj_bbox = crop_masked_object(obj_img, obj_mask)
+    cropped_obj, obj_bbox = crop_masked_object(obj_img, obj_mask, smooth_edges, morph_level, scale, rotation, transparency)
     if cropped_obj is None:
         return None, None, None, "**❌ لم يتم تحديد أي كائن في الصورة الثانية."
     result_img = paste_object_on_mask(base_img, base_mask, cropped_obj, obj_bbox)
@@ -91,20 +109,7 @@ def create_interface():
         """)
         gr.Markdown("""
         # 🖼️ نقل كائن من صورة إلى أخرى مع القناع والمحاذاة
-
-        **يدعم جميع أنواع الأقنعة:**
-        - قناع أبيض/أسود (ثنائي)
-        - قناع متدرج (Soft Mask)
-        - قناع ملون (RGB)
-        - قناع فيه قناة ألفا شفافة (RGBA)
-        - حتى أقنعة التقسيم (كل كائن بلون مختلف، سيتم نقل المناطق غير سوداء)
-
-        **طريقة الاستخدام:**
-        1. ارفع الصورة الأولى (صورة الخلفية/مكان اللصق).
-        2. ارفع الصورة الثانية (الصورة التي بها الكائن).
-        3. ارسم أو ارفع أي نوع قناع للكائن في الصورة الثانية.
-        4. ارسم أو ارفع قناع في مكان اللصق في الصورة الأولى.
-        5. اضغط زر "نقل الكائن" وستظهر النتيجة!
+        يمكنك تعديل إعدادات نقل الكائن من هنا، أو تركها على الإعدادات الافتراضية.
         """)
         with gr.Row():
             with gr.Column():
@@ -133,6 +138,13 @@ def create_interface():
                     height=200,
                     elem_classes=["silver-border"]
                 )
+        with gr.Row():
+            scale = gr.Slider(0.5, 2.0, value=1.0, step=0.05, label="حجم الكائن (Scale)", elem_classes=["silver-border"])
+            transparency = gr.Slider(0.2, 1.0, value=1.0, step=0.05, label="شفافية الكائن (Transparency)", elem_classes=["silver-border"])
+            rotation = gr.Slider(-180, 180, value=0, step=1, label="تدوير الكائن (Rotation)", elem_classes=["silver-border"])
+        with gr.Row():
+            smooth_edges = gr.Checkbox(value=True, label="تفعيل الحواف الناعمة (Smooth Edges/Anti-aliasing)", elem_classes=["silver-border"])
+            morph_level = gr.Slider(0, 10, value=0, step=1, label="مستوى تنعيم القناع (Morphological Smoothing)", elem_classes=["silver-border"])
         with gr.Row():
             transfer_btn = gr.Button("🚀 نقل الكائن", variant="primary")
         with gr.Row():
@@ -142,7 +154,7 @@ def create_interface():
         status = gr.Markdown()
         transfer_btn.click(
             fn=process_transfer,
-            inputs=[image1_editor, mask1_upload, image2_editor, mask2_upload],
+            inputs=[image1_editor, mask1_upload, image2_editor, mask2_upload, scale, transparency, rotation, smooth_edges, morph_level],
             outputs=[base_mask, obj_mask, result_img, status]
         )
     return demo