Update app.py

app.py

@@ -355,12 +355,36 @@ def infer(
 
     # --- NEW: Load default image if no input ---
     if not pil_images:
-        default_path = os.path.join(os.path.dirname(__file__), "1.jpg")
-        if os.path.exists(default_path):
-            pil_images = [Image.open(default_path).convert("RGB")]
-            print("Loaded default image: 1.jpg")
+        # 1. Định nghĩa đường dẫn đến thư mục /Face/
+        # os.path.dirname(__file__) lấy thư mục chứa file hiện tại (app.py)
+        face_dir = os.path.join(os.path.dirname(__file__), "Face")
+    
+        if os.path.isdir(face_dir):
+            # 2. Lấy danh sách tất cả các file trong thư mục /Face/
+            all_files = os.listdir(face_dir)
+    
+            # 3. Lọc ra các file ảnh (bạn có thể tùy chỉnh các phần mở rộng)
+            image_extensions = ('.jpg', '.jpeg', '.png', '.webp')
+            image_files = [f for f in all_files if f.lower().endswith(image_extensions)]
+    
+            if image_files:
+                # 4. Chọn ngẫu nhiên một file ảnh
+                random_image_name = random.choice(image_files)
+                random_image_path = os.path.join(face_dir, random_image_name)
+    
+                # 5. Tải ảnh và thêm vào pil_images
+                try:
+                    pil_images = [Image.open(random_image_path).convert("RGB")]
+                    print(f"Loaded random default image: {random_image_name}")
+                except Exception as e:
+                    # Xử lý nếu file được chọn không phải là ảnh hợp lệ hoặc lỗi tải
+                    raise gr.Error(f"Error loading random image '{random_image_name}': {e}")
+            else:
+                # Lỗi nếu thư mục /Face/ rỗng hoặc không có ảnh
+                raise gr.Error(f"No input images provided and no image files found in '{face_dir}'.")
         else:
-            raise gr.Error("No input images and '1.jpg' not found in app directory.")
+            # Lỗi nếu thư mục /Face/ không tồn tại
+            raise gr.Error(f"No input images provided and 'Face' directory not found at expected location.")
 
     if height==256 and width==256:
         height, width = None, None