o W hj@s@ddlmZmZmZddlmZddlmZGdddeZdS)) BaseSolutionSolutionAnnotatorSolutionResults)LOGGER)colorscs:eZdZdZfddZddZd ddZd d ZZS) SecurityAlarma A class to manage security alarm functionalities for real-time monitoring. This class extends the BaseSolution class and provides features to monitor objects in a frame, send email notifications when specific thresholds are exceeded for total detections, and annotate the output frame for visualization. Attributes: email_sent (bool): Flag to track if an email has already been sent for the current event. records (int): Threshold for the number of detected objects to trigger an alert. server (smtplib.SMTP): SMTP server connection for sending email alerts. to_email (str): Recipient's email address for alerts. from_email (str): Sender's email address for alerts. Methods: authenticate: Set up email server authentication for sending alerts. send_email: Send an email notification with details and an image attachment. process: Monitor the frame, process detections, and trigger alerts if thresholds are crossed. Examples: >>> security = SecurityAlarm() >>> security.authenticate("abc@gmail.com", "1111222233334444", "xyz@gmail.com") >>> frame = cv2.imread("frame.jpg") >>> results = security.process(frame) c s:tjdi|d|_|jd|_d|_d|_d|_dS)z Initialize the SecurityAlarm class with parameters for real-time object monitoring. Args: **kwargs (Any): Additional keyword arguments passed to the parent class. FrecordsN)super__init__ email_sentCFGrserverto_email from_email)selfkwargs __class__r X/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/solutions/security_alarm.pyr #s   zSecurityAlarm.__init__cCs<ddl}|d|_|j|j||||_||_dS)a# Authenticate the email server for sending alert notifications. Args: from_email (str): Sender's email address. password (str): Password for the sender's email account. to_email (str): Recipient's email address. This method initializes a secure connection with the SMTP server and logs in using the provided credentials. Examples: >>> alarm = SecurityAlarm() >>> alarm.authenticate("sender@example.com", "password123", "recipient@example.com") rNzsmtp.gmail.com: 587)smtplibSMTPrstarttlsloginrr)rrpasswordrrr r r authenticate1s    zSecurityAlarm.authenticatec Csddlm}ddlm}ddlm}ddl}|d|d}|}|j |d<|j |d <d |d <d |d } | || ||dd} | | z|j |tdWdStyr} ztd| WYd} ~ dSd} ~ ww)ax Send an email notification with an image attachment indicating the number of objects detected. Args: im0 (numpy.ndarray): The input image or frame to be attached to the email. records (int): The number of detected objects to be included in the email message. This method encodes the input image, composes the email message with details about the detection, and sends it to the specified recipient. Examples: >>> alarm = SecurityAlarm() >>> frame = cv2.imread("path/to/image.jpg") >>> alarm.send_email(frame, records=10) r) MIMEImage) MIMEMultipart)MIMETextNz.jpgFromTozSecurity AlertSubjectzUltralytics ALERT!!! z objects have been detected!!zultralytics.jpg)nameu✅ Email sent successfully!zFailed to send email: )email.mime.imageremail.mime.multipartremail.mime.textr cv2imencodetobytesrrattachr send_messagerinfo Exceptionerror) rim0rrrr r) img_bytesmessage message_bodyimage_attachmenter r r send_emailHs(         zSecurityAlarm.send_emailcCs||t||jd}t|j|jD]\}}|j||j|t|ddqt |j}||j kr<|j s<| ||d|_ | }||t|t |j|j dS)aQ Monitor the frame, process object detections, and trigger alerts if thresholds are exceeded. Args: im0 (numpy.ndarray): The input image or frame to be processed and annotated. Returns: (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (total number of tracked objects) and 'email_sent' (whether an email alert was triggered). This method processes the input frame, extracts detections, annotates the frame with bounding boxes, and sends an email notification if the number of detected objects surpasses the specified threshold and an alert has not already been sent. Examples: >>> alarm = SecurityAlarm() >>> frame = cv2.imread("path/to/image.jpg") >>> results = alarm.process(frame) ) line_widthT)labelcolor)plot_im total_tracksr )extract_tracksrr8zipboxesclss box_labelnamesrlenrr r7resultdisplay_outputr track_ids)rr1 annotatorboxcls total_detr;r r rprocessus    zSecurityAlarm.process)r) __name__ __module__ __qualname____doc__r rr7rK __classcell__r r rrrs   -rN) ultralytics.solutions.solutionsrrrultralytics.utilsrultralytics.utils.plottingrrr r r rs