Lifeinhockey commited on
Commit
1dac264
·
verified ·
1 Parent(s): df6b046

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +69 -11
app.py CHANGED
@@ -214,22 +214,80 @@ with gr.Blocks(css=css) as demo:
214
  value=512,
215
  )
216
 
217
- with gr.Blocks(): #as demo:
218
- with gr.Row():
219
- use_control_net = gr.Checkbox(label="Use ControlNet", value=False)
220
 
221
- #with gr.Column(visible=False) as controlnet_column:
222
- with gr.Column(visible=True) as controlnet_column:
223
- with gr.Row():
224
- control_strength_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Control Strength")
225
 
226
- with gr.Row():
227
- control_mode_dropdown = gr.Dropdown(['edge_detection', 'pose_estimation'], label='Control Mode')
228
 
229
- with gr.Row():
230
- control_image_input = gr.Image(type="pil", label="Upload Image for ControlNet")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
231
 
 
 
 
 
 
 
232
 
 
 
 
 
 
 
 
 
 
 
233
  #controlnet_column.visible=True
234
  # def toggle_controlnet_options(use_control_net):
235
  # if use_control_net:
 
214
  value=512,
215
  )
216
 
217
+ # with gr.Blocks(): #as demo:
218
+ # with gr.Row():
219
+ # use_control_net = gr.Checkbox(label="Use ControlNet", value=False)
220
 
221
+ # #with gr.Column(visible=False) as controlnet_column:
222
+ # with gr.Column(visible=True) as controlnet_column:
223
+ # with gr.Row():
224
+ # control_strength_slider = gr.Slider(minimum=0, maximum=1, step=0.01, label="Control Strength")
225
 
226
+ # with gr.Row():
227
+ # control_mode_dropdown = gr.Dropdown(['edge_detection', 'pose_estimation'], label='Control Mode')
228
 
229
+ # with gr.Row():
230
+ # control_image_input = gr.Image(type="pil", label="Upload Image for ControlNet")
231
+
232
+
233
+
234
+
235
+ def process_input(image, use_control_net, control_strength, control_mode):
236
+ if use_control_net:
237
+ # Логика для обработки с использованием ControlNet
238
+ result = f"ControlNet активен! Режим: {control_mode}, Интенсивность: {control_strength}"
239
+ else:
240
+ # Логика для обработки без ControlNet
241
+ result = "ControlNet отключен."
242
+ return result
243
+
244
+ # Создание интерфейса
245
+ with gr.Blocks():
246
+ with gr.Row():
247
+ # Чекбокс для включения/отключения ControlNet
248
+ use_control_net = gr.Checkbox(
249
+ label="Use ControlNet",
250
+ value=False,
251
+ )
252
+
253
+ # Дополнительные опции для ControlNet
254
+ with gr.Row(visible=False) as control_net_options:
255
+ # Слайдер для настройки интенсивности
256
+ control_strength = gr.Slider(
257
+ label="Control Strength",
258
+ minimum=0.0,
259
+ maximum=1.0,
260
+ value=0.5,
261
+ step=0.05,
262
+ )
263
+
264
+ # Выпадающий список для выбора режима
265
+ control_mode = gr.Dropdown(
266
+ label="Control Mode",
267
+ choices=["edge_detection", "pose_estimation"],
268
+ value="edge_detection",
269
+ )
270
+
271
+ # Окно для загрузки изображений
272
+ control_image = gr.Image(label="Upload Control Image")
273
 
274
+ # Логика для отображения/скрытия дополнительных опций
275
+ use_control_net.change(
276
+ fn=lambda x: gr.Row.update(visible=x),
277
+ inputs=use_control_net,
278
+ outputs=control_net_options,
279
+ )
280
 
281
+ # Кнопка для запуска обработки
282
+ output = gr.Textbox(label="Output")
283
+ gr.Button("Run").click(
284
+ fn=process_input,
285
+ inputs=[control_image, use_control_net, control_strength, control_mode],
286
+ outputs=output,
287
+ )
288
+
289
+
290
+
291
  #controlnet_column.visible=True
292
  # def toggle_controlnet_options(use_control_net):
293
  # if use_control_net: