wrapper function

app.py

@@ -230,6 +230,14 @@ def video_detection(video, det_conf_thres, clf_conf_thres, target_fps, codec):
                            callback=callback, target_fps=int(target_fps), codec=codec)
     return target_path
 
+def wrap_bool_output(fn):
+    def wrapped(*args, **kwargs):
+        result = fn(*args, **kwargs)
+        if isinstance(result, bool):
+            return {"success": result}
+        return result
+    return wrapped
+
 # Building Gradio UI
 
 with gr.Blocks() as demo:
@@ -304,7 +312,7 @@ with gr.Blocks() as demo:
                     process_files_button = gr.Button("Separate files")
                     process_result = gr.Text("Click on 'Separate files' once you see the JSON file", label="Separated files:")
                     process_btn.click(batch_path_detection, inputs=[inp_path, bth_conf_fs], outputs=bth_out2)
-                    process_files_button.click(pw_utils.detection_folder_separation, inputs=[bth_out2, inp_path, out_path, bth_conf_fs], outputs=process_result)
+                    process_files_button.click(wrap_bool_output(pw_utils.detection_folder_separation, inputs=[bth_out2, inp_path, out_path, bth_conf_fs], outputs=process_result))
     with gr.Tab("Batch Image Process"):
         with gr.Row():
             with gr.Column():