From 89f02679d369a8d2287dca4322a669976aa9c638 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Wed, 21 Aug 2024 23:13:11 +0800
Subject: [PATCH] fix mask_dir

---
 iopaint/file_manager/file_manager.py | 2 ++
 iopaint/web_config.py                | 1 +
 2 files changed, 3 insertions(+)

diff --git a/iopaint/file_manager/file_manager.py b/iopaint/file_manager/file_manager.py
index c24f54f..ba1d695 100644
--- a/iopaint/file_manager/file_manager.py
+++ b/iopaint/file_manager/file_manager.py
@@ -81,6 +81,8 @@ class FileManager:
 
     @staticmethod
     def _media_names(directory: Path) -> List[MediasResponse]:
+        if directory is None:
+            return []
         names = sorted([it.name for it in glob_img(directory)])
         res = []
         for name in names:
diff --git a/iopaint/web_config.py b/iopaint/web_config.py
index d582536..7501fd4 100644
--- a/iopaint/web_config.py
+++ b/iopaint/web_config.py
@@ -44,6 +44,7 @@ default_configs = dict(
     cpu_textencoder=False,
     device=Device.cuda,
     input=None,
+    mask_dir=None,
     output_dir=None,
     quality=95,
     enable_interactive_seg=False,