Update app.py

app.py

@@ -9,10 +9,25 @@ model = YOLO("best.pt")  # Ensure the path to your model is correct
 # Set the title of the app
 st.title("Live Fire Detection App")
 
-cap = cv2.VideoCapture(0)  # 0 for default camera
+def capture_video():
+    """Attempts to capture video from the default camera (index 0).
 
-if not cap.isOpened():
-    print("Error: Could not open video capture device.")
+    Returns:
+        cv2.VideoCapture: The video capture object if successful, None otherwise.
+    """
+
+    cap = cv2.VideoCapture(0)
+
+    if not cap.isOpened():
+        print("Error: Could not open video capture device.")
+        return None
+
+    return cap
+
+# Capture video
+cap = capture_video()
+if cap is None:
+    # Handle capture failure gracefully
     exit()
 
 while True:
@@ -28,5 +43,6 @@ while True:
     if cv2.waitKey(1) & 0xFF == ord('q'):
         break
 
+# Release resources
 cap.release()
 cv2.destroyAllWindows()