Update app.py

app.py

@@ -1,6 +1,55 @@
+import os
+import cv2
+import smtplib
 import gradio as gr
+from email import encoders
 from ultralytics import YOLO
-import cv2
+from email.mime.text import MIMEText
+from email.mime.base import MIMEBase
+from email.mime.multipart import MIMEMultipart
+
+sender_email = os.environ.get("sender_email")
+receiver_email = os.environ.get("receiver_email")
+sender_password = os.environ.get("sender_password")
+smtp_port = 8080
+smtp_server = "smtp.gmail.com"
+subject = "Accident detected"
+
+
+def send_email(accident_type):
+    body = accident_type
+
+    msg = MIMEMultipart()
+    msg['From'] = sender_email
+    msg['To'] = receiver_email
+    msg['Subject'] = subject
+
+    msg.attach(MIMEText(body, 'plain'))
+
+    filename = "res.png"
+    folder = "./result/"
+    fullFileName = folder + filename
+    attachment = open(fullFileName, 'rb')
+
+    attachment_package = MIMEBase('application', 'octet-stream')
+    attachment_package.set_payload((attachment).read())
+    encoders.encode_base64(attachment_package)
+    attachment_package.add_header('Content-Disposition', "attachment; filename= " + filename)
+    msg.attach(attachment_package)
+
+    text = msg.as_string()
+
+    print("Connecting to server")
+    gmail_server = smtplib.SMTP(smtp_server, smtp_port)
+    gmail_server.starttls()
+    gmail_server.login(sender_email, sender_password)
+    print("Successfully Connected to Server")
+
+    print("Sending email to ", receiver_email)
+    gmail_server.sendmail(sender_email, receiver_email, text)
+    print("Email sent to ", receiver_email)
+
+    gmail_server.quit()
 
 def check_acc(box):
     res_index_list = box.cls.tolist()
@@ -35,11 +84,13 @@ def image_predict(image):
     res = ""
     model_path = "best.pt"
     model = YOLO(model_path)
-    results = model.predict(image,conf = 0.8,iou = 0.3,imgsz = 512)
+    results = model.predict(image,conf = 0.6,iou = 0.3,imgsz = 512)
     box = results[0].boxes
     res = check_acc(box)
     annotated_frame = results[0].plot()
     if len(res) >0:
+        annotated_frame.save("./result/res.png")
+        send_email(res)
         return (res, annotated_frame)
 
     return ("No Class Detected", None)
@@ -62,12 +113,14 @@ def extract_frames(video):
         if (frame_no % (int(fps / nof))) == 0:
             model_path = "best.pt"
             model = YOLO(model_path)
-            results = model.predict(image,conf = 0.8,iou = 0.3,imgsz = 512)  
+            results = model.predict(image,conf = 0.6,iou = 0.3,imgsz = 512)  
             box = results[0].boxes
             res = check_acc(box) 
             annotated_frame = results[0].plot()
 
             if len(res) >0:
+                annotated_frame.save("./result/res.png")
+                send_email(res)
                 return (res, annotated_frame)
             
         frame_no += 1  # Increment frame number