Update app.py

app.py

@@ -16,7 +16,7 @@ smtp_server = "smtp.gmail.com"
 subject = "Accident detected"
 
 
-def send_email(accident_type):
+def send_email(accident_type,image):
     body = accident_type
 
     msg = MIMEMultipart()
@@ -26,11 +26,11 @@ def send_email(accident_type):
 
     msg.attach(MIMEText(body, 'plain'))
 
-    fullFileName = "./result/res.png"
-    attachment = open(fullFileName, 'rb')
+    is_success, buffer = cv2.imencode(".jpg", image)
+    attachment = buffer.tobytes()
 
     attachment_package = MIMEBase('application', 'octet-stream')
-    attachment_package.set_payload((attachment).read())
+    attachment_package.set_payload((attachment)
     encoders.encode_base64(attachment_package)
     attachment_package.add_header('Content-Disposition', "attachment; filename= " + filename)
     msg.attach(attachment_package)
@@ -88,8 +88,7 @@ def image_predict(image):
     annotated_frame = results[0].plot()
     if len(res) >0:
         annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
-        cv2.imwrite("./result/res.png", annotated_frame_bgr)
-        send_email(res)
+        send_email(res, annotated_frame_bgr)
         return (res, annotated_frame)
 
     return ("No Class Detected", None)
@@ -119,8 +118,7 @@ def extract_frames(video):
 
             if len(res) >0:
                 annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
-                cv2.imwrite("./result/res.png", annotated_frame_bgr)
-                send_email(res)
+                send_email(res, annotated_frame_bgr)
                 return (res, annotated_frame)
             
         frame_no += 1  # Increment frame number