Update app.py

app.py

@@ -385,9 +385,26 @@ def infer(
                 pil_images.append(img)
                 print(f"Loaded image from URL: {url_list[0]}")
             else:
-                imgChar = Image.open(os.path.join(face_dir, url_list[0])).convert("RGB")
-                pil_images.append(imgChar)
-                print(f"Loaded image from Local: {url_list[0]}")
+                imgPath = os.path.join(face_dir, url_list[0])
+                if os.path.exists(imgPath):
+                    imgChar = Image.open(imgPath).convert("RGB")
+                    pil_images.append(imgChar)
+                    print(f"Loaded image from Local: {url_list[0]}")
+                else:
+                    ll_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)]
+                    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.append(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}")
             
         if(len(url_list) > 1):
             img = load_image_from_url(url_list[1])