Update app.py

app.py

@@ -99,6 +99,7 @@ def gen_fnsix(model_str, prompt):
         return None
     noisesix = str(randint(1941, 2023)) #str(randint(0, 99999999999))
     return models_load[model_str](f'{prompt} {noisesix}')
+
 with gr.Blocks() as demo:
     gr.HTML(
     """
@@ -114,73 +115,14 @@ with gr.Blocks() as demo:
         </p></div>
     """
 )  
-    with gr.Tab('One Image'):
-        model_choice = gr.Dropdown(models, label = f'Choose a model from the {len(models)} available! Try clearing the box and typing on it to filter them!', value = models[0], filterable = True)
-        txt_input = gr.Textbox(label = 'Your prompt:')
-        
-        max_imagesone = 1
-        num_imagesone = gr.Slider(1, max_imagesone, value = max_imagesone, step = 1, label = 'Nobody gets to see this label so I can put here whatever I want!', visible = False)
-        
-        gen_button = gr.Button('Generate')
-        #stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None)
-        
-        with gr.Row():
-            output = [gr.Image(label = '') for _ in range(max_imagesone)]
-
-        for i, o in enumerate(output):
-            img_in = gr.Number(i, visible = False)
-            num_imagesone.change(lambda i, n: gr.update(visible = (i < n)), [img_in, num_imagesone], o, show_progress = False)
-            gen_event = gen_button.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_in, num_imagesone, model_choice, txt_input], o, concurrency_limit=None, queue=False)
-            #stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
-        with gr.Row():
-            gr.HTML(
-    """
-        <div class="footer">
-        <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77, Omnibus's Maximum Multiplier, and <a href="https://huggingface.co/spaces/Yntec/ToyWorld">Toy World</a>!
-        </p>
-    """
-)
-    with gr.Tab('Seed it!'):
-        model_choiceseed = gr.Dropdown(models, label = f'Choose a model from the {len(models)} available! Try clearing the box and typing on it to filter them!', value = models[0], filterable = True)
-        txt_inputseed = gr.Textbox(label = 'Your prompt:')
-        seed = gr.Slider(label="Use a seed to replicate the same image later", info="Max 3999999999", minimum=0, maximum=MAX_SEED, step=1, value=1)
-        
-        max_imagesseed = 1
-        num_imagesseed = gr.Slider(1, max_imagesone, value = max_imagesone, step = 1, label = 'One, because more would make it produce identical images with the seed', visible = False)
-        
-        gen_buttonseed = gr.Button('Generate an image using the seed')
-        #stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None)
-        
-        with gr.Row():
-            outputseed = [gr.Image(label = '') for _ in range(max_imagesseed)]
-
-        for i, o in enumerate(outputseed):
-            img_is = gr.Number(i, visible = False)
-            num_imagesseed.change(lambda i, n: gr.update(visible = (i < n)), [img_is, num_imagesseed], o, show_progress = False)
-            #gen_eventseed = gen_buttonseed.click(lambda i, n, m, t, n1: gen_fnseed(m, t, n1) if (i < n) else None, [img_is, num_imagesseed, model_choiceseed, txt_inputseed, useseed], o, concurrency_limit=None, queue=False)
-
-            gen_eventseed = gr.on(triggers=[gen_buttonseed.click, txt_inputseed.submit],
-                               fn=lambda i, n, m, t, n1: gen_fnseed(m, t, n1) if (i < n) else None,
-                               inputs=[img_is, num_imagesseed, model_choiceseed, txt_inputseed, seed], outputs=[o],
-                                       concurrency_limit=None, queue=False) # Be sure to delete ", queue=False" when activating the stop button
-                        
-            #stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
-        with gr.Row():
-            gr.HTML(
-    """
-        <div class="footer">
-        <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77, Omnibus's Maximum Multiplier, and <a href="https://huggingface.co/spaces/Yntec/ToyWorld">Toy World</a>!
-        </p>
-    """
-)
     with gr.Tab('Up To Six'):
         model_choice2 = gr.Dropdown(models, label = f'Choose a model from the {len(models)} available! Try clearing the box and typing on it to filter them!', value = models[0], filterable = True)
         txt_input2 = gr.Textbox(label = 'Your prompt:')
         
         max_images = 6
         num_images = gr.Slider(1, max_images, value = max_images, step = 1, label = 'Number of images (if you want less than 6 decrease them slowly until they match the boxes below)')
+        width_images = gr.Slider(128, 1440, value = 1024, step = 32, label = 'Image width')
+        height_images = gr.Slider(128, 1440, value = 1024, step = 32, label = 'Image height')
         
         gen_button2 = gr.Button('Generate up to 6 images in up to 3 minutes total')
         #stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)