delete redudant

grpc_server.py

@@ -1,178 +0,0 @@
-import sys
-import tempfile
-import grpc
-from concurrent import futures
-import os
-from deepface import DeepFace
-
-
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), "proto"))
-from proto import face_recognition_pb2, face_recognition_pb2_grpc
-
-
-EMPLOYEE_DB_PATH = "data/employees"
-
-
-class FaceRecognitionServicer(face_recognition_pb2_grpc.FaceRecognitionServiceServicer):
-
-    def Recognize(self, request, context):
-        print("[INFO] Received recognition request...")
-        temp_file = None
-        try:
-            # Save incoming image to a temporary file
-            with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
-                temp_file = tmp.name
-                tmp.write(request.image)
-
-            print("[INFO] Running DeepFace search...")
-            result = DeepFace.find(
-                img_path=temp_file,
-                db_path=EMPLOYEE_DB_PATH,
-                detector_backend="opencv",
-                enforce_detection=False
-            )
-
-            if isinstance(result, list):
-                result = result[0]
-
-            if result is None or len(result) == 0:
-                print("[INFO] No match found.")
-                return face_recognition_pb2.FaceResponse(
-                    name="Unknown",
-                    confidence=0.0,
-                    image=b""
-                )
-
-            best_row = result.iloc[0]
-            matched_image_path = best_row["identity"]
-            name = os.path.splitext(os.path.basename(matched_image_path))[0]
-
-            # Convert distance to rough confidence
-            distance = best_row.get("VGG-Face_cosine", 0.3)
-            confidence = float(max(0.0, 1.0 - distance))
-
-            # Read matched image bytes
-            with open(matched_image_path, "rb") as f:
-                matched_image_bytes = f.read()
-
-            print(f"[INFO] Match: {name}, confidence: {confidence:.2f}")
-            return face_recognition_pb2.FaceResponse(
-                name=name,
-                confidence=confidence,
-                image=matched_image_bytes
-            )
-
-        except Exception as e:
-            print("[ERROR] DeepFace failed:", e)
-            return face_recognition_pb2.FaceResponse(
-                name="Error",
-                confidence=0.0,
-                image=b""
-            )
-
-        finally:
-            if temp_file and os.path.exists(temp_file):
-                try:
-                    os.remove(temp_file)
-                except Exception as cleanup_error:
-                    print(
-                        f"[WARN] Could not delete temp file {temp_file}: {cleanup_error}")
-
-    def EnrollFace(self, request, context):
-        """
-        Handles enrollment of a new employee.
-        Saves the uploaded image in EMPLOYEE_DB_PATH with the employee's name.
-        """
-        try:
-            name = request.name
-            image_bytes = request.image
-
-            if not name or not image_bytes:
-                return face_recognition_pb2.EnrollFaceResponse(
-                    success=False,
-                    message="Name or image missing"
-                )
-
-            os.makedirs(EMPLOYEE_DB_PATH, exist_ok=True)
-            file_path = os.path.join(EMPLOYEE_DB_PATH, f"{name}.jpg")
-
-            with open(file_path, "wb") as f:
-                f.write(image_bytes)
-
-            print(f"[INFO] Enrolled employee: {name}, saved at {file_path}")
-
-            return face_recognition_pb2.EnrollFaceResponse(
-                success=True,
-                message="Enrollment successful"
-            )
-
-        except Exception as e:
-            print(f"[ERROR] Enrollment failed: {e}")
-            return face_recognition_pb2.EnrollFaceResponse(
-                success=False,
-                message=str(e)
-            )
-
-    def GetAllEmployees(self, request, context):
-        try:
-            employee_files = [
-                f for f in os.listdir(EMPLOYEE_DB_PATH)
-                if f.lower().endswith((".jpg", ".png", ".jpeg"))
-            ]
-
-            employees = []
-            for file in employee_files:
-                path = os.path.join(EMPLOYEE_DB_PATH, file)
-                name = os.path.splitext(file)[0]
-
-                with open(path, "rb") as f:
-                    image_bytes = f.read()
-
-                employees.append(face_recognition_pb2.Employee(
-                    name=name,
-                    image=image_bytes
-                ))
-
-            return face_recognition_pb2.EmployeeListResponse(
-                employees=employees
-            )
-        except Exception as e:
-            print("[ERROR] Failed to list employees:", e)
-            return face_recognition_pb2.EmployeeListResponse(employees=[])
-
-    def DeleteEmployee(self, request, context):
-        try:
-            found = False
-            for file in os.listdir(EMPLOYEE_DB_PATH):
-                if os.path.splitext(file)[0] == request.name:
-                    os.remove(os.path.join(EMPLOYEE_DB_PATH, file))
-                    found = True
-                    break
-            message = "Deleted successfully" if found else "Employee not found"
-            return face_recognition_pb2.DeleteEmployeeResponse(
-                success=found,
-                message=message
-            )
-        except Exception as e:
-            print("[ERROR] Failed to delete employee:", e)
-            return face_recognition_pb2.DeleteEmployeeResponse(
-                success=False,
-                message=str(e)
-            )
-
-
-def serve():
-    server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
-    face_recognition_pb2_grpc.add_FaceRecognitionServiceServicer_to_server(
-        FaceRecognitionServicer(), server
-    )
-
-    server.add_insecure_port("[::]:50051")
-    server.start()
-    print("[INFO] gRPC Face Recognition server running on port 50051")
-    server.wait_for_termination()
-
-
-if __name__ == "__main__":
-    os.makedirs(EMPLOYEE_DB_PATH, exist_ok=True)
-    serve()