From 8249ddc1859e200ecdec6cdf9df67cee81e70799 Mon Sep 17 00:00:00 2001
From: alienatedsec <73167064+alienatedsec@users.noreply.github.com>
Date: Thu, 5 Sep 2024 15:11:42 +0100
Subject: [PATCH] [Amend] Considering Multiple Executions and OS variations

---
 main.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 485afb34f..b3037f0d0 100644
--- a/main.py
+++ b/main.py
@@ -7,10 +7,23 @@ import folder_paths
 import time
 from comfy.cli_args import args
 from app.logger import setup_logger
+import datetime
+import tempfile
 
-pid = str(os.getpid())
-with open("/tmp/ComfyUI.pid", "w") as f:
-    f.write(pid)
+def write_pid_to_file(identifier="ComfyUI"):
+    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") 
+    filename = f"{identifier}_{timestamp}_{os.getpid()}.pid"  
+
+    temp_dir = tempfile.gettempdir()
+    full_filepath = os.path.join(temp_dir, filename)
+
+    with open(full_filepath, "w") as temp_file:
+        temp_file.write(str(os.getpid()))
+
+    return full_filepath 
+
+pid_filepath = write_pid_to_file()
+print(f"PID written to file: {pid_filepath}")
 
 setup_logger(verbose=args.verbose)