From 95ed7d54262728ebad169646317dc0fd43142b44 Mon Sep 17 00:00:00 2001
From: Peyton <peytond@methodsandmastery.com>
Date: Sat, 19 Aug 2023 20:48:23 -0500
Subject: [PATCH] Added logic to deal with hidden variables.

---
 comfyui_to_python.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/comfyui_to_python.py b/comfyui_to_python.py
index 4f7f48c..0e27353 100644
--- a/comfyui_to_python.py
+++ b/comfyui_to_python.py
@@ -4,6 +4,7 @@ import inspect
 import json
 import logging
 import os
+import random
 import sys
 
 import black
@@ -230,6 +231,9 @@ class CodeGenerator:
 
             # Remove any keyword arguments from **inputs if they are not in class_def_params
             inputs = {key: value for key, value in inputs.items() if key in class_def_params}
+            # Deal with hidden variables
+            if 'unique_id' in class_def_params:
+                inputs['unique_id'] = random.randint(1, 2**64)
 
             # Create executed variable and generate code
             executed_variables[idx] = f'{class_type.lower().strip()}_{idx}'
@@ -429,8 +433,8 @@ class ComfyUItoPython:
 
 if __name__ == '__main__':
     # Update class parameters here
-    input_file = 'workflow_api_serge.json'
-    output_file = 'workflow_api_serge.py'
+    input_file = 'workflow_api_impact3.json'
+    output_file = 'workflow_api_impact3.py'
     queue_size = 10
 
     # Convert ComfyUI workflow to Python