Update chatbot/__main__.py

chatbot/__main__.py

@@ -39,6 +39,33 @@ logging.getLogger("pyrogram.syncer").setLevel(logging.WARNING)
 logging.getLogger("pyrogram.client").setLevel(logging.WARNING)
 loop = asyncio.get_event_loop()
 
+async def premium_expiry_monitor(client):
+    while True:
+        try:
+            now = dt.now()
+            expired = db.user_premium.find({
+                "is_premium": True,
+                "premium_expiry": {"$lt": now}
+            })
+            
+            async for user in expired:
+                await db.user_premium.update_one(
+                    {"user_id": user["user_id"]},
+                    {"$set": {"is_premium": False}}
+                )
+                with suppress(Exception):
+                    await client.send_message(
+                        user["user_id"],
+                        "⚠️ Your premium subscription has expired\n"
+                        "You've been downgraded to free tier (5 images/day)\n\n"
+                    )
+            
+            await asyncio.sleep(3600)
+            
+        except Exception as e:
+            LOGS.error(f"Premium monitor error: {e}")
+            await asyncio.sleep(600)
+
 async def process_unfreezes(client):
     while True:
         try:
@@ -83,6 +110,7 @@ async def main():
         randydev = Randydev(loop=loop)
         
         await randydev.start()
+        asyncio.create_task(premium_expiry_monitor(randydev))
         asyncio.create_task(process_unfreezes(randydev))
         LOGS.info("Application startup complete")