Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 from deepface import DeepFace
 import os
 import uuid
-import shutil
+import numpy as np
+from PIL import Image
 import json
 
 # Thư mục lưu ảnh upload
@@ -10,12 +11,15 @@ UPLOAD_FOLDER = "uploads"
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
-# 1️⃣ API: Push ảnh lên server
+# 1️⃣ API: Upload ảnh
 def upload_image(image):
     try:
+        # Chuyển PIL sang numpy nếu cần
+        if isinstance(image, Image.Image):
+            image = np.array(image)
         image_id = str(uuid.uuid4())
         save_path = os.path.join(UPLOAD_FOLDER, f"{image_id}.png")
-        image.save(save_path)  # nếu image là PIL.Image
+        Image.fromarray(image).save(save_path)
         return f"Ảnh lưu thành công với ID: {image_id}", image_id
     except Exception as e:
         return f"Lỗi khi upload: {str(e)}", None
@@ -23,11 +27,12 @@ def upload_image(image):
 # 2️⃣ API: Nhận diện khuôn mặt
 def recognize_face(image, image_id):
     try:
+        if isinstance(image, Image.Image):
+            image = np.array(image)
         target_path = os.path.join(UPLOAD_FOLDER, f"{image_id}.png")
         if not os.path.exists(target_path):
             return "Ảnh ID không tồn tại"
         result = DeepFace.verify(img1_path=image, img2_path=target_path)
-        # result là dict có 'verified': True/False, 'distance': float
         output = {
             "Verified": result["verified"],
             "Khoảng cách": round(result["distance"], 4)
@@ -39,6 +44,8 @@ def recognize_face(image, image_id):
 # 3️⃣ API: Phân tích khuôn mặt
 def analyze_face(image):
     try:
+        if isinstance(image, Image.Image):
+            image = np.array(image)
         result = DeepFace.analyze(
             img_path=image,
             actions=["age", "gender", "emotion", "race"],
@@ -59,16 +66,16 @@ def analyze_face(image):
 with gr.Blocks() as demo:
     gr.Markdown("## 📁 Upload ảnh")
     with gr.Row():
-        upload_input = gr.Image(label="Upload ảnh", type="pil")
+        upload_input = gr.Image(label="Upload ảnh", type="numpy")
         upload_btn = gr.Button("Upload")
         upload_output = gr.Textbox(label="Kết quả Upload")
-        image_id_store = gr.State()  # lưu image_id
+        image_id_store = gr.State()
 
     upload_btn.click(upload_image, inputs=upload_input, outputs=[upload_output, image_id_store])
 
     gr.Markdown("## 🆔 Nhận diện khuôn mặt")
     with gr.Row():
-        recognize_input = gr.Image(label="Ảnh cần nhận diện", type="pil")
+        recognize_input = gr.Image(label="Ảnh cần nhận diện", type="numpy")
         recognize_btn = gr.Button("Nhận diện")
         recognize_output = gr.Textbox(label="Kết quả nhận diện")
 
@@ -76,7 +83,7 @@ with gr.Blocks() as demo:
 
     gr.Markdown("## 🔍 Phân tích khuôn mặt")
     with gr.Row():
-        analyze_input = gr.Image(label="Ảnh cần phân tích", type="pil")
+        analyze_input = gr.Image(label="Ảnh cần phân tích", type="numpy")
         analyze_btn = gr.Button("Phân tích")
         analyze_output = gr.Textbox(label="Kết quả phân tích")