Update app.py

app.py

@@ -1,8 +1,6 @@
 #@title Prepare the Concepts Library to be used
 
 
-
-
 import requests
 import os
 import gradio as gr
@@ -271,6 +269,8 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as new_welcome:
       <a href="http://www.astronaut.horse">http://www.astronaut.horse</a>
   ''')
   dropdown = gr.Dropdown([dropdown for dropdown in list(DROPDOWNS) if 'ahx-model' in dropdown], label="choose style...")
+  dropdown = gr.Dropdown(['square', 'landscape', 'portrait'], label="image size / dimensions...")
+  dropdown = gr.Dropdown(['1 image', '2 images', '3 images', '4 images'], label="output image count...")
 
   # with gr.Row():
   prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
@@ -363,10 +363,33 @@ with gr.Blocks() as beta:
   output_text = gr.Text(elem_id="output-text")
   go_button.click(fn=simple_image_prompt, inputs=[prompt, dropdown], outputs=[output, output_text])
 
+
+    
+# -----------------------------------------------------------------------------------------------
+
+# ~~~ NOISE STEPS TAB ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+
+with gr.Blocks() as noise_steps:
+  gr.Markdown('''
+      # 🧑‍🚀 Noise Steps Loader
+
+      This tool desn't exist yet!
+  ''')
+  dropdown = gr.Dropdown([dropdown for dropdown in list(DROPDOWNS) if 'ahx-beta' in dropdown], label="choose style...")
+
+  # with gr.Row():
+  prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
+
+  go_button = gr.Button("generate image", elem_id="go-button")
+  output = gr.Image(elem_id="output-image")
+  output_text = gr.Text(elem_id="output-text")
+  go_button.click(fn=simple_image_prompt, inputs=[prompt, dropdown], outputs=[output, output_text])
+
     
     
 # -----------------------------------------------------------------------------------------------
 
 
-tabbed_interface = gr.TabbedInterface([new_welcome, dropdown_tab, beta], ["Welcome!", "Advanced Prompting", "Beta Concepts"])
+tabbed_interface = gr.TabbedInterface([new_welcome, dropdown_tab, beta, noise_steps], ["Welcome!", "Advanced Prompting", "Beta Concepts", "Noise Steps"])
 tabbed_interface.launch()