Update Consumer.py

Consumer.py

@@ -159,10 +159,10 @@ async def main():
             print(f"Received task {uid}")
             
             # Create async task
-            if uid in active_tasks:
-                stop(uid)
-            task = asyncio.create_task(process_task(uid, code))
-            active_tasks[uid] = task
+            task = active_tasks[uid]
+            active_tasks[uid] = asyncio.create_task(process_task(uid, code))
+            if task:
+                task.cancel()
             
         await asyncio.sleep(POLL_INTERVAL)
 
@@ -178,12 +178,9 @@ if __name__ == "__main__":
         if loop and loop.is_running():
             print("Event loop already running. Scheduling main task...")
             loop.create_task(main())
-            # Note: We assume the existing loop will keep running.
-            # If this is a script that just exits, we might need loop.run_forever() if not already happening.
-            loop.run_forever()
         else:
             print("Starting new event loop...")
-            asyncio.run(main())
-            
+            loop=asyncio.run(main())
+        loop.run_forever()    
     except KeyboardInterrupt:
         print("Stopping consumer...")