Add beta tab

app.py

@@ -247,6 +247,9 @@ def simple_image_prompt(prompt, dropdown):
       )
 
 
+    
+# ~~~ WELCOME TAB ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+    
 with gr.Blocks(css=".gradio-container {max-width: 650px}") as new_welcome:
   gr.Markdown('''
       # 🧑‍🚀 Astronaut Horse Concept Loader
@@ -272,6 +275,7 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as new_welcome:
 # -----------------------------------------------------------------------------------------------
 
 
+
 def infer(text, dropdown):
   images_list = pipe(
               [f"{text} in the style of <{dropdown}>"],
@@ -284,6 +288,9 @@ def infer(text, dropdown):
 css = ""
 examples = []
 
+
+# ~~~ DEMO TAB? ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
 with gr.Blocks(css=css) as demo:
   state = gr.Variable({
         'selected': -1
@@ -321,8 +328,46 @@ with gr.Blocks(css=css) as demo:
         inputs=inputs,
         outputs=[infer_outputs, community_icon, loading_icon]
     )
-  
-  # -----------------------------------------------------------------------------------------------
+
+
+# -----------------------------------------------------------------------------------------------
+
+# ~~~ BETA TAB ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+# def infer(text, dropdown):
+#   images_list = pipe(
+#               [f"{text} in the style of <{dropdown}>"],
+#               num_inference_steps=30,
+#               guidance_scale=7.5
+#   )
+#   return images_list.images, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+
+
+# css = ""
+# examples = []
+
+with gr.Blocks() as new_welcome:
+  gr.Markdown('''
+      # 🧑‍🚀 Beta Concept Loader
+
+      This tool allows you to test out newly trained beta concepts trained by artists. These are experimental and may be removed if they are problematic or uninteresting. If they end up being interesting or successful though they'll be renamed and moved into the primary prompting drop-down.
+      
+      Artists can now freely train new models / concepts using the link below. This uses free access to Google's GPUs but will require a password / key that you can ask for on our discord. After a new concept / model is trained it will be automatically added to this tab after ~24 hours. Check it out!
+      <a href="https://colab.research.google.com/drive/1FhOpcEjHT7EN53Zv9MFLQTytZp11wjqg#scrollTo=hzUluHT-I42O">https://colab.research.google.com/astronaut-horse-training-tool</a>
+  ''')
+  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], ["Welcome!", "Advanced Prompting"])