From 810aec51352e25211bad5f83c61ed28b2fc6e0d7 Mon Sep 17 00:00:00 2001
From: bennykok <itechbenny@gmail.com>
Date: Thu, 25 Apr 2024 13:15:29 +0800
Subject: [PATCH] fix: empty inputs causing run issues

---
 custom_routes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_routes.py b/custom_routes.py
index ff94dec..db8319c 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -118,7 +118,7 @@ def apply_inputs_to_workflow(workflow_api: Any, inputs: Any, sid: str = None):
                 if (value["class_type"] == "ComfyDeployWebscoketImageInput"):
                     value['inputs']["client_id"] = sid
                     
-            if "input_id" in value['inputs'] and value['inputs']['input_id'] in inputs:
+            if "input_id" in value['inputs'] and inputs is not None and value['inputs']['input_id'] in inputs:
                 new_value = inputs[value['inputs']['input_id']]
                 
                 # Lets skip it if its an image