PrintingPressDx

Running

App Files Files Community

Yntec commited on Feb 27

Commit

bd650f6

•

1 Parent(s): 08e8fe5

Update app.py

Browse files

Files changed (1) hide show

app.py +18 -28

app.py CHANGED Viewed

@@ -44,34 +44,24 @@ def gen_fn(model_str, prompt):
 with gr.Blocks() as demo:
     with gr.Tab('Printing Press'):
-        txt_input = gr.Textbox(label = 'Your prompt:', lines=4).style(container=False,min_width=1200)
-        gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
-        stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
-        gr.HTML(
-        """
-            <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
-              <div>
-                <body>
-                <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images and select models.</p>
-                </div>
-                </body>
-              </div>
-            </div>
-        """
-               )
-        with gr.Row():
-            output = [gr.Image(label = m, min_width=480) for m in default_models]
-            current_models = [gr.Textbox(m, visible = False) for m in default_models]
-            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])
-        with gr.Accordion('Model selection'):
-            #model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 686 available!', value = default_models, multiselect = True, max_choices = num_models, interactive = True, filterable = False)
-            gr.Dropdown(models, label = 'Choose a model from the 686 available!', value = models[0], filterable = False)
-            #model_choice.change(update_imgbox, model_choice, output)
-            #model_choice.change(extend_choices, model_choice, current_models)
         with gr.Row():
             gr.HTML(
     """

 with gr.Blocks() as demo:
     with gr.Tab('Printing Press'):
+        model_choice2 = gr.Dropdown(models, label = 'Choose a model from the 686 available!', value = models[0], filterable = False)
+        txt_input2 = gr.Textbox(label = 'Your prompt:')
+        current_images = 1
+        max_images = 6
+        gen_button2 = gr.Button('Generate up to 6 images in up to 3 minutes total')
+        stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)
+        gen_button2.click(lambda s: gr.update(interactive = True), None, stop_button2)
+        with gr.Column():
+            num_images = gr.Slider(current_images, max_images, value = 1, step = 1, label = 'Number of images')
+            output2 = [gr.Image(label = '') for _ in range(6)]
+        for i, o in enumerate(output2):
+            num_models = num_images
+            img_i = gr.Number(i, visible = False)
+            num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o, show_progress = False)
+            gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o)
+            stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
         with gr.Row():
             gr.HTML(
     """