Update app.py

app.py

@@ -52,13 +52,30 @@ def choice_group_a(group_model_choice):
     print(choice)
     return choice
 
-def choice_group_b(group_model_choice):
-    output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]
-    return output
-
-def choice_group_c(group_model_choice):
-    current_models = [gr.Textbox(m, visible=False) for m in default_models]
-    return current_models   
+def choice_group_b(choice):
+    return [gr.Image(label=m, min_width=170, height=170) for m in choice]
+
+def choice_group_c(choice):
+    return [gr.Textbox(m, visible=False) for m in choice]   
+
+def choice_groupe_d(current_models,output,group_model_choice):
+    choice=choice_group_a(group_model_choice)
+    current_models=choice_group_c(choice)
+    output=choice_group_b(choice)
+    for m, o in zip(current_models, output):
+        gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+        stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
+    return (current_models,output)
+
+def def_test(var_Test):
+    (gen_button,stop_button,group_model_choice)=var_Test
+    choice=choice_group_a(group_model_choice)
+    current_models=choice_group_c(choice)
+    output=choice_group_b(choice)
+    for m, o in zip(current_models, output):
+        gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+        stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
+    return (current_models,output)
 
 def gen_fn(model_str, prompt):
     if model_str == 'NA':
@@ -94,7 +111,7 @@ def make_me():
                 stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
         """with gr.Accordion('Model selection'):
             model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
-            model_choice.change(update_imgbox, model_choice, output)
+            model_choice.change(update_imgbox, (gen_button,stop_button,group_model_choice), output)
             model_choice.change(extend_choices, model_choice, current_models)
         """     
         """with gr.Accordion('Group Model selection'):
@@ -105,6 +122,7 @@ def make_me():
             """group_model_choice.change(choice_group_a,group_model_choice,choices)
             group_model_choice.change(update_imgbox,choices,output)
             group_model_choice.change(extend_choices,choices,current_models)"""
+            group_model_choice.change(def_test,(gen_button,stop_button,group_model_choice),(current_models,output))
         with gr.Row():
             gr.HTML("""
                 <div class="footer">