YACWC

snap_version.py

@@ -1,13 +1,15 @@
 from common_code.settings import get_logger
-from utils import run_model, start_snap_manager
+import logging
+logger = get_logger('live_inference',
+                    file_path='/var/log/live_inference.log',
+                    stdout=True, level = logging.WARN)
+
+from utils import run_model, start_snap_manager, kafka_poster
 import threading
 import queue
 import multiprocessing
 import time
 
-logger = get_logger('live_inference',
-                    file_path='/var/log/live_inference.log',
-                    stdout=True)
 
 all_cameras_config = {
     "camera_sidefeeder": {
@@ -63,14 +65,22 @@ def start_system():
     # Create shared multiprocessing queue for model
     img_scoring_queue = multiprocessing.Queue(maxsize=len(cameras_config) * 2)
 
+
+    kafka_results_queue = multiprocessing.Queue()
     # Create threading queues for each camera
     for cam, details in cameras_config.items():
         details['msg_queue'] = queue.Queue(maxsize=1)
         details['img_scoring_queue'] = img_scoring_queue
+        details['kafka_queue'] = kafka_results_queue
 
     # Start model process
+    kafka_process = multiprocessing.Process(target=kafka_poster, args=(kafka_results_queue, ))
+    kafka_process.daemon = True
+    kafka_process.start()
+    logger.info('Started kafka poster process')
+    
     model_process = multiprocessing.Process(target=run_model,
-                                            args=(img_scoring_queue, ))
+                                            args=(img_scoring_queue,kafka_results_queue ))
     model_process.daemon = True
     model_process.start()
     logger.info('Started model process')