Upload 4 files

README.md

@@ -1,13 +1,12 @@
 ---
-title: Crystal
-emoji: 📚
+title: Back Ground Removal
+emoji: 📊
 colorFrom: gray
 colorTo: purple
-sdk: gradio
-sdk_version: 5.49.1
+sdk: streamlit
+sdk_version: 1.51.0
 app_file: app.py
 pinned: false
-short_description: 'Crystal AI Tools '
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,214 @@
+import streamlit as st
+from PIL import Image
+import torch
+from torchvision import transforms
+from transformers import AutoModelForImageSegmentation
+import io
+import os
+import requests
+from io import BytesIO
+
+# --- 1. إعدادات الصفحة والعلامة التجارية ---
+st.set_page_config(
+    page_title="SkyData - Background Removal",
+    layout="wide"
+)
+
+# --- 2. قاموس الترجمة (عربي / إنجليزي) ---
+localization = {
+    'en': {
+        'brand_title': "",
+        'lang_select': "Select Language",
+        'input_method': "Select Input Method",
+        'tab_upload': "Image Upload",
+        'tab_url': "URL Input",
+        'tab_file': "File Output",
+        'upload_prompt': "Upload an image",
+        'upload_prompt_file': "Upload an image for file output",
+        'url_prompt': "Paste an image URL",
+        'processed_caption': "Processed Image",
+        'download_button': "Download PNG",
+        'error_fetching': "Error fetching image from URL:",
+        'error_processing': "Error processing image:",
+        'generic_error': "An error occurred:",
+        'page_title': "SkyData - Background Removal",
+        'spinner_text': "Processing image, please wait..."
+    },
+    'ar': {
+        'brand_title': "",
+        'lang_select': "اختر اللغة",
+        'input_method': "اختر طريقة الإدخال",
+        'tab_upload': "رفع صورة",
+        'tab_url': "رابط صورة",
+        'tab_file': "تصدير ملف",
+        'upload_prompt': "ارفع صورة",
+        'upload_prompt_file': "ارفع صورة لتصدير الملف",
+        'url_prompt': "الصق رابط الصورة",
+        'processed_caption': "الصورة المعالجة",
+        'download_button': "تحميل بصيغة PNG",
+        'error_fetching': "خطأ في جلب الصورة من الرابط:",
+        'error_processing': "خطأ في معالجة الصورة:",
+        'generic_error': "حدث خطأ:",
+        'page_title': "SkyData - أداة إزالة الخلفية",
+        'spinner_text': "جاري معالجة الصورة، يرجى الانتظار..."
+    }
+}
+
+# --- 3. تحميل النموذج ---
+
+torch.set_float32_matmul_precision(["high", "highest"][0])
+
+use_cuda = torch.cuda.is_available()
+device = "cuda" if use_cuda else "cpu"
+
+@st.cache_resource
+def load_model():
+    model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True)
+    model.to(device)
+    if use_cuda:
+        model = model.half()
+    return model
+
+birefnet = load_model()
+
+transform_image = transforms.Compose([
+    transforms.Resize((1024, 1024)), # الدقة العالية
+    transforms.ToTensor(),
+    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
+])
+
+# --- 4. دالة المعالجة الرئيسية (معدلة) ---
+@st.cache_data
+def process(image):
+    # المعالجة بالذكاء الاصطناعي
+    image_size = image.size
+    input_images = transform_image(image).unsqueeze(0).to(device)
+    if use_cuda:
+        input_images = input_images.half()
+
+    with torch.no_grad():
+        preds = birefnet(input_images)[-1].sigmoid().cpu()
+    
+    pred = preds[0].squeeze()
+    pred_pil = transforms.ToPILImage()(pred)
+    mask = pred_pil.resize(image_size)
+    
+    # تطبيق القناع
+    image.putalpha(mask) # تعديل الصورة لتصبح شفافة
+    
+    # تحويل الصورة إلى bytes للتحميل
+    img_bytes = io.BytesIO()
+    image.save(img_bytes, format="PNG") # 'image' هي الآن الصورة الشفافة
+    img_bytes = img_bytes.getvalue()
+
+    # إرجاع الصورة (للعرض) و الـ bytes (للتحميل)
+    return image, img_bytes
+
+# --- 5. واجهة التطبيق (Sidebar) ---
+
+LOGO_URL = "https://i.ibb.co/v4vwvcGq/skydatafull.webp"
+st.sidebar.image(LOGO_URL)
+
+lang_choice = st.sidebar.radio(
+    label="Select Language / اختر اللغة", 
+    options=["English", "العربية"], 
+    horizontal=True
+)
+lang_code = 'ar' if lang_choice == 'العربية' else 'en'
+loc = localization[lang_code]
+
+if lang_code == 'ar':
+    st.markdown(
+        """
+        <style>
+        html, body, [data-testid="stSidebar"], [data-testid="stHeader"] {
+            direction: rtl !important;
+        }
+        [data-testid="stSidebar"] [data-testid="stRadio"] div[role="radiogroup"] {
+            flex-direction: row-reverse;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+# --- 6. واجهة التطبيق الرئيسية (Main App) ---
+
+st.sidebar.title(loc['brand_title']) 
+selected_tab = st.sidebar.radio(loc['input_method'], [loc['tab_upload'], loc['tab_url'], loc['tab_file']])
+st.title(loc['page_title'])
+
+# --- تم التعديل: إضافة زر التحميل لكل التبويبات ---
+
+if selected_tab == loc['tab_upload']:
+    uploaded_file = st.file_uploader(loc['upload_prompt'], type=["jpg", "jpeg", "png"])
+    if uploaded_file is not None:
+        try:
+            image = Image.open(uploaded_file).convert("RGB")
+            with st.spinner(loc['spinner_text']):
+                processed_image, file_bytes = process(image) # <-- تم التعديل
+            
+            st.image(processed_image, caption=loc['processed_caption'])
+            
+            # --- تمت الإضافة: زر التحميل ---
+            st.download_button(
+                label=loc['download_button'],
+                data=file_bytes,
+                file_name=f"{uploaded_file.name.rsplit('.', 1)[0]}.png",
+                mime="image/png",
+            )
+        except Exception as e:
+            st.error(f"{loc['generic_error']} {e}")
+
+
+elif selected_tab == loc['tab_url']:
+    image_url = st.text_input(loc['url_prompt'])
+    if image_url:
+        try:
+            response = requests.get(image_url, stream=True)
+            response.raise_for_status()
+            image = Image.open(BytesIO(response.content)).convert("RGB")
+            
+            with st.spinner(loc['spinner_text']):
+                processed_image, file_bytes = process(image) # <-- تم التعديل
+            
+            st.image(processed_image, caption=loc['processed_caption'])
+            
+            # --- تمت الإضافة: زر التحميل ---
+            try: # محاولة استخراج اسم الملف من الرابط
+                file_name = image_url.split('/')[-1].rsplit('.', 1)[0] + ".png"
+            except Exception:
+                file_name = "processed_image.png"
+                
+            st.download_button(
+                label=loc['download_button'],
+                data=file_bytes,
+                file_name=file_name,
+                mime="image/png",
+            )
+        except requests.exceptions.RequestException as e:
+            st.error(f"{loc['error_fetching']} {e}")
+        except Exception as e:
+            st.error(f"{loc['error_processing']} {e}")
+
+
+elif selected_tab == loc['tab_file']:
+    uploaded_file = st.file_uploader(loc['upload_prompt_file'], type=["jpg", "jpeg", "png"])
+    if uploaded_file is not None:
+        try: # <-- تم نقل معالجة الأخطاء هنا
+            image = Image.open(uploaded_file).convert("RGB")
+            
+            with st.spinner(loc['spinner_text']):
+                processed_image, file_bytes = process(image) # <-- تم التعديل
+            
+            st.image(processed_image, caption=loc['processed_caption'])
+            
+            # زر التحميل موجود هنا أصلاً
+            st.download_button(
+                label=loc['download_button'],
+                data=file_bytes,
+                file_name=f"{uploaded_file.name.rsplit('.', 1)[0]}.png",
+                mime="image/png",
+            )
+        except Exception as e:
+            st.error(f"{loc['generic_error']} {e}")

gitattributes.txt

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -0,0 +1,9 @@
+Pillow==11.1.0
+Requests==2.32.3
+streamlit==1.42.0
+torch==2.6.0
+torchvision==0.21.0
+transformers==4.48.3
+einops>=0.4.0
+kornia>=0.6.0
+timm>=0.6.0