Upload folder using huggingface_hub

core/pipelines/sd_image_pipeline.py

@@ -198,13 +198,14 @@ class SdImagePipeline(BasePipeline):
 
         if task_type == 'img2img':
             input_image_pil = ui_inputs.get('img2img_image')
-            if input_image_pil:
-                temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
-                input_image_pil.save(temp_file_path, "PNG")
-                ui_inputs['input_image'] = os.path.basename(temp_file_path)
-                temp_files_to_clean.append(temp_file_path)
-                ui_inputs['width'] = input_image_pil.width
-                ui_inputs['height'] = input_image_pil.height
+            if not input_image_pil:
+                raise gr.Error("Please upload an image for Image-to-Image.")
+            temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
+            input_image_pil.save(temp_file_path, "PNG")
+            ui_inputs['input_image'] = os.path.basename(temp_file_path)
+            temp_files_to_clean.append(temp_file_path)
+            ui_inputs['width'] = input_image_pil.width
+            ui_inputs['height'] = input_image_pil.height
         
         elif task_type == 'inpaint':
             inpaint_dict = ui_inputs.get('inpaint_image_dict')
@@ -231,22 +232,24 @@ class SdImagePipeline(BasePipeline):
 
         elif task_type == 'outpaint':
             input_image_pil = ui_inputs.get('outpaint_image')
-            if input_image_pil:
-                temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
-                input_image_pil.save(temp_file_path, "PNG")
-                ui_inputs['input_image'] = os.path.basename(temp_file_path)
-                temp_files_to_clean.append(temp_file_path)
+            if not input_image_pil:
+                raise gr.Error("Please upload an image for Outpainting.")
+            temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
+            input_image_pil.save(temp_file_path, "PNG")
+            ui_inputs['input_image'] = os.path.basename(temp_file_path)
+            temp_files_to_clean.append(temp_file_path)
             
             ui_inputs['megapixels'] = 0.25
             ui_inputs['grow_mask_by'] = ui_inputs.get('feathering', 10)
 
         elif task_type == 'hires_fix':
             input_image_pil = ui_inputs.get('hires_image')
-            if input_image_pil:
-                temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
-                input_image_pil.save(temp_file_path, "PNG")
-                ui_inputs['input_image'] = os.path.basename(temp_file_path)
-                temp_files_to_clean.append(temp_file_path)
+            if not input_image_pil:
+                raise gr.Error("Please upload an image for Hires Fix.")
+            temp_file_path = os.path.join(INPUT_DIR, f"temp_input_{random.randint(1000, 9999)}.png")
+            input_image_pil.save(temp_file_path, "PNG")
+            ui_inputs['input_image'] = os.path.basename(temp_file_path)
+            temp_files_to_clean.append(temp_file_path)
 
         embedding_data = ui_inputs.get('embedding_data', [])
         embedding_filenames = []

requirements.txt

@@ -1,5 +1,5 @@
 comfyui-frontend-package==1.43.18
-comfyui-workflow-templates==0.9.72
+comfyui-workflow-templates==0.9.73
 comfyui-embedded-docs==0.4.4
 torch==2.10.0
 torchsde