Upload 2 files

app.py

@@ -7,6 +7,10 @@ import io
 import os
 import requests
 from io import BytesIO
+import firebase_admin
+from firebase_admin import credentials, db
+import datetime
+import json
 
 # --- 1. إعدادات الصفحة ---
 st.set_page_config(
@@ -14,115 +18,101 @@ st.set_page_config(
     layout="wide"
 )
 
-# --- !!! هذا هو الكود الجديد لتغيير التصميم !!! ---
+# --- كود التصميم (كما هو) ---
 CUSTOM_CSS = """
 <style>
+    /* ... (نفس كود CSS الذي أصلحناه سابقاً) ... */
     /* --- 1. الخلفية الرئيسية والخطوط --- */
     body, [data-testid="stAppViewContainer"], [data-testid="stHeader"] {
-        background-color: #121212 !important; /* الخلفية الرئيسية */
-        color: #e0e0e0 !important;
+        background-color: #121212 !important; color: #e0e0e0 !important;
     }
     [data-testid="stHeader"] {
-        background-color: #1e1e1e !important; /* لون شريط الهيدر */
-        border-bottom: 1px solid #333;
-    }
-    h1, h2, h3, h4, h5, h6 {
-        color: #ffffff !important; /* ألوان العناوين */
-    }
-    p, label, [data-testid="stMarkdownContainer"] {
-        color: #e0e0e0 !important; /* لون النص العادي */
+        background-color: #1e1e1e !important; border-bottom: 1px solid #333;
     }
+    h1, h2, h3, h4, h5, h6 { color: #ffffff !important; }
+    p, label, [data-testid="stMarkdownContainer"] { color: #e0e0e0 !important; }
 
     /* --- 2. الشريط الجانبي (Sidebar) --- */
-    
-    /* <<< تم التعديل: استهداف الشريط الجانبي "المفتوح" فقط لحل مشكلة الجلتش */
     [data-testid="stSidebar"][aria-expanded="true"] {
-        background-color: #1e1e1e !important; /* خلفية الشريط الجانبي */
-        border-right: 1px solid #333 !important; /* <<< تم التعديل: إضافة important */
+        background-color: #1e1e1e !important; border-right: 1px solid #333 !important;
     }
-    
-    /* <<< تم التعديل: استهداف الأزرار داخل الشريط الجانبي "المفتوح" فقط */
     [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] label { 
-        background-color: #2a2a2a;
-        padding: 10px;
-        border-radius: 5px;
-        margin: 5px 0;
-        transition: all 0.3s ease;
+        background-color: #2a2a2a; padding: 10px; border-radius: 5px;
+        margin: 5px 0; transition: all 0.3s ease;
     }
-    /* <<< تم التعديل */
     [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] label:hover {
-        background-color: #333;
-        color: #007bff;
+        background-color: #333; color: #007bff;
     }
-    /* <<< تم التعديل */
     [data-testid="stSidebar"][aria-expanded="true"] [data-testid="stRadio"] input:checked + div {
          color: #007bff; 
     }
 
     /* --- 3. زر التحميل (Download Button) --- */
     [data-testid="stDownloadButton"] > button {
-        background-color: #007bff !important;
-        color: #ffffff !important;
-        border: none !important;
-        border-radius: 5px !important;
-        font-weight: bold !important;
-        padding: 12px 20px !important;
-        transition: all 0.3s ease !important;
-        width: 100%; 
+        background-color: #007bff !important; color: #ffffff !important;
+        border: none !important; border-radius: 5px !important;
+        font-weight: bold !important; padding: 12px 20px !important;
+        transition: all 0.3s ease !important; width: 100%; 
     }
     [data-testid="stDownloadButton"] > button:hover {
-        background-color: #0056b3 !important;
-        transform: scale(1.03) !important;
+        background-color: #0056b3 !important; transform: scale(1.03) !important;
     }
     
     /* --- 4. زر رفع الملفات (File Uploader) --- */
     [data-testid="stFileUploader"] section {
-        background-color: #1e1e1e !important; /* خلفية صندوق الرفع */
-        border: 1px dashed #333 !important;
+        background-color: #1e1e1e !important; border: 1px dashed #333 !important;
         border-radius: 8px !important;
     }
-    /* زر "Browse files" */
     [data-testid="stFileUploader"] section > button { 
-        background-color: #007bff !important;
-        color: #ffffff !important;
-        border: none !important;
-        border-radius: 5px !important;
-        font-weight: bold !important;
-        transition: all 0.3s ease !important;
+        background-color: #007bff !important; color: #ffffff !important;
+        border: none !important; border-radius: 5px !important;
+        font-weight: bold !important; transition: all 0.3s ease !important;
     }
     [data-testid="stFileUploader"] section > button:hover {
-        background-color: #0056b3 !important;
-        transform: scale(1.03) !important;
-    }
-    [data-testid="stFileUploader"] label { /* عنوان صندوق الرفع */
-         color: #ffffff !important;
+        background-color: #0056b3 !important; transform: scale(1.03) !important;
     }
-    /* نص "Drag and drop" */
+    [data-testid="stFileUploader"] label { color: #ffffff !important; }
     [data-testid="stFileUploader"] section [data-testid="stMarkdownContainer"] p {
          color: #aaa !important;
     }
 
     /* --- 5. عناصر أخرى --- */
-    [data-testid="stImage"] figcaption { /* نص "Processed Image" */
-        color: #aaa !important;
-    }
-    [data-testid="stSpinner"] > div { /* نص التحميل "Processing..." */
-        color: #e0e0e0 !important;
-    }
-    /* صندوق إدخال الرابط */
+    [data-testid="stImage"] figcaption { color: #aaa !important; }
+    [data-testid="stSpinner"] > div { color: #e0e0e0 !important; }
     [data-testid="stTextInput"] input { 
-        background-color: #1e1e1e !important;
-        color: #e0e0e0 !important;
-        border: 1px solid #333 !important;
-        border-radius: 5px !important;
+        background-color: #1e1e1e !important; color: #e0e0e0 !important;
+        border: 1px solid #333 !important; border-radius: 5px !important;
     }
 </style>
 """
 st.markdown(CUSTOM_CSS, unsafe_allow_html=True)
 # --- نهاية كود التصميم ---
 
+# --- <<< جديد: إعداد Firebase ---
+@st.cache_resource
+def init_firebase():
+    try:
+        # (انظر خطوة الإعداد) جلب المفتاح السري من Hugging Face Secrets
+        creds_json_str = os.environ.get("FIREBASE_CREDS")
+        if creds_json_str:
+            creds_dict = json.loads(creds_json_str)
+            cred = credentials.Certificate(creds_dict)
+            # (هام) استبدل هذا برابط قاعدة بياناتك من ملف index.html
+            db_url = "https://pixelrpg-5ebc3-default-rtdb.europe-west1.firebasedatabase.app"
+            
+            if not firebase_admin._apps:
+                firebase_admin.initialize_app(cred, {'databaseURL': db_url})
+            return True
+        else:
+            st.error("خطأ فادح: لم يتم العثور على مفتاح Firebase. يرجى الاتصال بمسؤول الموقع.")
+            return False
+    except Exception as e:
+        st.error(f"خطأ في الاتصال بـ Firebase: {e}")
+        return False
 
-# --- 2. قاموس الترجمة (عربي / إنجليزي) ---
+firebase_initialized = init_firebase()
+
+# --- 2. قاموس الترجمة (معدل) ---
 localization = {
     'en': {
         'brand_title': "",
@@ -140,7 +130,13 @@ localization = {
         'error_processing': "Error processing image:",
         'generic_error': "An error occurred:",
         'page_title': "SkyData - Background Removal",
-        'spinner_text': "Processing image, please wait..."
+        'spinner_text': "Processing image, please wait...",
+        # <<< جديد: رسائل نظام المفاتيح
+        'api_key_prompt': "Enter your API Key",
+        'api_key_missing': "Please enter your API Key to use the tool.",
+        'api_key_invalid': "Invalid API Key. Please get a key from our website.",
+        'limit_reached': "You have reached your daily limit (30 attempts).",
+        'attempts_left': "Attempts left today:"
     },
     'ar': {
         'brand_title': "",
@@ -158,15 +154,17 @@ localization = {
         'error_processing': "خطأ في معالجة الصورة:",
         'generic_error': "حدث خطأ:",
         'page_title': "SkyData - أداة إزالة الخلفية",
-        'spinner_text': "جاري معالجة الصورة، يرجى الانتظار..."
+        'spinner_text': "جاري معالجة الصورة، يرجى الانتظار...",
+        # <<< جديد: رسائل نظام المفاتيح
+        'api_key_prompt': "أدخل مفتاح الاستخدام (API Key)",
+        'api_key_missing': "الرجاء إدخال مفتاح الاستخدام لتشغيل الأداة.",
+        'api_key_invalid': "مفتاح استخدام غير صالح. الرجاء الحصول على مفتاح من موقعنا.",
+        'limit_reached': "لقد وصلت إلى الحد الأقصى (30 محاولة). يرجى المحاولة غداً.",
+        'attempts_left': "المحاولات المتبقية اليوم:"
     }
 }
 
-# --- 3. تحميل النموذج ---
-torch.set_float32_matmul_precision(["high", "highest"][0])
-use_cuda = torch.cuda.is_available()
-device = "cuda" if use_cuda else "cpu"
-
+# --- 3. تحميل النموذج (كما هو) ---
 @st.cache_resource
 def load_model():
     model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True)
@@ -175,39 +173,61 @@ def load_model():
         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. دالة المعالجة الرئيسية ---
+# --- 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()
-
+    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)
-    
     img_bytes = io.BytesIO()
     image.save(img_bytes, format="PNG")
     img_bytes = img_bytes.getvalue()
-    
     return image, img_bytes
 
-# --- 5. واجهة التطبيق (Sidebar) ---
+# --- <<< جديد: دالة التحقق من حد الاستخدام ---
+DAILY_LIMIT = 30
+@st.cache_data(ttl=60) # تخزين نتيجة التحقق لمدة 60 ثانية لتجنب الضغط على Firebase
+def check_and_update_limit(api_key, loc):
+    if not firebase_initialized:
+        return False, loc['generic_error']
+        
+    try:
+        ref = db.reference(f'users/{api_key}')
+        data = ref.get()
+        
+        if not data:
+            return False, loc['api_key_invalid']
+            
+        today_date = datetime.date.today().isoformat()
+        last_reset = data.get('lastResetDate', '1970-01-01')
+        count = int(data.get('usageCount', 0))
+        
+        # تصفير العداد إذا كان يوماً جديداً
+        if last_reset != today_date:
+            count = 0
+            ref.update({'lastResetDate': today_date, 'usageCount': 0})
+            
+        # التحقق من الحد
+        if count >= DAILY_LIMIT:
+            st.session_state.remaining_attempts = 0
+            return False, loc['limit_reached']
+        
+        # (مهم) تحديث العداد في قاعدة البيانات
+        ref.update({'usageCount': count + 1})
+        st.session_state.remaining_attempts = DAILY_LIMIT - (count + 1)
+        return True, "" # نجح
+
+    except Exception as e:
+        return False, f"Error checking key: {e}"
 
-# اختيار اللغة
+# --- 5. واجهة التطبيق (Sidebar) (معدلة) ---
 lang_choice = st.sidebar.radio(
     label="Select Language / اختر اللغة", 
     options=["English", "العربية"], 
@@ -216,10 +236,17 @@ lang_choice = st.sidebar.radio(
 lang_code = 'ar' if lang_choice == 'العربية' else 'en'
 loc = localization[lang_code]
 
+# --- <<< جديد: إضافة خانة المفتاح ---
+st.sidebar.title(loc['brand_title'])
+api_key = st.sidebar.text_input(loc['api_key_prompt'], type="password")
+
+# --- <<< جديد: إظهار المحاولات المتبقية (إذا تم إدخال المفتاح)
+if 'remaining_attempts' in st.session_state:
+    st.sidebar.info(f"{loc['attempts_left']} {st.session_state.remaining_attempts}")
 
-# --- 6. واجهة التطبيق الرئيسية (Main App) ---
+selected_tab = st.sidebar.radio(loc['input_method'], [loc['tab_upload'], loc['tab_url'], loc['tab_file']])
 
-# الشعار والعنوان في الصفحة الرئيسية
+# --- 6. واجهة التطبيق الرئيسية (Main App) (كما هي) ---
 LOGO_URL = "https://i.ibb.co/v4vwvcGq/skydatafull.webp"
 col1, col2 = st.columns([1, 6]) 
 with col1:
@@ -227,76 +254,77 @@ with col1:
 with col2:
     st.title(loc['page_title'])
 
-# بقية الشريط الجانبي
-st.sidebar.title(loc['brand_title'])
-selected_tab = st.sidebar.radio(loc['input_method'], [loc['tab_upload'], loc['tab_url'], loc['tab_file']])
-
-# --- منطق التبويبات ---
+# --- منطق التبويبات (معدل بالكامل) ---
 
-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}")
+# (ملاحظة: قمت بإنشاء دالة موحدة للمعالجة لتجنب تكرار الكود)
+def handle_processing(image, original_name):
+    try:
+        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"{original_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}")
+# --- التحقق من المفتاح قبل أي شيء ---
+if not firebase_initialized:
+    st.error("خطأ في الاتصال بقاعدة البيانات. لا يمكن تشغيل الأداة.")
+else:
+    if selected_tab == loc['tab_upload']:
+        uploaded_file = st.file_uploader(loc['upload_prompt'], type=["jpg", "jpeg", "png"])
+        if uploaded_file is not None:
+            if not api_key:
+                st.info(loc['api_key_missing'])
+            else:
+                is_ok, message = check_and_update_limit(api_key, loc)
+                if is_ok:
+                    image = Image.open(uploaded_file).convert("RGB")
+                    handle_processing(image, uploaded_file.name)
+                else:
+                    st.warning(message)
 
+    elif selected_tab == loc['tab_url']:
+        image_url = st.text_input(loc['url_prompt'])
+        if image_url:
+            if not api_key:
+                st.info(loc['api_key_missing'])
+            else:
+                is_ok, message = check_and_update_limit(api_key, loc)
+                if is_ok:
+                    try:
+                        response = requests.get(image_url, stream=True)
+                        response.raise_for_status()
+                        image = Image.open(BytesIO(response.content)).convert("RGB")
+                        
+                        try:
+                            file_name = image_url.split('/')[-1]
+                        except Exception:
+                            file_name = "processed_image.png"
+                            
+                        handle_processing(image, file_name)
+                    except requests.exceptions.RequestException as e:
+                        st.error(f"{loc['error_fetching']} {e}")
+                    except Exception as e:
+                        st.error(f"{loc['error_processing']} {e}")
+                else:
+                    st.warning(message)
 
-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}")
+    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:
+            if not api_key:
+                st.info(loc['api_key_missing'])
+            else:
+                is_ok, message = check_and_update_limit(api_key, loc)
+                if is_ok:
+                    image = Image.open(uploaded_file).convert("RGB")
+                    handle_processing(image, uploaded_file.name)
+                else:
+                    st.warning(message)

requirements.txt

@@ -7,3 +7,4 @@ transformers==4.48.3
 einops>=0.4.0
 kornia>=0.6.0
 timm>=0.6.0
+firebase-admin==6.5.0