Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import cv2
 import tempfile
+import os
 import smtplib
 from email.mime.text import MIMEText
 from ultralytics import YOLO
@@ -14,14 +15,18 @@ model_path = hf_hub_download(
 
 model = YOLO(model_path)
 
-# -------- EMAIL ALERT --------
+# -------- EMAIL ALERT (SECURE) --------
 def send_alert():
-    sender = "your_email@gmail.com"
-    password = "your_app_password"
-    receiver = "receiver_email@gmail.com"
+    sender = os.getenv("EMAIL_USER")
+    password = os.getenv("EMAIL_PASS")
+    receiver = os.getenv("EMAIL_TO")
+
+    if not sender or not password or not receiver:
+        print("Email credentials not set.")
+        return
 
     msg = MIMEText("⚠️ Alert: Person detected!")
-    msg["Subject"] = "YOLO Alert"
+    msg["Subject"] = "YOLO Surveillance Alert"
     msg["From"] = sender
     msg["To"] = receiver
 
@@ -47,7 +52,7 @@ def detect_image(image):
 
     return annotated
 
-# -------- VIDEO DETECTION --------
+# -------- VIDEO DETECTION + TRACKING --------
 def detect_video(video):
     cap = cv2.VideoCapture(video)
 
@@ -61,6 +66,8 @@ def detect_video(video):
 
     out = cv2.VideoWriter(output_path, fourcc, fps, (w, h))
 
+    alert_sent = False
+
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
@@ -69,11 +76,13 @@ def detect_video(video):
         results = model.track(frame, persist=True)
         annotated = results[0].plot()
 
-        for box in results[0].boxes:
-            cls = int(box.cls[0])
-            if model.names[cls] == "person":
-                send_alert()
-                break
+        if not alert_sent:
+            for box in results[0].boxes:
+                cls = int(box.cls[0])
+                if model.names[cls] == "person":
+                    send_alert()
+                    alert_sent = True
+                    break
 
         out.write(annotated)
 
@@ -82,9 +91,9 @@ def detect_video(video):
 
     return output_path
 
-# -------- WEBCAM --------
-def webcam_detect(frame):
-    results = model.track(frame, persist=True)
+# -------- WEBCAM (SAFE VERSION FOR SPACES) --------
+def webcam_detect(image):
+    results = model(image)
     annotated = results[0].plot()
 
     for box in results[0].boxes:
@@ -97,23 +106,27 @@ def webcam_detect(frame):
 
 # -------- UI --------
 with gr.Blocks() as demo:
-    gr.Markdown("# 🔐 YOLO Surveillance (Hugging Face Model)")
+    gr.Markdown("# 🔐 Smart Surveillance System (YOLO + Alerts)")
 
-    with gr.Tab("Image"):
+    # IMAGE TAB
+    with gr.Tab("📷 Image"):
         img_in = gr.Image(type="numpy")
         img_out = gr.Image()
-        btn1 = gr.Button("Detect")
+        btn1 = gr.Button("Run Detection")
         btn1.click(detect_image, inputs=img_in, outputs=img_out)
 
-    with gr.Tab("Video"):
+    # VIDEO TAB
+    with gr.Tab("🎥 Video"):
         vid_in = gr.Video()
         vid_out = gr.Video()
-        btn2 = gr.Button("Detect + Track")
+        btn2 = gr.Button("Run Detection + Tracking")
         btn2.click(detect_video, inputs=vid_in, outputs=vid_out)
 
-    with gr.Tab("Webcam"):
-        cam_in = gr.Image(source="webcam", streaming=True)
+    # WEBCAM TAB (Stable Version)
+    with gr.Tab("📡 Webcam"):
+        cam_in = gr.Image(sources=["webcam"], type="numpy")
         cam_out = gr.Image()
-        cam_in.stream(webcam_detect, inputs=cam_in, outputs=cam_out)
+        btn3 = gr.Button("Capture & Detect")
+        btn3.click(webcam_detect, inputs=cam_in, outputs=cam_out)
 
 demo.launch()