Update

app.py

@@ -161,15 +161,14 @@ class webui:
     def __init__(self):
         self.demo = gr.Blocks()
 
-    def main(self, image_path):
+    def main(self, image_path, kernel_size, sigma, k_sigma, gamma):
         image = Image.open(image_path).convert('RGBA')
-        #拡張子を取り除いたファイル名を取得
         image_name = os.path.splitext(image_path)[0]
         alpha = image.getchannel('A') if image.mode == 'RGBA' else None
         image.save(image_path)
         image = Image.open(image_path).convert('RGBA')
         rgb_image = image.convert('RGB')
-        lineart = process_XDoG(image_path).convert('L')
+        lineart = process_XDoG(image_path, kernel_size, sigma, k_sigma, gamma).convert('L')
         replace_color_image = process_image(rgb_image, lineart).convert('RGBA')
         
         if alpha:
@@ -197,6 +196,10 @@ class webui:
             with gr.Row():
                 with gr.Column():
                     input_image = gr.Image(type='filepath', image_mode="RGBA", label="Original Image")
+                    kernel_size_input = gr.Number(value=0, label="Kernel Size", precision=0)
+                    sigma_input = gr.Number(value=1.0, label="Sigma")
+                    k_sigma_input = gr.Number(value=2.0, label="K Sigma")
+                    gamma_input = gr.Number(value=1.5, label="Gamma")
                     submit = gr.Button(value="Start")
                 with gr.Row():
                     with gr.Column():
@@ -205,7 +208,7 @@ class webui:
                         output_file = gr.File()
             submit.click(
                 self.main, 
-                inputs=[input_image], 
+                inputs=[input_image, kernel_size_input, sigma_input, k_sigma_input, gamma_input], 
                 outputs=[output_0, output_file]
             )