Update app.py

app.py

@@ -7,6 +7,7 @@ from email.mime.multipart import MIMEMultipart
 from email.mime.base import MIMEBase
 from email import encoders
 import smtplib
+import os
 
 sender_email = "sudipdatta2002@gmail.com"
 receiver_email = "2021csb006.sudip@students.iiests.ac.in"
@@ -56,6 +57,15 @@ def send_email(accident_type):
 
     gmail_server.quit()
 
+def send_sms(accident_type):
+    print(os.environ.get("account_sid"))
+    print(os.environ.get("auth_token"))
+    print(os.environ.get("my_twilio_number"))
+    print(os.environ.get("receiver_number"))
+    print(os.environ.get("sender_email"))
+    print(os.environ.get("receiver_email"))
+    print(os.environ.get("sender_password"))
+
 
 def check_acc(box):
     # return format(box.cls)
@@ -65,18 +75,25 @@ def check_acc(box):
     for index in res_index_list:
         if index ==3:
             result = "car car accident detected"
+            break
         elif index==4:
             result = "car bike accident detected"
+            break
         elif index==5:
             result = "car person accident detected"
+            break
         elif index==6:
             result = "bike bike accident detected"
+            break
         elif index==7:
             result = "bike person accident detected"
+            break
         elif index ==8:
             result = "car object accident detected"
+            break
         elif index == 9:
             result = "bike object accident detected"
+            break
     
     return result
 
@@ -95,9 +112,13 @@ def image_predict(image):
     # if res == "car car accident detected":
     #     cv2.imwrite("./result/res.png", render)
     # print(render)
-    if res == "car car accident detected":
-        render.save("./result/res.png")
-        send_email(res)
+    # if res == "car car accident detected":
+    #     render.save("./result/res.png")
+    #     # send_email(res)
+    #     send_sms
+    if len(res) >0:
+        # render.save("./result/res.png")
+        send_sms(res)
 
     return (res, render)