[feature]Add log info.

consumer/asr.py

@@ -12,6 +12,15 @@ from pydantic import BaseModel, Field, ValidationError, model_validator
 from .base import BasicMessageReceiver, BasicMessageSender
 
 
+def setup_logger():
+    logger = logging.getLogger(__name__)
+    logging.basicConfig(level=logging.INFO)
+    return logger
+
+
+logger = setup_logger()
+
+
 class TranscribeInputMessage(BaseModel):
     uuid: str = Field(title='Request Unique Id.')
     audio_file: str
@@ -46,9 +55,11 @@ class TranscribeConsumer(BasicMessageReceiver):
 
         self.setup_consume_parameters()
 
+        logger.info('Loading model...')
         model_size = os.environ.get('WHISPER-MODEL-SIZE', 'large-v3')
         # Run on GPU with FP16
         self.asr_model = WhisperModel(model_size, device='cuda', compute_type='float16')
+        logger.info('Load model finished.')
 
     def setup_consume_parameters(self):
         self.declare_exchange(self.exchange_name)
@@ -65,6 +76,10 @@ class TranscribeConsumer(BasicMessageReceiver):
             body=message,
             headers=None
         )
+        logger.info(f'{"-" * 80}')
+        logger.info(f"Send message to Exchange: {self.exchange_name}, Routing-key: {routing_key}, \n"
+                    f"Messgae body: {message}")
+        logger.info(f'{"-" * 80}')
 
     def send_success_message(self, uuid: str, transcribed_text):
         message = TranscribeOutputMessage(uuid=uuid, if_success=True, msg='Transcribe finished.',
@@ -76,6 +91,7 @@ class TranscribeConsumer(BasicMessageReceiver):
         self.send_message(message.model_dump_json())
 
     def consume(self, channel, method, properties, message):
+        logger.info(f'Recevied a message: {message}')
         try:
             body = self.decode_message(message)
         except json.JSONDecodeError as exc: