Update app.py


app.py

@@ -34,6 +34,66 @@ def main():
 
     # Load Haar Cascade
     face_cascade = load_cascade()
+    import streamlit as st
+import cv2
+import numpy as np
+from PIL import Image
+import os
+
+# Load Haar Cascade
+@st.cache_resource
+def load_cascade():
+    face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
+    if face_cascade.empty():
+        raise Exception("Haar Cascade file not loaded!")
+    return face_cascade
+
+# Detect faces
+def detect_faces(image, face_cascade):
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.3, minNeighbors=5)
+    for (x, y, w, h) in faces:
+        cv2.rectangle(image, (x, y), (x + w, y + h), (255, 0, 0), 2)
+    return image, len(faces)
+
+# Streamlit app
+def main():
+    st.title("Face Detection App")
+    st.write("Upload an image, and the app will detect faces!")
+
+    # Load Haar Cascade
+    face_cascade = load_cascade()
+
+    # File uploader
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+    if uploaded_file is not None:
+        # Save the uploaded image temporarily
+        img_path = '/tmp/uploaded_image.jpg'
+        with open(img_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+
+        # Read and process the image
+        image = np.array(Image.open(uploaded_file))
+        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+
+        # Detect faces
+        result_image, face_count = detect_faces(image, face_cascade)
+
+        # Convert BGR to RGB for display
+        result_image = cv2.cvtColor(result_image, cv2.COLOR_BGR2RGB)
+
+        # Display the image with face detection
+        st.image(result_image, caption=f"Detected {face_count} face(s).", use_column_width=True)
+
+        # Show success or failure message based on face count
+        if face_count > 0:
+            st.success("Face Detection Successful!")
+        else:
+            st.error("No faces detected. Please try again with a clearer image.")
+
+if __name__ == "__main__":
+    main()
+
 
     # File uploader
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])