Create app.py

app.py

@@ -0,0 +1,82 @@
+import streamlit as st
+from deepface import DeepFace
+import requests
+from PIL import Image
+from gtts import gTTS
+import os
+
+KNOWN_FOLDER = "known_faces"
+ESP32_SERVER_URL = "https://esp32-upload-server.onrender.com"
+# FLASK_UPLOAD_URL = "https://flask-upload-pzch.onrender.com/upload"
+
+# Set Streamlit layout
+st.set_page_config(page_title="Second Eye - Face Recognition", layout="centered")
+
+# Sidebar navigation
+st.sidebar.title("Navigation")
+page = st.sidebar.radio("Go to", ["Face Recognition"])
+
+# Fetch latest image from ESP32 server
+def get_latest_image():
+    try:
+        r = requests.get(f"{ESP32_SERVER_URL}/latest")
+        if r.status_code != 200:
+            return None
+        filename = r.json()["filename"]
+        return f"{ESP32_SERVER_URL}/uploads/{filename}"
+    except:
+        return None
+
+# Face detection check
+def is_face_detected(image_path):
+    try:
+        faces = DeepFace.extract_faces(img_path=image_path, enforce_detection=True)
+        return len(faces) > 0
+    except:
+        return False
+
+# Face comparison using Facenet
+def compare_with_known_faces(unknown_img_path):
+    for filename in os.listdir(KNOWN_FOLDER):
+        known_img_path = os.path.join(KNOWN_FOLDER, filename)
+        try:
+            result = DeepFace.verify(
+                img1_path=unknown_img_path,
+                img2_path=known_img_path,
+                model_name="Facenet",
+                enforce_detection=True
+            )
+            if result["verified"]:
+                return filename.split('.')[0]
+        except Exception as e:
+            print(f"Comparison failed with {filename}: {e}")
+    return None
+
+# -------------- PAGE 1: Face Recognition --------------
+if page == "Face Recognition":
+    st.title("ESP32-CAM Face Recognition")
+
+    if st.button("Check for New Image"):
+        image_url = get_latest_image()
+        if image_url:
+            st.image(image_url, caption="Captured Image", use_container_width=True)
+            response = requests.get(image_url)
+            with open("latest.jpg", "wb") as f:
+                f.write(response.content)
+
+            if is_face_detected("latest.jpg"):
+                match = compare_with_known_faces("latest.jpg")
+                if match:
+                    st.success(f"✅ Match found: {match}")
+                    tts = gTTS(f"Match found: {match}")
+                else:
+                    st.error("❌ No match found")
+                    tts = gTTS("No match found")
+            else:
+                st.warning("😕 No face detected in the image.")
+                tts = gTTS("No face detected")
+
+            tts.save("result.mp3")
+            st.audio("result.mp3", autoplay=True)
+        else:
+            st.warning("No image found on ESP32 server.")