Spaces:
Runtime error
Runtime error
File size: 2,734 Bytes
63e2f2d | 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 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | from fastapi import FastAPI,Form,HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from model.face_model import generate_facenet512_embedding,base64_to_cv2_image
from utils.client import upsert_embedding,search_embedding
from utils.supa_client import add_user,search_user
from model.face_match import face_match
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
class ImageData(BaseModel):
image: str
@app.post("/user")
async def upload_image(data: ImageData):
try:
embedding = generate_facenet512_embedding(data.image)
result = search_embedding(embedding)
id = result["matches"][0]["id"]
response = search_user(id)
img1 = base64_to_cv2_image(data.image)
img2 = base64_to_cv2_image(response.data[0]['captured_image'])
final = face_match(img1,img2)
if(final):
return {
"message": "Embedding generated successfully",
"embedding": response
}
else:
raise HTTPException(status_code=400, detail="Face not founf Proprly")
except Exception as e:
return {"error": str(e)}
@app.post("/new_user")
async def new_user(
username: str = Form(...),
age: int = Form(...),
phone: str = Form(...),
address: str = Form(...),
email: str = Form(...),
gender: str = Form(...),
image: str = Form(...) # base64 string from frontend
):
try:
# Generate embedding and ID
embedding = generate_facenet512_embedding(image)
result = search_embedding(embedding)
id = result["matches"][0]["id"]
res = search_user(id)
img1 = base64_to_cv2_image(image)
img2 = base64_to_cv2_image(res.data[0]['captured_image'])
final = face_match(img1,img2)
print(final)
if final:
raise HTTPException(status_code=400, detail="User already exists")
id = upsert_embedding(embedding)
user_data = {
"username": username,
"age": age,
"phone": phone,
"address": address,
"email": email,
"gender": gender,
"captured_image": image
}
# Add user to Supabase
response = add_user(id, user_data)
return {
"success": True,
"message": "User added",
"id": id
}
except Exception as e:
return {"success": False, "error": str(e)}
|