Spaces:
Sleeping
Sleeping
File size: 3,309 Bytes
c4dc0b3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
from fastapi import APIRouter,BackgroundTasks,HTTPException
from os.path import exists
from os import listdir
import subprocess
from passlib.context import CryptContext
import sqlite3
ModelTrainer=APIRouter(prefix="/Trainer")
TrainingProcess={}
def TrainModel(UserId):
global TrainingProcess
if TrainingProcess.get(UserId,False) and TrainingProcess[UserId].poll() is None:
raise HTTPException(status_code=400,detail="Model is already training.")
TrainingProcess=subprocess.Popen(["python","./FaceRecognition/ModelTrainer.py",f"{UserId}"])
@ModelTrainer.post("/TrainFaceModel")
async def SpeachToTextEndPoint(Tasks:BackgroundTasks,Email:str,Password:str):
global TrainingProcess
try:
State=False
pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
connect=sqlite3.connect("DataBase/DataBase.bd")
cursor=connect.execute(f'''
SELECT UserId,Password FROM Users where Email='{Email}'
''')
Data=cursor.fetchall()
if len(Data) !=0 :
if Data[0][0]==None:
return {"Status":True,"Message":"Email or Password Is Incorrect"}
HasedPassword=Data[0][1]
UserId=Data[0][0]
State=pwd_context.verify(Password,HasedPassword)
if exists(f"./FaceRecognition/ExtactedFaces/{UserId}/Train"):
for UserName in listdir(f"./FaceRecognition/ExtactedFaces/{UserId}/Train"):
if len(listdir(f"./FaceRecognition/ExtactedFaces/{UserId}/Train/"+UserName))<2:
return {"Status":False,"Message":f"{UserName} has only {len(listdir(f'./FaceRecognition/ExtactedFaces/{UserId}/Train/'+UserName))} image and it must be 2 or more"}
if not State:
return {"Status":False,"Message":"Email or Password is not correct"}
if TrainingProcess.get(UserId,False) and TrainingProcess[UserId].poll() is None:
raise HTTPException(status_code=400,detail="Model is already training.")
Tasks.add_task(TrainModel ,args=[UserId])
return{"message":"Training Started"}
except Exception as e:
return{"Stats":False,"message":f"{e}"}
@ModelTrainer.post("/TrainFaceModelStatus")
async def SpeachToTextEndPoint(Email:str,Password:str):
global TrainingProcess
try:
State=False
pwd_context=CryptContext(schemes=["bcrypt"],deprecated="auto")
connect=sqlite3.connect("DataBase/DataBase.bd")
cursor=connect.execute(f'''
SELECT UserId,Password FROM Users where Email='{Email}'
''')
if len(cursor.fetchall())==1:
HasedPassword=cursor.fetchall()[0][1]
UserId=cursor.fetchall()[0][0]
State=pwd_context.verify(Password,HasedPassword)
if not State:
return {"Status":False,"Message":"Email or Password is not correct"}
if TrainingProcess.get(UserId,False) and TrainingProcess[UserId].poll() is None:
return{"message":"Model still Training "}
else:
return{"message":"Model Training ended"}
except Exception as e:
return{"Stats":True,"message":f"{e}"} |