Update app.py

app.py

@@ -1,25 +1,20 @@
 import gradio as gr
 from deepface import DeepFace
-import os
-import uuid
 import numpy as np
 from PIL import Image
 import json
+import uuid
 
-# Thư mục lưu ảnh upload
-UPLOAD_FOLDER = "uploads"
-if not os.path.exists(UPLOAD_FOLDER):
-    os.makedirs(UPLOAD_FOLDER)
+# --- Lưu ảnh tạm trong session (dictionary) ---
+image_storage = {}
 
 # 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.fromarray(image).save(save_path)
+        image_storage[image_id] = image
         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
@@ -29,10 +24,10 @@ 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):
+        if image_id not in image_storage:
             return "Ảnh ID không tồn tại"
-        result = DeepFace.verify(img1_path=image, img2_path=target_path)
+        stored_image = image_storage[image_id]
+        result = DeepFace.verify(img1_path=image, img2_path=stored_image)
         output = {
             "Verified": result["verified"],
             "Khoảng cách": round(result["distance"], 4)
@@ -62,14 +57,14 @@ def analyze_face(image):
     except Exception as e:
         return f"Lỗi khi phân tích: {str(e)}"
 
-# Tạo giao diện Gradio
+# --- Tạo giao diện Gradio ---
 with gr.Blocks() as demo:
     gr.Markdown("## 📁 Upload ảnh")
     with gr.Row():
         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()
+        image_id_store = gr.State()  # lưu image_id
 
     upload_btn.click(upload_image, inputs=upload_input, outputs=[upload_output, image_id_store])