import gradio as gr
import face_recognition
import cv2
import numpy as np
from PIL import Image
import pickle
import firebase_admin
from firebase_admin import credentials
from firebase_admin import db
from datetime import datetime
from firebase_admin import storage

# Initialize Firebase
cred = credentials.Certificate("serviceAccountKey.json")  # Update with your credentials path
firebase_app = firebase_admin.initialize_app(cred, {
    'databaseURL': 'https://faceantendancerealtime-default-rtdb.firebaseio.com/',
    'storageBucket': 'faceantendancerealtime.appspot.com'
})
bucket = storage.bucket()

# Function to download face encodings from Firebase Storage
def download_encodings():
    blob = bucket.blob('EncodeFile.p')
    blob.download_to_filename('EncodeFile.p')
    with open('EncodeFile.p', 'rb') as file:
        return pickle.load(file)

encodeListKnownWithIds = download_encodings()
encodeListKnown, studentsIds = encodeListKnownWithIds

def recognize_face(input_image):
    # Convert PIL Image to numpy array
    img = np.array(input_image)
    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)

    # Detect faces and encode
    face_locations = face_recognition.face_locations(img)
    face_encodings = face_recognition.face_encodings(img, face_locations)

    # Initialize the database reference
    ref = db.reference('Students')
    recognized_faces_info = []

    for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
        matches = face_recognition.compare_faces(encodeListKnown, face_encoding)
        name = "Unknown"

        face_distances = face_recognition.face_distance(encodeListKnown, face_encoding)
        best_match_index = np.argmin(face_distances)
        if matches[best_match_index]:
            student_id = studentsIds[best_match_index]
            student_ref = ref.child(student_id)
            student_info = student_ref.get()

            if student_info:
                name = student_info['name']
                # Increment total_attendance
                student_info['total_attendance'] += 1
                # Update last attendance time
                student_info['last_attendance_time'] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                # Write back to the database
                student_ref.update(student_info)
                recognized_faces_info.append(student_info)
            else:
                recognized_faces_info.append({'name': 'Unknown'})
        
        # Draw rectangles around the faces
        cv2.rectangle(img, (left, top), (right, bottom), (0, 0, 255), 2)
        cv2.putText(img, name, (left + 6, bottom - 6), cv2.FONT_HERSHEY_COMPLEX, 0.5, (255, 255, 255), 1)

    # Convert back to PIL Image
    pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
    return pil_img, recognized_faces_info

# Gradio interface
iface = gr.Interface(
    fn=recognize_face,
    inputs=gr.components.Image(source="webcam", type="pil", tool="editor"),
    outputs=[
        gr.components.Image(type="pil"),
        gr.components.JSON(label="Student Information")
    ],
    title="Real-time Face Recognition Attendance System",
    description="Activate your webcam and take a photo to check attendance."
)

if __name__ == "__main__":
    iface.launch()