From f5818adb64e8dfc814bf36a17994457f0886efa8 Mon Sep 17 00:00:00 2001
From: Yoland Y <4950057+yoland68@users.noreply.github.com>
Date: Wed, 17 Jul 2024 17:23:28 -0700
Subject: [PATCH] Add 10 min timeout for websocket

---
 action.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/action.py b/action.py
index 34b3f9b..6ebdaac 100644
--- a/action.py
+++ b/action.py
@@ -159,7 +159,16 @@ def main(args):
         start_time = int(datetime.datetime.now().timestamp())
         try:
             result = subprocess.run(
-                ["comfy", "--skip-prompt", "--no-enable-telemetry", "run", "--workflow", file_path],
+                [
+                    "comfy",
+                    "--skip-prompt",
+                    "--no-enable-telemetry",
+                    "run",
+                    "--workflow",
+                    file_path,
+                    "--timeout",
+                    600
+                ],
                 check=True,
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,