main.py

import os
from typing import Any, Dict, List
import cv2
from src.core.face_predictions_on_image import face_predictions_on_image
from src.core.exceptions.person_not_found_exception import PersonNotFoundException
from src.core.get_faces_from_images import get_faces_from_image_data_model
from src.core.process_images import process_images
from src.core.show_image_with_faces import show_image_with_faces
from src.data_models.image_data_model import ImageDataModel
from src.data_models.images_from_csv import images_from_csv
from src.data_models.person_data_model import PersonDataModel
from src.data_models.persons_from_csv import persons_from_csv
from src.repository import Repository
from src.data_models.face_data_model import FaceDataModel
from src.data_models.person_data_model import PersonDataModel
from fastapi import FastAPI
from typing import List
from fastapi.responses import FileResponse, JSONResponse
from fastapi import FastAPI, HTTPException
from fastapi.responses import JSONResponse
from typing import List
from fastapi.middleware.cors import CORSMiddleware
import uuid


app = FastAPI()
# Enable CORS for all routes
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # Set this to your actual frontend origin in production
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)
repository = Repository()

@app.get("/")
def read_root():
    return {"mi": "ze"}

@app.get("/images", response_model=List[ImageDataModel])
def get_images():
    images = repository.get_images()
    return images

@app.post("/images")
def add_images(images: List[ImageDataModel]):
    repository.add_images(images)
    return {"message": "Image created successfully"}

@app.delete("/images/{image_id}")
def delete_image(image_id: str):
    repository.delete_image(image_id)
    return {"message": "Image deleted successfully"}

@app.put("/images/{image_id}")
def update_image(image_id: str, image: ImageDataModel):
    repository.update_image(image_id, image)
    return {"message": "Image updated successfully"}

@app.get("/faces", response_model=List[FaceDataModel])
def get_faces():
    return repository.get_faces()

@app.get("/persons", response_model=List[PersonDataModel])
def get_persons():
    return repository.get_persons()

@app.get("/persons/{person_id}", response_model=PersonDataModel)
def get_person(person_id: str):
    return repository.get_person(person_id)

@app.post("/persons")
def add_persons(persons_list: List[PersonDataModel]):
    for p in persons_list:
        p.person_id = str(uuid.uuid4())
    repository.add_persons(persons_list)
    return {"message": "Persons created successfully"}

@app.put("/persons/{person_id}")
def update_person(person_id: str, person: PersonDataModel):
    repository.update_person(person_id, person)
    return {"message": "Person updated successfully"}

@app.post("/persons/{person_id}/verified_faces")
def add_verified_faces_to_person(person_id: str, faces: List[FaceDataModel]):
    try:
        person = repository.get_person(person_id)
        repository.add_verified_faces_to_person(person, faces)
        return {"message": "Verified faces added successfully"}
    except PersonNotFoundException as e:
        raise HTTPException(status_code=404, detail=str(e))

@app.delete("/persons/{person_id}/verified_faces")
def remove_verified_faces(person_id: str, faces: List[FaceDataModel]):
    try:
        person = repository.get_person(person_id)
        repository.remove_verified_faces(person, faces)
        return {"message": "Verified faces removed successfully"}
    except PersonNotFoundException as e:
        raise HTTPException(status_code=404, detail=str(e))

@app.post("/faces/{face_id}")
def update_face(face_id: str, face: FaceDataModel):
    repository.update_face(face_id, face)
    return {"message": "Face updated successfully"}

@app.get("/images/media/{image_id}")
def get_image(image_id: str):
    image:ImageDataModel = repository.get_image(image_id)
    return FileResponse(image.get_image_path())



@app.post("/images/{image_id}/predict")
def predict_on_image(image_id: str):
    image:ImageDataModel = repository.get_image(image_id)
    people = repository.get_persons()
    repository.predict_image(image_id=image_id)
    return repository.get_image(image_id=image_id)


@app.post('/update-database')
def update_database():
    repository.update_database()
    return {"message": "Database updated successfully"}














# processed_images = repository.get_images()
# people = repository.get_persons()
# faces:List[FaceDataModel] = repository.get_faces()




# for image in processed_images:
#     image = face_predictions_on_image(image=image, persons=people)
#     show_image_with_faces(image=image)
#     for face in image.faces:
#         if face.certainty > 0.4:
#             repository.update_face(face_id=face.face_id, face=face)
   
        
        



# face = repository.get_faces()[0]
# repository.update_face(face_id=face.face_id, face=face)


#dummy data
            
# images = [ImageDataModel( image_path=r'images\test3.jpg',image_id="test3"),ImageDataModel( image_path=r'images\test_image.png',image_id="test"),ImageDataModel( image_path=r'images\test4.jpg',image_id="test4"),ImageDataModel( image_path=r'images\test_image2.jpeg',image_id="test2"), ImageDataModel( image_path=r'images\test5.jpg',image_id="test5"), ImageDataModel( image_path=r'images\test6.jpg',image_id="test6"), ImageDataModel( image_path=r'images\test7.jpg',image_id="test7"), ImageDataModel( image_path=r'images\test8.jpg',image_id="test8"), ImageDataModel( image_path=r'images\test9.jpg',image_id="test9"), ImageDataModel( image_path=r'images\test10.jpg',image_id="test10"), ImageDataModel( image_path=r'images\test11.jpg',image_id="test11"), ImageDataModel( image_path=r'images\test12.jpg',image_id="test12"), ImageDataModel( image_path=r'images\test13.jpg',image_id="test13"), ImageDataModel( image_path=r'images\test14.jpg',image_id="test14"), ImageDataModel( image_path=r'images\test15.jpg',image_id="test15"), ImageDataModel( image_path=r'images\test16.jpg',image_id="test16"), ImageDataModel( image_path=r'images\test17.jpg',image_id="test17"), ImageDataModel( image_path=r'images\test18.jpg',image_id="test18"), ImageDataModel( image_path=r'images\test19.jpg',image_id="test19"), ImageDataModel( image_path=r'images\test20.jpg',image_id="test20"), ImageDataModel( image_path=r'images\test21.jpg',image_id="test21"), ImageDataModel( image_path=r'images\test22.jpg',image_id="test22"), ImageDataModel( image_path=r'images\test23.jpg',image_id="test23"), ]
# processed_images = repository.add_images(images=images)

# people = [ PersonDataModel(name='Kami Tazyig',person_id="2"), PersonDataModel(name='Omri Berchman',person_id="1"),PersonDataModel(name='Yaron Shaharabani',person_id="3"), PersonDataModel(name='Amit Inbar',person_id="4")]
# repository.add_persons(persons_list=people)
# people = repository.get_persons()
# faces = repository.get_faces()
# kami_faces = [face for face in faces if face.face_id == "test3_face_5"] 
# omri_faces = [face for face in faces if face.face_id == "test3_face_4"  ]
# yaron_faces = [face for face in faces if face.face_id == "test3_face_3" ]
# amit_faces = [face for face in faces if face.face_id == "test3_face_2" ]
# repository.add_verified_faces_to_person(person=people[0], faces=kami_faces)
# repository.add_verified_faces_to_person(person=people[1], faces=omri_faces)
# repository.add_verified_faces_to_person(person=people[2], faces=yaron_faces)
# repository.add_verified_faces_to_person(person=people[3], faces=amit_faces)