import grpc
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "proto"))
from proto import face_recognition_pb2, face_recognition_pb2_grpc

# Load the test image
with open("sample_aladdin.jpg", "rb") as f:
    image_bytes = f.read()

# Connect to gRPC server
channel = grpc.insecure_channel("localhost:50051")
stub = face_recognition_pb2_grpc.FaceRecognitionServiceStub(channel)

# Create request with model_name = "Facenet"
request = face_recognition_pb2.FaceRequest(
    image=image_bytes,
    model_name="OpenFace"  # specify the model here
)

# Call the Recognize RPC
response = stub.Recognize(request)

# Print results
print("Name:", response.name)
print("Confidence:", response.confidence)