Graduation / Logic /GetModels /FaceModel.py
Mohammedallyl's picture
Upload 27 files
c4dc0b3 verified
raw
history blame
1.46 kB
from fastapi import APIRouter
from fastapi.responses import FileResponse
from passlib.context import CryptContext
import sqlite3
FaceModelRouter=APIRouter(prefix="/GetModel")
@FaceModelRouter.post("/GetFaceModel")
def FaceModelEndPoint(Email:str,Password:str):
try:
State=False
pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
connect=sqlite3.connect("DataBase/DataBase.bd")
cursor=connect.execute(f'''
SELECT Password FROM Users where Email='{Email}'
''')
if len(cursor.fetchall())==1:
HasedPassword=cursor.fetchall()[0][0]
State=pwd_context.verify(Password,HasedPassword)
if not State:
return {"Status":False,"Message":"Email or Password is not correct"}
pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
Password=pwd_context.hash(Password)
connect=sqlite3.connect("DataBase/DataBase.bd")
cursor=connect.execute(f'''
SELECT UserId FROM Users where Email='{Email}'
''')
if len(cursor.fetchall())==0:
return {"Status":False,"Message":"somthing went wrong"}
UserId=cursor.fetchone()["UserId"]
return FileResponse(f"./ExtactedFaces/FaceModel/{UserId}/FinalFaceModel.tflite")
except Exception as e:
return {"Status":False,"Message":e}