Upload app.py

app.py

@@ -7,10 +7,6 @@ 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(
@@ -88,31 +84,8 @@ CUSTOM_CSS = """
 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
 
-firebase_initialized = init_firebase()
-
-# --- 2. قاموس الترجمة (معدل) ---
+# --- 2. قاموس الترجمة (عربي / إنجليزي) ---
 localization = {
     'en': {
         'brand_title': "",
@@ -131,12 +104,7 @@ localization = {
         'generic_error': "An error occurred:",
         'page_title': "SkyData - Background Removal",
         '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:"
+        'limit_warning': "You have reached the limit (10 attempts). Please try again later.", # <<< جديد
     },
     'ar': {
         'brand_title': "",
@@ -155,16 +123,15 @@ localization = {
         'generic_error': "حدث خطأ:",
         'page_title': "SkyData - أداة إزالة الخلفية",
         'spinner_text': "جاري معالجة الصورة، يرجى الانتظار...",
-        # <<< جديد: رسائل نظام المفاتيح
-        'api_key_prompt': "أدخل مفتاح الاستخدام (API Key)",
-        'api_key_missing': "الرجاء إدخال مفتاح الاستخدام لتشغيل الأداة.",
-        'api_key_invalid': "مفتاح استخدام غير صالح. الرجاء الحصول على مفتاح من موقعنا.",
-        'limit_reached': "لقد وصلت إلى الحد الأقصى (30 محاولة). يرجى المحاولة غداً.",
-        'attempts_left': "المحاولات المتبقية اليوم:"
+        'limit_warning': "لقد وصلت إلى الحد الأقصى (10 محاولات). يرجى المحاولة مرة أخرى لاحقاً.", # <<< جديد
     }
 }
 
-# --- 3. تحميل النموذج (كما هو) ---
+# --- 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)
@@ -173,61 +140,38 @@ def load_model():
         model = model.half()
     return model
 
-# --- 4. دالة المعالجة (كما هي) ---
+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()
+    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
 
-# --- <<< جديد: دالة التحقق من حد الاستخدام ---
-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) ---
 
-# --- 5. واجهة التطبيق (Sidebar) (معدلة) ---
 lang_choice = st.sidebar.radio(
     label="Select Language / اختر اللغة", 
     options=["English", "العربية"], 
@@ -236,17 +180,8 @@ 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}")
-
-selected_tab = st.sidebar.radio(loc['input_method'], [loc['tab_upload'], loc['tab_url'], loc['tab_file']])
+# --- 6. واجهة التطبيق الرئيسية (Main App) ---
 
-# --- 6. واجهة التطبيق الرئيسية (Main App) (كما هي) ---
 LOGO_URL = "https://i.ibb.co/v4vwvcGq/skydatafull.webp"
 col1, col2 = st.columns([1, 6]) 
 with col1:
@@ -254,77 +189,106 @@ 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 'attempt_count' not in st.session_state:
+    st.session_state.attempt_count = 0
 
-# (ملاحظة: قمت بإنشاء دالة موحدة للمعالجة لتجنب تكرار الكود)
-def handle_processing(image, original_name):
-    try:
-        with st.spinner(loc['spinner_text']):
-            processed_image, file_bytes = process(image)
+# --- منطق التبويبات (معدل) ---
+
+if selected_tab == loc['tab_upload']:
+    uploaded_file = st.file_uploader(loc['upload_prompt'], type=["jpg", "jpeg", "png"])
+    if uploaded_file is not None:
         
-        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}")
+        # <<< جديد: التحقق من الحد
+        if st.session_state.attempt_count >= 10:
+            st.warning(loc['limit_warning'])
+        else:
+            try:
+                image = Image.open(uploaded_file).convert("RGB")
+                with st.spinner(loc['spinner_text']):
+                    processed_image, file_bytes = process(image)
+                
+                # <<< جديد: زيادة العداد بعد النجاح
+                st.session_state.attempt_count += 1
+                
+                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}")
 
 
-# --- التحقق من المفتاح قبل أي شيء ---
-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 st.session_state.attempt_count >= 10:
+            st.warning(loc['limit_warning'])
+        else:
+            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.session_state.attempt_count += 1
+                
+                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_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:
-            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_file']:
+    uploaded_file = st.file_uploader(loc['upload_prompt_file'], type=["jpg", "jpeg", "png"])
+    if uploaded_file is not None:
+        
+        # <<< جديد: التحقق من الحد
+        if st.session_state.attempt_count >= 10:
+            st.warning(loc['limit_warning'])
+        else:
+            try:
+                image = Image.open(uploaded_file).convert("RGB")
+                
+                with st.spinner(loc['spinner_text']):
+                    processed_image, file_bytes = process(image)
+                
+                # <<< جديد: زيادة العداد بعد النجاح
+                st.session_state.attempt_count += 1
+                
+                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}")
+
+# <<< جديد: إظهار العداد للمستخدم (اختياري، يمكنك حذف هذا السطر)
+st.sidebar.info(f"المحاولات المتبقية: {10 - st.session_state.attempt_count}")