Update app.py

app.py

@@ -12,20 +12,24 @@ from roop.core import (
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
 import os
+import random
 from PIL import Image
-import uuid
 import onnxruntime as ort
 import cv2
 from roop.face_analyser import get_one_face
 
 @spaces.GPU
 def swap_face(source_file, target_file, doFaceEnhancer):
-    session_id = str(uuid.uuid4())  # Tạo một UUID duy nhất cho mỗi phiên làm việc
-    session_dir = f"temp/{session_id}"
+    session_dir = "temp"  # Sử dụng thư mục cố định
     os.makedirs(session_dir, exist_ok=True)
 
-    source_path = os.path.join(session_dir, "input.jpg")
-    target_path = os.path.join(session_dir, "target.jpg")
+    # Tạo tên file ngẫu nhiên
+    source_filename = f"source_{random.randint(1000, 9999)}.jpg"
+    target_filename = f"target_{random.randint(1000, 9999)}.jpg"
+    output_filename = f"output_{random.randint(1000, 9999)}.jpg"
+
+    source_path = os.path.join(session_dir, source_filename)
+    target_path = os.path.join(session_dir, target_filename)
 
     source_image = Image.fromarray(source_file)
     source_image.save(source_path)
@@ -45,12 +49,11 @@ def swap_face(source_file, target_file, doFaceEnhancer):
     if target_face is None:
         raise gradio.exceptions.Error("No face in target path detected.")
 
-    output_path = os.path.join(session_dir, "output.jpg")
+    output_path = os.path.join(session_dir, output_filename)
     normalized_output_path = normalize_output_path(source_path, target_path, output_path)
 
     frame_processors = ["face_swapper", "face_enhancer"] if doFaceEnhancer else ["face_swapper"]
 
-
     for frame_processor in get_frame_processors_modules(frame_processors):
         if not frame_processor.pre_check():
             print(f"Pre-check failed for {frame_processor}")