Update app.py

app.py

@@ -68,7 +68,7 @@ def infer(
     seed=4, 
     guidance_scale=7.5, 
     lora_scale=0.5,
-    #use_control_net=False,  # Добавляем параметр для управления включением ControlNet
+    use_control_net=False,  # Добавляем параметр для управления включением ControlNet
     #control_net_weight=1.0,  control_strength_slider
     progress=gr.Progress(track_tqdm=True)
     ):
@@ -214,84 +214,80 @@ with gr.Blocks(css=css) as demo:
                     value=512,
                 )
 
-        # # Функция для работы с ControlNet ---------------------------------------------------------------------
-        # def process_input_ControlNet(image, use_control_net, control_strength, control_mode):
-        #     if use_control_net:
-        #         # Логика для обработки с использованием ControlNet
-        #         result = f"ControlNet активен! Режим: {control_mode}, Интенсивность: {control_strength}"
-        #     else:
-        #         # Логика для обработки без ControlNet
-        #         result = "ControlNet отключен."
-        #     return result
-
-        # with gr.Blocks():
-        #     with gr.Row():
-        #         # Чекбокс для включения/отключения ControlNet
-        #         use_control_net = gr.Checkbox(
-        #             label="Use ControlNet",
-        #             value=False,
-        #         )
-
-        #     # Дополнительные опции для ControlNet
-        #     with gr.Column(visible=False) as control_net_options:    
-        #         # Слайдер для настройки интенсивности
-        #         control_strength = gr.Slider(
-        #             label="Control Strength",
-        #             minimum=0.0,
-        #             maximum=1.0,
-        #             value=0.5,
-        #             step=0.05,
-        #         )
-        
-        #         # Выпадающий список для выбора режима
-        #         control_mode = gr.Dropdown(
-        #             label="Control Mode",
-        #             choices=[
-        #                 "edge_detection",
-        #                 "canny_edge_detection",
-        #                 "pose_estimation",
-        #                 "depth_map",
-        #                 "segmentation_map",
-        #                 "scribble_sketch",
-        #                 "normal_map",
-        #                 "hed_edge_detection",
-        #                 "openpose",
-        #                 "mlsd_line_detection",
-        #                 "scribble_diffusion",
-        #                 "semantic_segmentation",
-        #                 "style_transfer",
-        #                 "colorization",
-        #                 "custom_map"
-        #             ],
-        #             value="pose_estimation",
-        #         )
-        
-        #         # Окно для загрузки изображений
-        #         control_image = gr.Image(label="Upload Control Image")
-
-        #         # Кнопка для запуска работы ControlNet
-        #         run_button = gr.Button("Run")
+        # Функция для работы с ControlNet ---------------------------------------------------------------------
+        def process_input_ControlNet(image, use_control_net, control_strength, control_mode):
+            if use_control_net:
+                # Логика для обработки с использованием ControlNet
+                result = f"ControlNet активен! Режим: {control_mode}, Интенсивность: {control_strength}"
+            else:
+                # Логика для обработки без ControlNet
+                result = "ControlNet отключен."
+            return result
 
-        #         # Текстовое поле для вывода результата
-        #         output = gr.Textbox(label="Output")
+        with gr.Blocks():
+            with gr.Row():
+                # Чекбокс для включения/отключения ControlNet
+                use_control_net = gr.Checkbox(
+                    label="Use ControlNet",
+                    value=False,
+                )
 
-        #     # Логика для отображения/скрытия дополнительных опций
-        #     use_control_net.change(
-        #         fn=lambda x: gr.Row.update(visible=x),
-        #         inputs=use_control_net,
-        #         outputs=control_net_options,
-        #     )
+            # Дополнительные опции для ControlNet
+            with gr.Column(visible=False) as control_net_options:    
+                # Слайдер для настройки интенсивности
+                control_strength = gr.Slider(
+                    label="Control Strength",
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.5,
+                    step=0.05,
+                )
+        
+                # Выпадающий список для выбора режима
+                control_mode = gr.Dropdown(
+                    label="Control Mode",
+                    choices=[
+                        "edge_detection",
+                        "canny_edge_detection",
+                        "pose_estimation",
+                        "depth_map",
+                        "segmentation_map",
+                        "scribble_sketch",
+                        "normal_map",
+                        "hed_edge_detection",
+                        "openpose",
+                        "mlsd_line_detection",
+                        "scribble_diffusion",
+                        "semantic_segmentation",
+                        "style_transfer",
+                        "colorization",
+                        "custom_map"
+                    ],
+                    value="pose_estimation",
+                )
+        
+                # Окно для загрузки изображений
+                control_image = gr.Image(label="Upload Control Image")
 
-        #     # Привязка кнопки Run к функции работы с ControlNet
-        #     run_button.click(
-        #         fn=process_input_ControlNet,
-        #         inputs=[control_image, use_control_net, control_strength, control_mode],
-        #         outputs=output,
-        #     )
+                # Кнопка для запуска работы ControlNet
+                run_button = gr.Button("Run")
 
+                # Текстовое поле для вывода результата
+                output = gr.Textbox(label="Output")
 
+            # Логика для отображения/скрытия дополнительных опций
+            use_control_net.change(
+                fn=lambda x: gr.Row.update(visible=x),
+                inputs=use_control_net,
+                outputs=control_net_options,
+            )
 
-        
+            # Привязка кнопки Run к функции работы с ControlNet
+            run_button.click(
+                fn=process_input_ControlNet,
+                inputs=[control_image, use_control_net, control_strength, control_mode],
+                outputs=output,
+            )
 
         # Функция для работы с IP-adapter ----------------------------------------------------------------------------
         def process_input_IP_adapter(image, use_ip_adapter, ip_adapter_scale, ip_adapter_image):
@@ -339,7 +335,7 @@ with gr.Blocks(css=css) as demo:
                 outputs=ip_adapter_options,
             )
 
-            # Привязка кнопки к функции обработки
+            # Привязка кнопки Run к функции работы с IP-adapter
             run_button.click(
                 fn=process_input_IP_adapter,
                 inputs=[ip_adapter_image, use_ip_adapter, ip_adapter_scale, ip_adapter_image],
@@ -374,7 +370,7 @@ with gr.Blocks(css=css) as demo:
             seed,
             guidance_scale,
             lora_scale,
-            #use_control_net,  # Передаем состояние чекбокса
+            use_control_net,  # Передаем состояние чекбокса
             #control_net_weight,  # Передаем вес ControlNet
         ],
         outputs=[result],