fix: add atexit and signal handlers for ZMQ cleanup

- Register terminate_zmq() with atexit to ensure cleanup on normal exit
- Add signal handlers for SIGINT (Ctrl+C) and SIGTERM
- Improve terminate_zmq() with better logging and error handling
- Clear zmq_ports dict after cleanup to prevent double-cleanup

Author: Sahil-u07

concore.py

@@ -4,6 +4,8 @@
 import sys
 import re
 import zmq # Added for ZeroMQ
+import atexit
+import signal
 
 # if windows, create script to kill this process 
 # because batch files don't provide easy way to know pid of last command
@@ -92,12 +94,31 @@ def init_zmq_port(port_name, port_type, address, socket_type_str):
         print(f"An unexpected error occurred during ZMQ port initialization for {port_name}: {e}")
 
 def terminate_zmq():
-    for port in zmq_ports.values():
+    """Clean up all ZMQ sockets and contexts before exit."""
+    if not zmq_ports:
+        return  # No ports to clean up
+    
+    print("\nCleaning up ZMQ resources...")
+    for port_name, port in zmq_ports.items():
         try:
             port.socket.close()
             port.context.term()
+            print(f"Closed ZMQ port: {port_name}")
         except Exception as e:
-            print(f"Error while terminating ZMQ port {port.address}: {e}")
+            print(f"Error while terminating ZMQ port {port_name} ({port.address}): {e}")
+    zmq_ports.clear()
+
+def signal_handler(sig, frame):
+    """Handle interrupt signals gracefully."""
+    print(f"\nReceived signal {sig}, shutting down gracefully...")
+    terminate_zmq()
+    sys.exit(0)
+
+# Register cleanup handlers
+atexit.register(terminate_zmq)
+signal.signal(signal.SIGINT, signal_handler)   # Handle Ctrl+C
+signal.signal(signal.SIGTERM, signal_handler)  # Handle termination
+
 # --- ZeroMQ Integration End ---
 
 # ===================================================================