Delete sprite_cutter.py

sprite_cutter.py

@@ -1,476 +0,0 @@
-import os
-import sys
-import cv2
-import numpy as np
-import torch
-import torchvision.transforms as transforms
-from PIL import Image, ImageDraw
-import zipfile
-from pathlib import Path
-import tempfile
-import shutil
-from scipy import ndimage
-from datetime import datetime
-import traceback
-
-try:
-    import gradio as gr
-except ImportError:
-    print("❌ Gradio not installed. Installing...")
-    os.system("pip install -q gradio torch torchvision pillow opencv-python scipy numpy")
-    import gradio as gr
-
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-TEMP_DIR = tempfile.mkdtemp(prefix="sprite_cutter_")
-
-def cleanup_temp():
-    """تنظيف الملفات المؤقتة"""
-    if os.path.exists(TEMP_DIR):
-        shutil.rmtree(TEMP_DIR)
-
-def remove_background_ai(image_cv2):
-    """
-    حذف الخلفية باستخدام Color Thresholding + Morphological Operations
-    يعمل بسرعة عالية على الـ CPU
-    """
-    try:
-        if len(image_cv2.shape) == 2:
-            image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_GRAY2BGR)
-        
-        bgr = image_cv2.copy()
-        hsv = cv2.cvtColor(bgr, cv2.COLOR_BGR2HSV)
-        
-        # كشف الألوان الموحدة (الخلفيات النموذجية للـ Pixel Art)
-        lower_blue = np.array([90, 100, 100])
-        upper_blue = np.array([130, 255, 255])
-        
-        lower_pink = np.array([140, 50, 50])
-        upper_pink = np.array([180, 255, 255])
-        
-        mask_blue = cv2.inRange(hsv, lower_blue, upper_blue)
-        mask_pink = cv2.inRange(hsv, lower_pink, upper_pink)
-        
-        # دمج الأقنعة
-        mask = cv2.bitwise_or(mask_blue, mask_pink)
-        
-        # تطبيق عمليات مورفولوجية
-        kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
-        mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel, iterations=1)
-        mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel, iterations=1)
-        
-        # عكس القناع (الكائنات = أبيض)
-        mask = cv2.bitwise_not(mask)
-        
-        # تطبيق القناع على الصورة
-        result = cv2.bitwise_and(bgr, bgr, mask=mask)
-        result = cv2.cvtColor(result, cv2.COLOR_BGR2RGBA)
-        result[:, :, 3] = mask
-        
-        return result, mask
-    except Exception as e:
-        print(f"❌ خطأ في حذف الخلفية: {e}")
-        return None, None
-
-def detect_sprites_gpu(image_array, mask):
-    """
-    كشف الكائنات باستخدام Connected Components على الـ GPU
-    """
-    try:
-        if mask is None:
-            return []
-        
-        # تطبيق Connected Components
-        labeled, num_features = ndimage.label(mask > 128)
-        
-        sprites_info = []
-        
-        for i in range(1, num_features + 1):
-            component = (labeled == i).astype(np.uint8) * 255
-            
-            # حساب الإحصائيات
-            contours, _ = cv2.findContours(component, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-            
-            if not contours:
-                continue
-            
-            cnt = max(contours, key=cv2.contourArea)
-            area = cv2.contourArea(cnt)
-            
-            # تصفية الكائنات الصغيرة جداً
-            if area < 10:
-                continue
-            
-            x, y, w, h = cv2.boundingRect(cnt)
-            
-            # تحديد الأبعاد لأقرب 2 بكسل (محاذاة للـ pixel grid)
-            w = max(w, 2)
-            h = max(h, 2)
-            
-            sprites_info.append({
-                'x': int(x),
-                'y': int(y),
-                'w': int(w),
-                'h': int(h),
-                'area': float(area),
-                'label': i
-            })
-        
-        return sorted(sprites_info, key=lambda s: (s['y'], s['x']))
-    except Exception as e:
-        print(f"❌ خطأ في كشف الكائنات: {e}")
-        traceback.print_exc()
-        return []
-
-def extract_sprites(image_cv2, sprites_info, preview=False):
-    """
-    استخراج الكائنات وإرجاع الصور المقصوصة
-    """
-    extracted = []
-    preview_image = image_cv2.copy()
-    
-    try:
-        colors = [
-            (0, 255, 0),
-            (255, 0, 0),
-            (0, 0, 255),
-            (255, 255, 0),
-            (255, 0, 255),
-            (0, 255, 255)
-        ]
-        
-        for idx, sprite in enumerate(sprites_info):
-            x, y, w, h = sprite['x'], sprite['y'], sprite['w'], sprite['h']
-            
-            # استخراج الكائن
-            sprite_img = image_cv2[y:y+h, x:x+w].copy()
-            
-            extracted.append({
-                'image': sprite_img,
-                'filename': f"sprite_{idx+1:03d}_{w}x{h}.png",
-                'data': sprite
-            })
-            
-            # رسم المربعات على الصورة المعاينة
-            if preview:
-                color = colors[idx % len(colors)]
-                cv2.rectangle(preview_image, (x, y), (x+w, y+h), color, 2)
-                cv2.putText(preview_image, str(idx+1), (x+3, y+15), cv2.FONT_HERSHEY_SIMPLEX, 0.4, color, 1)
-        
-        return extracted, preview_image
-    except Exception as e:
-        print(f"❌ خطأ في استخراج الكائنات: {e}")
-        return extracted, preview_image
-
-def create_zip_archive(sprites_list):
-    """
-    إنشاء ملف ZIP يحتوي على جميع الكائنات المقصوصة
-    """
-    try:
-        zip_path = os.path.join(TEMP_DIR, f"sprites_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip")
-        
-        with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
-            for idx, sprite in enumerate(sprites_list):
-                img = sprite['image']
-                filename = sprite['filename']
-                
-                # تحويل إلى RGB إذا لزم الأمر
-                if len(img.shape) == 2:
-                    img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
-                elif img.shape[2] == 4:
-                    img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
-                
-                # حفظ في الذاكرة
-                success, buffer = cv2.imencode('.png', img)
-                if success:
-                    zipf.writestr(filename, buffer.tobytes())
-        
-        return zip_path
-    except Exception as e:
-        print(f"❌ خطأ في إنشاء ZIP: {e}")
-        traceback.print_exc()
-        return None
-
-def process_sprite_sheet(image_file, sensitivity_slider=0.5):
-    """
-    المعالج الرئيسي الذي يربط كل المراحل
-    """
-    try:
-        if image_file is None:
-            return None, None, None, "❌ من فضلك حمل صورة أولاً"
-        
-        # قراءة الصورة
-        image_path = image_file if isinstance(image_file, str) else image_file
-        image_cv2 = cv2.imread(image_path)
-        
-        if image_cv2 is None:
-            return None, None, None, "❌ فشل تحميل الصورة"
-        
-        image_cv2 = cv2.cvtColor(image_cv2, cv2.COLOR_BGR2RGB)
-        
-        # حذف الخلفية
-        image_rgba, mask = remove_background_ai(image_cv2)
-        if image_rgba is None:
-            return None, None, None, "❌ خطأ في حذف الخلفية"
-        
-        # كشف الكائنات
-        sprites_info = detect_sprites_gpu(image_cv2, mask)
-        
-        if not sprites_info:
-            return None, None, None, "⚠️ لم يتم كشف أي كائنات. حاول صورة مختلفة"
-        
-        # استخراج الكائنات
-        extracted_sprites, preview = extract_sprites(image_cv2, sprites_info, preview=True)
-        
-        # إنشاء ZIP
-        zip_path = create_zip_archive(extracted_sprites)
-        
-        # إعداد رسالة النتيجة
-        stats_msg = f"""
-✅ **تم بنجاح!**
-
-📊 الإحصائيات:
-• عدد الكائنات المكتشفة: **{len(sprites_info)}**
-• أبعاد الصورة الأصلية: **{image_cv2.shape[1]}x{image_cv2.shape[0]}** بكسل
-• متوسط حجم الكائن: **{int(np.mean([s['area'] for s in sprites_info]))}** بكسل²
-
-📦 الملف جاهز للتحميل!
-        """
-        
-        preview_rgb = cv2.cvtColor(preview, cv2.COLOR_BGR2RGB)
-        
-        return preview_rgb, extracted_sprites[0]['image'] if extracted_sprites else None, zip_path, stats_msg
-    
-    except Exception as e:
-        error_msg = f"❌ خطأ: {str(e)}\n{traceback.format_exc()}"
-        print(error_msg)
-        return None, None, None, error_msg
-
-def create_ui():
-    """
-    إنشاء واجهة Gradio احترافية وجميلة
-    """
-    
-    css = """
-    :root {
-        --primary: #ff006e;
-        --secondary: #00d9ff;
-        --dark: #0a0e27;
-        --card: #1a1f3a;
-        --text: #e0e0ff;
-    }
-    
-    body {
-        background: linear-gradient(135deg, var(--dark) 0%, #1a0033 100%);
-        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-        color: var(--text);
-    }
-    
-    #main_container {
-        background: rgba(10, 14, 39, 0.95);
-        border: 2px solid var(--secondary);
-        border-radius: 15px;
-        padding: 30px;
-        box-shadow: 0 0 50px rgba(0, 217, 255, 0.3);
-    }
-    
-    .title-section {
-        text-align: center;
-        margin-bottom: 30px;
-        padding-bottom: 20px;
-        border-bottom: 2px solid var(--primary);
-    }
-    
-    .title-section h1 {
-        margin: 0;
-        font-size: 2.5em;
-        background: linear-gradient(45deg, var(--secondary), var(--primary));
-        -webkit-background-clip: text;
-        -webkit-text-fill-color: transparent;
-        background-clip: text;
-        text-shadow: 0 0 20px rgba(0, 217, 255, 0.5);
-    }
-    
-    .subtitle {
-        color: rgba(224, 224, 255, 0.7);
-        margin-top: 10px;
-    }
-    
-    .control-panel {
-        background: var(--card);
-        border: 1px solid var(--secondary);
-        border-radius: 10px;
-        padding: 20px;
-        margin-bottom: 20px;
-    }
-    
-    .control-panel h3 {
-        color: var(--secondary);
-        margin-top: 0;
-    }
-    
-    .slider {
-        width: 100%;
-    }
-    
-    .button-group {
-        display: flex;
-        gap: 10px;
-        margin-top: 20px;
-    }
-    
-    .button-group button {
-        flex: 1;
-        padding: 12px 20px;
-        font-size: 1em;
-        background: linear-gradient(45deg, var(--primary), #ff4081);
-        color: white;
-        border: none;
-        border-radius: 8px;
-        cursor: pointer;
-        transition: all 0.3s ease;
-        font-weight: bold;
-    }
-    
-    .button-group button:hover {
-        transform: translateY(-2px);
-        box-shadow: 0 10px 25px rgba(255, 0, 110, 0.4);
-    }
-    
-    .output-section {
-        background: var(--card);
-        border: 2px solid var(--secondary);
-        border-radius: 10px;
-        padding: 20px;
-        margin-top: 20px;
-    }
-    
-    .preview-image {
-        border: 2px dashed var(--secondary);
-        border-radius: 8px;
-        padding: 10px;
-    }
-    
-    .status-message {
-        background: rgba(0, 217, 255, 0.1);
-        border-left: 4px solid var(--secondary);
-        padding: 15px;
-        border-radius: 5px;
-        margin-top: 15px;
-        font-weight: 500;
-    }
-    
-    .file-download {
-        background: linear-gradient(45deg, var(--secondary), #00ffe0);
-        color: var(--dark);
-        padding: 15px;
-        border-radius: 8px;
-        text-align: center;
-        font-weight: bold;
-        margin-top: 15px;
-    }
-    """
-    
-    with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
-        with gr.Column(elem_id="main_container"):
-            # Header
-            with gr.Group():
-                gr.HTML("""
-                <div class="title-section">
-                    <h1>🎮 أداة قص البكسل التلقائية</h1>
-                    <p class="subtitle">
-                        استخراج تلقائي للكائنات من صور Sprite Sheet
-                        <br/>
-                        <em>AI-Powered Pixel Art Sprite Extraction</em>
-                    </p>
-                </div>
-                """)
-            
-            # Upload Section
-            with gr.Group(elem_classes="control-panel"):
-                gr.Markdown("### 📤 حمل صورة البكسل")
-                image_input = gr.Image(
-                    label="صورة Sprite Sheet",
-                    type="filepath",
-                    accept_types=["image/*"],
-                    show_label=False
-                )
-                sensitivity = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.5,
-                    step=0.1,
-                    label="📊 حساسية الكشف (Detection Sensitivity)",
-                    info="أعلى = كشف أدق لكائنات صغيرة"
-                )
-            
-            # Process Button
-            with gr.Group(elem_classes="control-panel"):
-                process_btn = gr.Button(
-                    "🚀 ابدأ المعالجة",
-                    variant="primary",
-                    scale=1
-                )
-            
-            # Results Section
-            with gr.Group(elem_classes="output-section"):
-                gr.Markdown("### 📊 النتائج")
-                
-                with gr.Tabs():
-                    with gr.TabItem("🔍 معاينة"):
-                        preview_image = gr.Image(
-                            label="صورة المعاينة (مع تحديد الكائنات)",
-                            type="numpy",
-                            elem_classes="preview-image"
-                        )
-                    
-                    with gr.TabItem("✂️ أول كائن"):
-                        first_sprite = gr.Image(
-                            label="أول كائن تم قصه",
-                            type="numpy",
-                            elem_classes="preview-image"
-                        )
-                    
-                    with gr.TabItem("📦 الملف"):
-                        zip_file = gr.File(
-                            label="تحميل جميع الكائنات (ZIP)",
-                            type="filepath"
-                        )
-                
-                # Status Message
-                status_msg = gr.Markdown(
-                    "⏳ جاهز للعملية...",
-                    elem_classes="status-message"
-                )
-            
-            # Footer
-            gr.HTML("""
-            <div style="text-align: center; margin-top: 30px; padding-top: 20px; border-top: 1px solid rgba(224, 224, 255, 0.2); color: rgba(224, 224, 255, 0.6); font-size: 0.9em;">
-                ⚡ معالجة سريعة بقوة GPU | 🎨 دعم كامل للـ RTL العربي | 🔒 معالجة محلية آمنة
-                <br/>
-                <small>Version 1.0 | Built with ❤️ for Pixel Artists</small>
-            </div>
-            """)
-        
-        # Connect Events
-        process_btn.click(
-            fn=process_sprite_sheet,
-            inputs=[image_input, sensitivity],
-            outputs=[preview_image, first_sprite, zip_file, status_msg]
-        )
-        
-        return demo
-
-if __name__ == "__main__":
-    try:
-        demo = create_ui()
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False,
-            show_error=True
-        )
-    except Exception as e:
-        print(f"❌ خطأ في بدء التطبيق: {e}")
-        traceback.print_exc()
-    finally:
-        cleanup_temp()