Update app.py

app.py

@@ -3,58 +3,155 @@ from flask import Flask, render_template_string
 from playwright.sync_api import sync_playwright
 from PIL import Image
 import img2pdf
+from telebot import types
 
-# --- الإعدادات ---
+# --- الإعدادات العامة ---
 BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+ADMIN_HANDLE = "@svipfast"
 bot = telebot.TeleBot(BOT_TOKEN, threaded=False)
 app = Flask(__name__)
 
+# تخزين مؤقت لبيانات المستخدمين
+user_data = {}
+
 @app.route('/')
-def home(): return "<h1>Manga Engine Active</h1>"
+def home(): return "<h1>Manga Engine Professional is Running</h1>"
 
-def fetch_chapter_with_playwright(url):
+# --- محرك السحب (Playwright) ---
+def fetch_images(url):
     with sync_playwright() as p:
-        # تشغيل متصفح مخفي (Headless)
         browser = p.chromium.launch(headless=True)
         context = browser.new_context(user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
         page = context.new_page()
-        
         try:
-            # الذهاب للرابط والانتظار حتى يتم تحميل الشبكة بالكامل (مثل HakuNeko)
             page.goto(url, wait_until="networkidle", timeout=60000)
-            
-            # التمرير لأسفل الصفحة ببطء لضمان تحميل صور (Lazy Loading)
-            for _ in range(5):
-                page.mouse.wheel(0, 1000)
+            # التمرير لضمان تحميل الصور (Lazy Load)
+            for _ in range(4):
+                page.mouse.wheel(0, 1500)
                 time.sleep(1)
             
-            # استخراج روابط الصور التي تنتهي بامتدادات صور معروفة
             images = page.query_selector_all("img")
-            img_data_list = []
-            
+            img_list = []
             for img in images:
                 src = img.get_attribute("src") or img.get_attribute("data-src")
                 if src and any(ext in src.lower() for ext in ['.jpg', '.jpeg', '.png', '.webp']):
-                    # تحويل الصورة إلى Bytes
                     res = page.request.get(src)
-                    if res.status == 200 and len(res.body()) > 20000:
-                        img_data_list.append(Image.open(io.BytesIO(res.body())).convert('RGB'))
-            
+                    if res.status == 200 and len(res.body()) > 25000:
+                        img_list.append(Image.open(io.BytesIO(res.body())).convert('RGB'))
             browser.close()
-            return img_data_list
-        except Exception as e:
+            return img_list
+        except:
             browser.close()
             return []
 
-def process_engine(sample_url, start, end):
+# --- واجهة البوت ---
+
+@bot.message_handler(commands=['start'])
+def send_welcome(message):
+    markup = types.InlineKeyboardMarkup(row_width=1)
+    btn1 = types.InlineKeyboardButton("📥 ابدأ تحميل مانجا", callback_data="start_download")
+    btn2 = types.InlineKeyboardButton("ℹ️ شرح طريقة الروابط", callback_data="how_to_url")
+    btn3 = types.InlineKeyboardButton("👨‍💻 تواصل مع الإدارة", url=f"https://t.me/{ADMIN_HANDLE.replace('@','')}")
+    markup.add(btn1, btn2, btn3)
+    
+    welcome_text = (
+        "<b>مرحباً بك في نظام سحب المانجا الإحترافي 🚀</b>\n\n"
+        "هذا النظام مدعوم بمحرك محاكاة متطور لتجاوز الحماية وسحب الفصول بأعلى جودة ممكنة.\n\n"
+        f"للدعم الفني: {ADMIN_HANDLE}"
+    )
+    bot.send_message(message.chat.id, welcome_text, parse_mode="HTML", reply_markup=markup)
+
+@bot.callback_query_handler(func=lambda call: True)
+def callback_query(call):
+    if call.data == "start_download":
+        bot.answer_callback_query(call.id)
+        msg = bot.send_message(call.message.chat.id, "<b>الرجاء إرسال رابط المانجا المراد تحميلها:</b>", parse_mode="HTML")
+        bot.register_next_step_handler(msg, get_manga_url)
+        
+    elif call.data == "how_to_url":
+        help_text = (
+            "<b>💡 كيف تحصل على الرابط الصحيح؟</b>\n\n"
+            "يجب أن ينتهي الرابط برقم الفصل ليعمل النظام تلقائياً:\n\n"
+            "✅ <b>أزورا:</b>\n<code>https://azoramoon.com/series/manga-name/chapter-1</code>\n\n"
+            "✅ <b>أوليمبوس:</b>\n<code>https://olympustaff.com/series/manga-name/1</code>\n\n"
+            "⚠️ تأكد من نسخ الرابط من المتصفح مباشرة."
+        )
+        bot.send_message(call.message.chat.id, help_text, parse_mode="HTML")
+
+def get_manga_url(message):
+    url = message.text.strip()
+    if not url.startswith("http"):
+        bot.reply_to(message, "❌ الرابط غير صالح. يرجى البدء بـ http")
+        return
+    
+    user_data[message.chat.id] = {'url': url}
+    
+    markup = types.InlineKeyboardMarkup()
+    markup.add(types.InlineKeyboardButton("🎯 فصل واحد فقط", callback_data="type_single"),
+               types.InlineKeyboardButton("📦 حزمة فصول (مدى)", callback_data="type_range"))
+    markup.add(types.InlineKeyboardButton("⏩ تحميل 10 فصول قادمة", callback_data="type_auto_10"))
+    
+    bot.send_message(message.chat.id, "<b>اختر طريقة التحميل المفضلة:</b>", parse_mode="HTML", reply_markup=markup)
+
+@bot.callback_query_handler(func=lambda call: call.data.startswith("type_"))
+def handle_type_selection(call):
+    chat_id = call.message.chat.id
+    if chat_id not in user_data: return
+
+    if call.data == "type_single":
+        msg = bot.send_message(chat_id, "🔢 أرسل رقم الفصل المراد تحميله:")
+        bot.register_next_step_handler(msg, lambda m: start_process(m, "single"))
+    
+    elif call.data == "type_range":
+        msg = bot.send_message(chat_id, "🔢 أرسل النطاق المطلوب (مثال: 1-10):")
+        bot.register_next_step_handler(msg, lambda m: start_process(m, "range"))
+        
+    elif call.data == "type_auto_10":
+        msg = bot.send_message(chat_id, "🔢 أرسل رقم الفصل الذي تريد البدء منه:")
+        bot.register_next_step_handler(msg, lambda m: start_process(m, "auto10"))
+
+def start_process(message, mode):
+    chat_id = message.chat.id
+    url = user_data[chat_id]['url']
+    
+    try:
+        if mode == "single":
+            start, end = message.text, message.text
+        elif mode == "range":
+            start, end = message.text.split('-')
+        elif mode == "auto10":
+            start = int(message.text)
+            end = start + 9
+            
+        status_msg = bot.send_message(chat_id, f"<b>⏳ جاري تشغيل المحرك...</b>\n🔄 معالجة الفصول من {start} إلى {end}", parse_mode="HTML")
+        
+        # تنفيذ السحب
+        zip_path = run_engine_logic(url, start, end)
+        
+        if zip_path:
+            with open(zip_path, 'rb') as f:
+                caption = (
+                    "<b>✅ تمت عملية السحب بنجاح</b>\n\n"
+                    f"📂 <b>الحزمة:</b> {start} - {end}\n"
+                    f"👤 <b>بواسطة:</b> {ADMIN_HANDLE}"
+                )
+                bot.send_document(chat_id, f, caption=caption, parse_mode="HTML")
+            os.remove(zip_path)
+            bot.delete_message(chat_id, status_msg.message_id)
+        else:
+            bot.edit_message_text(f"❌ فشل المحرك في إيجاد الصور. تواصل مع {ADMIN_HANDLE}", chat_id, status_msg.message_id)
+            
+    except Exception as e:
+        bot.send_message(chat_id, "⚠️ حدث خطأ في إدخال البيانات. يرجى المحاولة مرة أخرى.")
+
+def run_engine_logic(sample_url, start, end):
     base_part = re.sub(r'/(?:chapter-)?\d+$', '', sample_url.strip().rstrip('/'))
     is_azora = "azoramoon" in sample_url
     pdf_files = []
 
     for i in range(int(start), int(end) + 1):
         target_url = f"{base_part}/chapter-{i}" if is_azora else f"{base_part}/{i}"
-        imgs = fetch_chapter_with_playwright(target_url)
-        
+        imgs = fetch_images(target_url)
         if imgs:
             fname = f"Chapter_{i}.pdf"
             imgs[0].save(fname, save_all=True, append_images=imgs[1:], format='PDF')
@@ -62,34 +159,14 @@ def process_engine(sample_url, start, end):
             
     if not pdf_files: return None
     
-    zip_name = f"Engine_Manga_{int(time.time())}.zip"
+    zip_name = f"Manga_Pack_{int(time.time())}.zip"
     with zipfile.ZipFile(zip_name, 'w') as zipf:
         for f in pdf_files:
             zipf.write(f); os.remove(f)
     return zip_name
 
-@bot.message_handler(func=lambda m: True)
-def handle_msg(message):
-    try:
-        url, range_part = message.text.strip().split(' ')
-        start, end = range_part.split('-')
-        
-        status = bot.reply_to(message, "🚀 جاري تشغيل محرك المحاكاة... قد يستغرق وقتاً أطول لتجاوز الحماية.")
-        zip_path = process_engine(url, start, end)
-        
-        if zip_path:
-            with open(zip_path, 'rb') as f:
-                bot.send_document(message.chat.id, f, caption=f"✅ المحرك نجح في سحب الفصول {start}-{end}")
-            os.remove(zip_path)
-            bot.delete_message(message.chat.id, status.message_id)
-        else:
-            bot.edit_message_text("❌ المحرك لم يجد صوراً. قد تكون الحماية متطورة جداً.", message.chat.id, status.message_id)
-    except:
-        bot.reply_to(message, "⚠️ التنسيق: الرابط مسافة 1-5")
-
-# تشغيل البوت والسيرفر
+# --- تشغيل التطبيق ---
 if __name__ == "__main__":
-    # تثبيت المتصفحات المطلوبة لـ Playwright عند التشغيل الأول
     os.system("playwright install chromium")
     threading.Thread(target=lambda: bot.infinity_polling(), daemon=True).start()
     app.run(host="0.0.0.0", port=7860)