Ok updated

chatbot/__main__.py

@@ -39,6 +39,17 @@ logging.getLogger("pyrogram.syncer").setLevel(logging.WARNING)
 logging.getLogger("pyrogram.client").setLevel(logging.WARNING)
 loop = asyncio.get_event_loop()
 
+async def startup_tasks(bot_client):
+    task1 = asyncio.create_task(
+        premium_expiry_monitor(bot_client),
+        name="premium_expiry_monitor"
+    )
+    task2 = asyncio.create_task(
+        process_unfreezes(bot_client),
+        name="process_unfreezes"
+    )
+    return [task1, task2]
+
 async def premium_expiry_monitor(client):
     while True:
         try:
@@ -61,7 +72,8 @@ async def premium_expiry_monitor(client):
                     )
             
             await asyncio.sleep(3600)
-            
+        except asyncio.CancelledError:
+            LOGS.info("Premium monitor stopped gracefully")
         except Exception as e:
             LOGS.error(f"Premium monitor error: {e}")
             await asyncio.sleep(600)
@@ -89,7 +101,9 @@ async def process_unfreezes(client):
                     )
             
             await asyncio.sleep(300)
-            
+
+        except asyncio.CancelledError:
+            LOGS.info("Unfreeze monitor stopped gracefully")
         except Exception as e:
             LOGS.error(f"Unfreeze task error: {str(e)}")
             await asyncio.sleep(60)
@@ -110,8 +124,7 @@ async def main():
         randydev = Randydev(loop=loop)
         
         await randydev.start()
-        asyncio.create_task(premium_expiry_monitor(randydev))
-        asyncio.create_task(process_unfreezes(randydev))
+        bg_tasks = await startup_tasks(randydev)
         LOGS.info("Application startup complete")
 
         await idle()
@@ -120,6 +133,9 @@ async def main():
         LOGS.critical(f"Fatal error: {str(e)}")
         traceback.print_exc()
     finally:
+        for task in bg_tasks:
+            task.cancel()
+        await asyncio.gather(*bg_tasks, return_exceptions=True)
         await shutdown(asyncio.get_event_loop())
 
 if __name__ == "__main__":