Upload folder using huggingface_hub

requirements.txt

@@ -1,7 +1,7 @@
-comfyui-frontend-package==1.33.10
-comfyui-workflow-templates==0.7.25
-comfyui-embedded-docs==0.3.1
-torch
+comfyui-frontend-package==1.37.11
+comfyui-workflow-templates==0.8.24
+comfyui-embedded-docs==0.4.0
+torch==2.9.1
 torchsde
 torchvision
 torchaudio
@@ -21,6 +21,8 @@ psutil
 alembic
 SQLAlchemy
 av>=14.2.0
+comfy-kitchen>=0.2.7
+requests
 
 #non essential dependencies:
 kornia>=0.7.1

ui/events.py

@@ -421,7 +421,7 @@ def attach_event_handlers(ui_components, demo):
 
 
     for prefix, task_type in [
-        ("txt2img", "txt2img"), ("img2img", "img2img"),
+        ("txt2img", "txt2img"), ("img2img", "img2img"), ("inpaint", "inpaint"),
         ("outpaint", "outpaint"), ("hires_fix", "hires_fix"),
     ]:
         if f'add_lora_button_{prefix}' in ui_components: 
@@ -481,25 +481,25 @@ def attach_event_handlers(ui_components, demo):
             model_dropdown = ui_components[f'base_model_{prefix}']
             aspect_ratio_dropdown.change(fn=on_aspect_ratio_change, inputs=[aspect_ratio_dropdown, model_dropdown], outputs=[width_component, height_component], show_progress=False)
             
-    # if 'view_mode_inpaint' in ui_components:
-    #     def toggle_inpaint_fullscreen_view(view_mode):
-    #         is_fullscreen = (view_mode == "Fullscreen View")
-    #         other_elements_visible = not is_fullscreen
-    #         editor_height = 800 if is_fullscreen else 272
-    #         return {
-    #             ui_components['model_and_run_row_inpaint']: gr.update(visible=other_elements_visible),
-    #             ui_components['prompts_column_inpaint']: gr.update(visible=other_elements_visible),
-    #             ui_components['params_and_gallery_row_inpaint']: gr.update(visible=other_elements_visible),
-    #             ui_components['accordion_wrapper_inpaint']: gr.update(visible=other_elements_visible),
-    #             ui_components['input_image_dict_inpaint']: gr.update(height=editor_height),
-    #         }
+    if 'view_mode_inpaint' in ui_components:
+        def toggle_inpaint_fullscreen_view(view_mode):
+            is_fullscreen = (view_mode == "Fullscreen View")
+            other_elements_visible = not is_fullscreen
+            editor_height = 800 if is_fullscreen else 272
+            return {
+                ui_components['model_and_run_row_inpaint']: gr.update(visible=other_elements_visible),
+                ui_components['prompts_column_inpaint']: gr.update(visible=other_elements_visible),
+                ui_components['params_and_gallery_row_inpaint']: gr.update(visible=other_elements_visible),
+                ui_components['accordion_wrapper_inpaint']: gr.update(visible=other_elements_visible),
+                ui_components['input_image_dict_inpaint']: gr.update(height=editor_height),
+            }
         
-    #     output_components = [
-    #         ui_components['model_and_run_row_inpaint'], ui_components['prompts_column_inpaint'],
-    #         ui_components['params_and_gallery_row_inpaint'], ui_components['accordion_wrapper_inpaint'],
-    #         ui_components['input_image_dict_inpaint']
-    #     ]
-    #     ui_components['view_mode_inpaint'].change(fn=toggle_inpaint_fullscreen_view, inputs=[ui_components['view_mode_inpaint']], outputs=output_components, show_progress=False)
+        output_components = [
+            ui_components['model_and_run_row_inpaint'], ui_components['prompts_column_inpaint'],
+            ui_components['params_and_gallery_row_inpaint'], ui_components['accordion_wrapper_inpaint'],
+            ui_components['input_image_dict_inpaint']
+        ]
+        ui_components['view_mode_inpaint'].change(fn=toggle_inpaint_fullscreen_view, inputs=[ui_components['view_mode_inpaint']], outputs=output_components, show_progress=False)
 
     def initialize_all_cn_dropdowns():
         cn_config = load_controlnet_config()
@@ -521,7 +521,7 @@ def attach_event_handlers(ui_components, demo):
                     break
 
         updates = {}
-        for prefix in ["txt2img", "img2img", "outpaint", "hires_fix"]:
+        for prefix in ["txt2img", "img2img", "inpaint", "outpaint", "hires_fix"]:
             if f'controlnet_types_{prefix}' in ui_components:
                 for type_dd in ui_components[f'controlnet_types_{prefix}']:
                     updates[type_dd] = gr.update(choices=all_types, value=default_type)
@@ -552,7 +552,7 @@ def attach_event_handlers(ui_components, demo):
         return all_updates
     
     all_load_outputs = []
-    for prefix in ["txt2img", "img2img", "outpaint", "hires_fix"]:
+    for prefix in ["txt2img", "img2img", "inpaint", "outpaint", "hires_fix"]:
         if f'controlnet_types_{prefix}' in ui_components:
             all_load_outputs.extend(ui_components[f'controlnet_types_{prefix}'])
             all_load_outputs.extend(ui_components[f'controlnet_series_{prefix}'])

ui/layout.py

@@ -25,8 +25,8 @@ def build_ui(event_handler_function):
         gr.Markdown(
             "This demo is a streamlined version of the [Comfy web UI](https://github.com/RioShiina47/comfy-webui)'s ImgGen functionality. "
             "Other versions are also available: "
+            "[FLUX.2](https://huggingface.co/spaces/RioShiina/ImageGen-FLUX.2), "
             "[Z-Image](https://huggingface.co/spaces/RioShiina/ImageGen-Z-Image), "
-            "[NewBie-Image](https://huggingface.co/spaces/RioShiina/ImageGen-NewBie-Image), "
             "[Illstrious](https://huggingface.co/spaces/RioShiina/ImageGen-Illstrious), "
             "[NoobAI](https://huggingface.co/spaces/RioShiina/ImageGen-NoobAI), "
             "[Pony](https://huggingface.co/spaces/RioShiina/ImageGen-Pony1), "
@@ -42,8 +42,8 @@ def build_ui(event_handler_function):
                     with gr.TabItem("Img2Img", id=1):
                         ui_components.update(img2img_ui.create_ui())
 
-                    # with gr.TabItem("Inpaint", id=2):
-                    #     ui_components.update(inpaint_ui.create_ui())
+                    with gr.TabItem("Inpaint", id=2):
+                        ui_components.update(inpaint_ui.create_ui())
 
                     with gr.TabItem("Outpaint", id=3):
                         ui_components.update(outpaint_ui.create_ui())