Merge pull request #3 from SentryCoderDev/SentryCoderDev-patch-FaceRecognition

Feature/FaceRecognition

Author: SentryCoderDev

app.py

@@ -1,15 +1,26 @@
+import logging
+from datetime import datetime
 from fastapi import FastAPI, WebSocket, Request
 from fastapi.responses import HTMLResponse, StreamingResponse
 from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
 import cv2
 import pytesseract
 import threading
 import serial
 import asyncio
+import requests
+import os
+import face_recognition
 
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
+# Log file settings
+logging.basicConfig(filename='access_log.txt', level=logging.INFO, format='%(asctime)s - %(message)s')
+
+# Start cameras and serial port
 # All Cameras
 cameras = [cv2.VideoCapture(i) for i in range(3)]
 ser = serial.Serial('/dev/ttyUSB0', 115200)  # Arduino's serial port
@@ -22,8 +33,24 @@
 }
 
 # Pytesseract configuration
+# Pytesseract config
 pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'  # Path to Tesseract OCR executable
 
+# Add known faces
+known_face_encodings = []
+known_face_names = []
+
+def load_known_faces():
+    known_faces_dir = "known_faces"
+    for filename in os.listdir(known_faces_dir):
+        image_path = os.path.join(known_faces_dir, filename)
+        image = face_recognition.load_image_file(image_path)
+        face_encoding = face_recognition.face_encodings(image)[0]
+        known_face_encodings.append(face_encoding)
+        known_face_names.append(os.path.splitext(filename)[0])
+
+load_known_faces()
+
 def gen_frames(cam_index):
     cam = cameras[cam_index]
     while True:
@@ -37,7 +64,7 @@ def gen_frames(cam_index):
             frame = buffer.tobytes()
             yield (b'--frame\r\n'
                    b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
-
+            
 async def garage_control():
     while True:
         ret, frame = cameras[0].read()
@@ -51,6 +78,34 @@ async def garage_control():
                 ser.write(b'OPEN_GARAGE\n')
                 print("Garage opens for plate 'your_target_plate'")
         await asyncio.sleep(1)
+        
+async def face_recognition_task():
+    while True:
+        ret, frame = cameras[2].read()
+        if ret:
+            hsv_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+            lower_yellow = (20, 100, 100)
+            upper_yellow = (30, 255, 255)
+            mask = cv2.inRange(hsv_frame, lower_yellow, upper_yellow)
+
+            if cv2.countNonZero(mask) == 0:
+                ser.write(b'FACE_RECOGNITION\n')
+
+                rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                face_locations = face_recognition.face_locations(rgb_frame)
+                face_encodings = face_recognition.face_encodings(rgb_frame, face_locations)
+
+                for face_encoding in face_encodings:
+                    matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
+                    name = "Unknown"
+
+                    if True in matches:
+                        first_match_index = matches.index(True)
+                        name = known_face_names[first_match_index]
+
+                    ser.write(f"IDENTIFIED:{name}\n".encode('utf-8'))
+
+        await asyncio.sleep(1)
 
 async def read_from_arduino():
     global result_message
@@ -59,8 +114,23 @@ async def read_from_arduino():
             line = ser.readline().decode('utf-8').strip()
             if line.startswith("RESULT:"):
                 result_message = line.split("RESULT:")[1]
+                log_access("User1", result_message)
+                send_to_nodemcu(result_message)
         await asyncio.sleep(0.1)
 
+def log_access(user, identifier):
+    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    logging.info(f"User: {user}, Identifier: {identifier}, Time: {current_time}")
+
+def send_to_nodemcu(message):
+    try:
+        url = "http://nodemcu-ip-address/announce"
+        data = {"message": message}
+        response = requests.post(url, json=data)
+        print("NodeMCU's Answer:", response.text)
+    except Exception as e:
+        print(f"Sending data to NodeMCU failed: {e}")
+
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
@@ -83,6 +153,11 @@ async def validate_password(user: str, password: str):
         return {"message": "Password Validated"}
     return {"message": "Invalid Password"}, 401
 
+@app.post("/open_garage")
+async def open_garage():
+    ser.write(b'OPEN_GARAGE\n')
+    return {"message": "Garaj Açıldı"}
+
 @app.get("/result")
 async def result():
     global result_message
@@ -98,10 +173,7 @@ async def websocket_endpoint(websocket: WebSocket):
 
 if __name__ == "__main__":
     threading.Thread(target=asyncio.run, args=(garage_control(),)).start()
+    threading.Thread(target=asyncio.run, args=(face_recognition_task(),)).start()
     threading.Thread(target=asyncio.run, args=(read_from_arduino(),)).start()
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)
-
-
-
-