Update bg_removal_engine.py

bg_removal_engine.py

@@ -1,52 +1,78 @@
 import telebot
+from telebot import types
+import io
 from rembg import remove
 from PIL import Image
-import io
-import os
 
-# الساروت ديالك اللي عطيتيني
+# التوكين ديالك (تأكد باللي هو الصحيح)
 TOKEN = '8238787092:AAHPlHZeE3woH2t-q14LUYhg8AynYD2aX1E'
-
 bot = telebot.TeleBot(TOKEN)
 
-# رسالة الترحيب
-@bot.message_handler(commands=['start', 'help'])
-def send_welcome(message):
-    bot.reply_to(message, "مرحباً يا جمال! صيفط ليا دابا أي صورة وغادي نحيد ليها الخلفية في ثانية باستعمال Hugging Face. 🚀")
+# 1. تخصيص رسالة الترحيب بأزرار
+@bot.message_handler(commands=['start'])
+def start(message):
+    # إنشاء أزرار الكيبورد (Reply Keyboard)
+    markup = types.ReplyKeyboardMarkup(resize_keyboard=True, row_width=2)
+    item1 = types.KeyboardButton("إزالة خلفية صورة 🖼️")
+    item2 = types.KeyboardButton("حول البوت ℹ️")
+    markup.add(item1, item2)
+    
+    welcome_text = (
+        f"🚀 **أهلاً بك يا {message.from_user.first_name}!**\n\n"
+        "أنا بوت **Removeybg** الاحترافي. استعمل أحدث تقنيات الذكاء الاصطناعي "
+        "لعزل الصور بدقة عالية.\n\n"
+        "👇 **اضغط على الزر أسفله أو أرسل صورتك مباشرة!**"
+    )
+    bot.send_message(message.chat.id, welcome_text, parse_mode='Markdown', reply_markup=markup)
+
+# 2. التعامل مع الأزرار أو النصوص
+@bot.message_handler(func=lambda message: True)
+def echo_all(message):
+    if message.text == "إزالة خلفية صورة 🖼️":
+        bot.reply_to(message, "تفضل! صيفط ليا التصويرة اللي بغيتي تحيد ليها الخلفية.")
+    elif message.text == "حول البوت ℹ️":
+        bot.reply_to(message, "هذا البوت تم تطويره بواسطة Jamal (Digital Entrepreneur) "
+                              "باستخدام تقنية rembg والذكاء الاصطناعي.")
 
-# معالجة الصور المرسلة
+# 3. معالجة الصور باحترافية
 @bot.message_handler(content_types=['photo'])
 def handle_photo(message):
     try:
-        # إشعار للمستخدم بالبدء
-        msg = bot.reply_to(message, "جاري تنقية الصورة... انتظر لحظة ⏳")
+        # إرسال رسالة انتظار وتخزينها فمتغير باش نقدرو نعدلوها
+        status_msg = bot.reply_to(message, "⏳ جاري تحميل الصورة... [25%]")
         
-        # تحميل الصورة من تيليجرام
+        # تحميل الملف
         file_info = bot.get_file(message.photo[-1].file_id)
         downloaded_file = bot.download_file(file_info.file_path)
         
-        # فتح الصورة بـ PIL
-        input_image = Image.open(io.BytesIO(downloaded_file))
+        # تحديث الحالة
+        bot.edit_message_text("⚙️ جاري العزل بالذكاء الاصطناعي... [60%]", 
+                             message.chat.id, status_msg.message_id)
         
-        # إزالة الخلفية بالذكاء الاصطناعي
+        # المعالجة بـ rembg
+        input_image = Image.open(io.BytesIO(downloaded_file))
         output_image = remove(input_image)
         
-        # تحويل النتيجة لملف PNG جاهز للإرسال
+        # تحويل النتيجة لملف
         img_byte_arr = io.BytesIO()
         output_image.save(img_byte_arr, format='PNG')
         img_byte_arr.seek(0)
         
-        # إرسال النتيجة كملف (Document) باش تبقى الجودة عالية
-        bot.send_document(message.chat.id, img_byte_arr, visible_file_name='Removeybg_Result.png')
+        # اللمسة الأخيرة: إرسال النتيجة ومسح رسالة الحالة
+        bot.edit_message_text("✅ تمت المعالجة بنجاح! [100%]", 
+                             message.chat.id, status_msg.message_id)
+        
+        bot.send_document(message.chat.id, img_byte_arr, 
+                         visible_file_name='Removeybg_Result.png',
+                         caption="تفضل صورتك بدون خلفية بدقة HD! 🚀")
         
-        # مسح رسالة الانتظار
-        bot.delete_message(message.chat.id, msg.message_id)
+        # مسح رسالة الحالة بعد الإرسال بـ 2 ثواني (اختياري)
+        bot.delete_message(message.chat.id, status_msg.message_id)
         
     except Exception as e:
-        bot.reply_to(message, "وقع مشكل تقني، جرب صورة أخرى.")
-        print(f"Error logic: {e}")
+        bot.reply_to(message, "❌ وقع مشكل تقني. تأكد أن الملف صورة وحاول مرة أخرى.")
+        print(f"Error: {e}")
 
-# تشغيل البوت بشكل مستمر
 if __name__ == "__main__":
-    print("الما��ينة ناضية في Hugging Face... البوت متصل دابا!")
+    print("البوت الاحترافي متصل الآن...")
     bot.infinity_polling()